TestCreateZoneRest1.java example

Explorer
zstack-master
package org.zstack.test.compute.zone;

import junit.framework.Assert;
import org.junit.Before;
import org.junit.Test;
import org.zstack.core.componentloader.ComponentLoader;
import org.zstack.core.db.DatabaseFacade;
import org.zstack.header.exception.CloudRuntimeException;
import org.zstack.header.message.APIDeleteMessage;
import org.zstack.header.rest.RESTFacade;
import org.zstack.header.zone.ZoneState;
import org.zstack.rest.RestConstants;
import org.zstack.sdk.*;
import org.zstack.test.Api;
import org.zstack.test.ApiSenderException;
import org.zstack.test.DBUtil;
import org.zstack.test.WebBeanConstructor;
import org.zstack.utils.Utils;
import org.zstack.utils.gson.JSONObjectUtil;
import org.zstack.utils.logging.CLogger;

public class TestCreateZoneRest1 {
    CLogger logger = Utils.getLogger(TestCreateZoneRest1.class);

    Api api;
    ComponentLoader loader;
    DatabaseFacade dbf;
    RESTFacade restf;

    @Before
    public void setUp() throws Exception {
        DBUtil.reDeployDB();
        WebBeanConstructor con = new WebBeanConstructor();
        /* This loads spring application context */
        loader = con.addXml("PortalForUnitTest.xml")
                .addXml("ZoneManager.xml")
                .addXml("AccountManager.xml")
                .addXml("rest.xml")
                .build();
        dbf = loader.getComponent(DatabaseFacade.class);
        restf = loader.getComponent(RESTFacade.class);
        api = new Api();
        api.startServer();
    }

    @Test
    public void test() throws ApiSenderException, InterruptedException {
        api.loginAsAdmin();

        ZSClient.configure(
                new ZSConfig.Builder()
                        .setHostname("127.0.0.1")
                        .setPort(8989)
                        .setWebHook(String.format("http://127.0.0.1:8989%s", RestConstants.UNIT_TEST_WEBHOOK_PATH))
                        .build()
        );

        CreateZoneAction action = new CreateZoneAction();
        action.sessionId = api.getAdminSession().getUuid();
        action.name = "zone1";
        CreateZoneAction.Result res = action.call();

        if (res.error != null) {
            throw new CloudRuntimeException(JSONObjectUtil.toJsonString(res.error));
        }

        logger.debug(JSONObjectUtil.toJsonString(res.value));

        QueryZoneAction qaction = new QueryZoneAction();
        qaction.sessionId = api.getAdminSession().getUuid();
        qaction.conditions.add("name=zone1");
        qaction.replyWithCount = true;
        QueryZoneAction.Result r = qaction.call();
        Assert.assertNull(r.error);

        QueryZoneResult qr = r.value;
        Assert.assertEquals(1, qr.inventories.size());
        logger.debug(JSONObjectUtil.toJsonString(r));

        ZoneInventory zone = r.value.inventories.get(0);
        ChangeZoneStateAction ca = new ChangeZoneStateAction();
        ca.uuid = zone.uuid;
        ca.sessionId = api.getAdminSession().getUuid();
        ca.stateEvent = "disable";
        ChangeZoneStateAction.Result z = ca.call();
        Assert.assertNull(z.error);
        Assert.assertEquals(ZoneState.Disabled.toString(), z.value.inventory.getState());

        DeleteZoneAction da = new DeleteZoneAction();
        da.uuid = zone.uuid;
        da.sessionId = api.getAdminSession().getUuid();
        da.deleteMode = APIDeleteMessage.DeletionMode.Enforcing.toString();
        da.call();

        qaction = new QueryZoneAction();
        qaction.sessionId = api.getAdminSession().getUuid();
        qaction.conditions.add(String.format("uuid=%s", zone.uuid));
        r = qaction.call();
        Assert.assertEquals(0, r.value.inventories.size());

        qaction = new QueryZoneAction();
        qaction.sessionId = api.getAdminSession().getUuid();
        qaction.conditions.add(String.format("name!=%s", zone.name));
        r = qaction.call();

        LogOutAction la = new LogOutAction();
        la.sessionUuid = api.getAdminSession().getUuid();
        la.call();
    }
}