view system-backend/src/test/java/com/redhat/thermostat/backend/system/SystemBackendTest.java @ 910:421d8a954893

Generic agent-side API for VM start/stop Introduce an API that all backends can use to find out when a VM starts up or shuts down. Currently, all backends re-implement this functionality using jvmstat explicitly. With an explicit API for this, we will have less duplicated code for each plugin and we can change/improve the implementation that detects when VMs start or stop without having to change each plugin. All classes implementing VmStatusChangeListener registered as OSGi services now get notified about the VM's pid when a vm starts or stops. The GC backend now obtains the pid from this new API but still uses the MonitoredVm class form jvmstat connect to the VM and gather data. Reviewed-by: ebaron, vanaltj Review-thread: http://icedtea.classpath.org/pipermail/thermostat/2013-January/005122.html PR1258
author Omair Majid <omajid@redhat.com>
date Tue, 15 Jan 2013 11:38:58 -0500
parents 9e6bcfc40ea1
children ccac9bcabaa0
line wrap: on
line source

/*
 * Copyright 2013 Red Hat, Inc.
 *
 * This file is part of Thermostat.
 *
 * Thermostat is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published
 * by the Free Software Foundation; either version 2, or (at your
 * option) any later version.
 *
 * Thermostat is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Thermostat; see the file COPYING.  If not see
 * <http://www.gnu.org/licenses/>.
 *
 * Linking this code with other modules is making a combined work
 * based on this code.  Thus, the terms and conditions of the GNU
 * General Public License cover the whole combination.
 *
 * As a special exception, the copyright holders of this code give
 * you permission to link this code with independent modules to
 * produce an executable, regardless of the license terms of these
 * independent modules, and to copy and distribute the resulting
 * executable under terms of your choice, provided that you also
 * meet, for each linked independent module, the terms and conditions
 * of the license of that module.  An independent module is a module
 * which is not derived from or based on this code.  If you modify
 * this code, you may extend this exception to your version of the
 * library, but you are not obligated to do so.  If you do not wish
 * to do so, delete this exception statement from your version.
 */

package com.redhat.thermostat.backend.system;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.junit.Before;
import org.junit.Test;

import com.redhat.thermostat.common.dao.DAOFactory;
import com.redhat.thermostat.common.dao.HostInfoDAO;
import com.redhat.thermostat.common.dao.NetworkInterfaceInfoDAO;
import com.redhat.thermostat.storage.core.Storage;

public class SystemBackendTest {

    private SystemBackend b;

    @Before
    public void setUp() {
        Storage s = mock(Storage.class);
        HostInfoDAO hDAO = mock(HostInfoDAO.class);
        NetworkInterfaceInfoDAO nDAO = mock(NetworkInterfaceInfoDAO.class);
        DAOFactory df = mock(DAOFactory.class);
        when(df.getStorage()).thenReturn(s);
        when(df.getHostInfoDAO()).thenReturn(hDAO);
        when(df.getNetworkInterfaceInfoDAO()).thenReturn(nDAO);

        VmStatusChangeNotifier notifier = mock(VmStatusChangeNotifier.class);

        b = new SystemBackend(notifier);
        b.setDAOFactory(df);
    }

    @Test
    public void testBasicBackend() {
        assertFalse(b.isActive());
        b.activate();
        assertTrue(b.isActive());
        b.deactivate();
        assertFalse(b.isActive());
    }

    @Test
    public void testActivateTwice() {
        b.activate();
        b.activate();
        assert(b.isActive());
    }

    @Test
    public void testDeactiateWhenNotActive() {
        b.deactivate();
        b.deactivate();
        assertFalse(b.isActive());
    }

}