InsideAgentSimulationTest.java

/*
 * RHQ Management Platform
 * Copyright (C) 2005-2012 Red Hat, Inc.
 * All rights reserved.
 *
 * This program 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 version 2 of the License.
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

package org.rhq.test.arquillian;

import static org.mockito.Matchers.any;
import static org.mockito.Mockito.when;

import java.util.Set;

import org.testng.Assert;
import org.testng.annotations.Test;

import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.TargetsContainer;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.arquillian.testng.Arquillian;
import org.jboss.shrinkwrap.api.ShrinkWrap;

import org.rhq.core.clientapi.server.discovery.InventoryReport;
import org.rhq.core.domain.resource.InventoryStatus;
import org.rhq.core.domain.resource.Resource;
import org.rhq.test.shrinkwrap.RhqAgentPluginArchive;

/**
 * Test for making sure that we can simulate the plugin container running inside a full agent
 * being connected to a mocked out server.
 *
 * @author Lukas Krejci
 */
public class InsideAgentSimulationTest extends Arquillian {

    @Deployment
    @TargetsContainer("connected-pc")
    public static RhqAgentPluginArchive getTestPlugin() {
        return ShrinkWrap.create(RhqAgentPluginArchive.class, "test-deep-plugin-1.0.0.jar")
            .addClasses(TestDiscoveryComponent.class, TestResourceComponent.class)
            .setPluginDescriptor("test-deep-rhq-plugin.xml");
    }

    @ArquillianResource
    private MockingServerServices serverServices;

    @DiscoveredResources(plugin = "testDeepPlugin", resourceType = "TestServer")
    private Set<Resource> discoveredServers;

    @DiscoveredResources(plugin = "testDeepPlugin", resourceType = "TestService")
    private Set<Resource> discoveredServices;

    private FakeServerInventory fakeServerInventory;

    private FakeServerInventory.CompleteDiscoveryChecker discoveryCompleteChecker;

    @BeforeDiscovery(order = 1)
    public void resetServerServices() {
        serverServices.resetMocks();
        fakeServerInventory = new FakeServerInventory();
    }

    @BeforeDiscovery(testMethods = "testDeepDiscovery", order = 2)
    public void setupDiscoveryMocks() throws Exception {
        discoveryCompleteChecker = fakeServerInventory.createAsyncDiscoveryCompletionChecker(3);
        //autoimport everything
        when(serverServices.getDiscoveryServerService().mergeInventoryReport(any(InventoryReport.class))).then(
            fakeServerInventory.mergeInventoryReport(InventoryStatus.COMMITTED));
        when(serverServices.getDiscoveryServerService().getResourceSyncInfo(any(int.class))).then(
            fakeServerInventory.getResourceSyncInfo());
    }

    @AfterDiscovery
    public void waitForAsyncDiscoveries() throws Exception {
        if (discoveryCompleteChecker != null) {
            discoveryCompleteChecker.waitForDiscoveryComplete();
        }
    }

    //we need to make sure that the no discovery test is run first, because the plugin container
    //would keep the inventory from the previous test.
    //the other two tests get each a new serverside, which, when synced with the PC, will cause
    //the PC to clear and rediscover the resources (the serverside will have an empty inventory
    //at first and hence the PC will clear its inventory, too).
    @Test
    public void testNoDiscovery() {
        Assert.assertEquals(discoveredServers.size(), 0, "There should be no server discovered");
        Assert.assertEquals(discoveredServices.size(), 0, "There should be no service discovered");
    }

    //the difference between this test and the deep discovery one is that for this test
    //the mocks should not be set up and hence only a top server discovery should occur
    @Test(dependsOnMethods = "testNoDiscovery")
    @RunDiscovery
    public void testShallowDiscovery() throws Exception {
        Assert.assertEquals(discoveredServers.size(), 1, "There should be 1 server discovered");
        Assert.assertEquals(discoveredServices.size(), 0, "There should be no service discovered");
    }

    @Test(dependsOnMethods = "testNoDiscovery")
    @RunDiscovery
    public void testDeepDiscovery() throws Exception {
        Assert.assertEquals(discoveredServers.size(), 1, "There should be 1 server discovered");
        Assert.assertEquals(discoveredServices.size(), 1, "There should be 1 service discovered");
    }
}