TestMultipleNode6.java example

Explorer
zstack-master
package org.zstack.test.multinodes;

import junit.framework.Assert;
import org.junit.Before;
import org.junit.Test;
import org.zstack.core.Platform;
import org.zstack.core.cloudbus.CloudBus;
import org.zstack.core.componentloader.ComponentLoader;
import org.zstack.core.thread.AsyncThread;
import org.zstack.header.errorcode.SysErrors;
import org.zstack.header.managementnode.ManagementNodeInventory;
import org.zstack.header.message.MessageReply;
import org.zstack.test.Api;
import org.zstack.test.ApiSenderException;
import org.zstack.test.BeanConstructor;
import org.zstack.utils.CollectionUtils;
import org.zstack.utils.Utils;
import org.zstack.utils.function.Function;
import org.zstack.utils.logging.CLogger;

import java.util.Arrays;
import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * start 2 nodes, call silent message to the node0 which won't reply
 * stop node0, we should receive error message which indicates node0 is unavailable
 */
public class TestMultipleNode6 {
    CLogger logger = Utils.getLogger(TestMultipleNode6.class);
    ComponentLoader loader;
    NodeManager nodeMgr;
    CloudBus bus;
    Api api;
    boolean success1 = false;
    boolean success2 = false;

    @Before
    public void setUp() throws Exception {
        BeanConstructor con = new BeanConstructor();
        con.addXml("PortalForUnitTest.xml");
        con.addXml("AccountManager.xml");
        // have to load silentService, otherwise message tracker won't care it
        con.addXml("silentService.xml");
        loader = con.build();
        bus = loader.getComponent(CloudBus.class);
        nodeMgr = new NodeManager();
        api = new Api();
        api.startServer();
        api.setTimeout(300);
    }


    @AsyncThread
    private void doTest1() throws ApiSenderException {
        List<ManagementNodeInventory> nodes = api.listManagementNodes();
        ManagementNodeInventory target = CollectionUtils.find(nodes, new Function<ManagementNodeInventory, ManagementNodeInventory>() {
            @Override
            public ManagementNodeInventory call(final ManagementNodeInventory arg) {
                if (!arg.getUuid().equals(Platform.getManagementServerId())) {
                    return arg;
                }
                return null;
            }
        });

        NeedReplySilentMsg smsg = new NeedReplySilentMsg();
        bus.makeServiceIdByManagementNodeId(smsg, SilentService.SERVICE_ID, target.getUuid());
        MessageReply reply = bus.call(smsg);
        if (!reply.isSuccess() && reply.getError().getCode().equals(SysErrors.MANAGEMENT_NODE_UNAVAILABLE_ERROR.toString())) {
            success1 = true;
        }
    }

    @AsyncThread
    private void doTest2() throws ApiSenderException {
        List<ManagementNodeInventory> nodes = api.listManagementNodes();
        ManagementNodeInventory target = CollectionUtils.find(nodes, new Function<ManagementNodeInventory, ManagementNodeInventory>() {
            @Override
            public ManagementNodeInventory call(final ManagementNodeInventory arg) {
                if (!arg.getUuid().equals(Platform.getManagementServerId())) {
                    return arg;
                }
                return null;
            }
        });

        NeedReplySilentMsg smsg = new NeedReplySilentMsg();
        bus.makeServiceIdByManagementNodeId(smsg, SilentService.SERVICE_ID, target.getUuid());
        List<MessageReply> replies = bus.call(Arrays.asList(smsg));
        MessageReply reply = replies.get(0);
        if (!reply.isSuccess() && reply.getError().getCode().equals(SysErrors.MANAGEMENT_NODE_UNAVAILABLE_ERROR.toString())) {
            success2 = true;
        }
    }

    @Test
    public void test() throws InterruptedException, ApiSenderException {
        try {
            nodeMgr.setProperties("-DbFacadeDataSource.maxPoolSize=50")
                    .createLightWeightNodeWithCore(1, 300, Arrays.asList("silentService.xml"));

            doTest1();
            doTest2();
            TimeUnit.SECONDS.sleep(3);

            nodeMgr.stopNode(0, 120);

            TimeUnit.SECONDS.sleep(5);
            Assert.assertTrue(success1);
            Assert.assertTrue(success2);
        } finally {
            nodeMgr.stopNodes(120);
        }
    }
}