TestKvmNfsPrimaryStorage.java example

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

import junit.framework.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.zstack.core.cloudbus.CloudBus;
import org.zstack.core.componentloader.ComponentLoader;
import org.zstack.core.config.GlobalConfigFacade;
import org.zstack.core.db.DatabaseFacade;
import org.zstack.header.cluster.ClusterInventory;
import org.zstack.header.host.APIAddHostEvent;
import org.zstack.header.host.HostInventory;
import org.zstack.header.identity.SessionInventory;
import org.zstack.header.storage.primary.APIAddPrimaryStorageEvent;
import org.zstack.header.storage.primary.PrimaryStorageInventory;
import org.zstack.header.storage.primary.PrimaryStorageVO;
import org.zstack.header.zone.ZoneInventory;
import org.zstack.kvm.APIAddKVMHostMsg;
import org.zstack.kvm.KVMHostFactory;
import org.zstack.simulator.kvm.KVMSimulatorConfig;
import org.zstack.simulator.storage.primary.nfs.NfsPrimaryStorageSimulatorConfig;
import org.zstack.storage.primary.nfs.APIAddNfsPrimaryStorageMsg;
import org.zstack.storage.primary.nfs.NfsPrimaryStorageGlobalConfig;
import org.zstack.test.*;
import org.zstack.test.deployer.Deployer;
import org.zstack.utils.Utils;
import org.zstack.utils.data.SizeUnit;
import org.zstack.utils.logging.CLogger;
import org.zstack.utils.path.PathUtil;

import java.util.concurrent.TimeUnit;

public class TestKvmNfsPrimaryStorage {
    static CLogger logger = Utils.getLogger(TestKvmNfsPrimaryStorage.class);
    static Deployer deployer;
    static Api api;
    static ComponentLoader loader;
    static CloudBus bus;
    static DatabaseFacade dbf;
    static KVMHostFactory kvmFactory;
    static SessionInventory session;
    static KVMSimulatorConfig config;
    static NfsPrimaryStorageSimulatorConfig nfsConfig;
    static GlobalConfigFacade gcf;
    static ClusterInventory cinv;
    static WebBeanConstructor con;

    @BeforeClass
    public static void setUp() throws Exception {
        DBUtil.reDeployDB();
        con = new WebBeanConstructor();
        deployer = new Deployer("deployerXml/kvm/TestAddKvmHost.xml", con);
        deployer.addSpringConfig("KVMRelated.xml");
        deployer.build();
        api = deployer.getApi();
        loader = deployer.getComponentLoader();
        kvmFactory = loader.getComponent(KVMHostFactory.class);
        bus = loader.getComponent(CloudBus.class);
        dbf = loader.getComponent(DatabaseFacade.class);
        config = loader.getComponent(KVMSimulatorConfig.class);
        nfsConfig = loader.getComponent(NfsPrimaryStorageSimulatorConfig.class);
        gcf = loader.getComponent(GlobalConfigFacade.class);
        session = api.loginAsAdmin();
        cinv = api.listClusters(null).get(0);
    }

    private HostInventory addHost() throws ApiSenderException {
        config.connectSuccess = true;
        config.connectException = false;
        config.hostFactSuccess = true;
        config.hostFactException = false;
        config.cpuNum = 1;
        config.cpuSpeed = 2600;
        config.totalMemory = SizeUnit.GIGABYTE.toByte(8);
        config.usedMemory = SizeUnit.MEGABYTE.toByte(512);
        config.usedCpu = 512;
        ClusterInventory cinv = api.listClusters(null).get(0);
        APIAddKVMHostMsg msg = new APIAddKVMHostMsg();
        msg.setName("KVM-1");
        msg.setClusterUuid(cinv.getUuid());
        msg.setManagementIp("localhost");
        msg.setUsername("admin");
        msg.setPassword("password");
        msg.setSession(session);
        ApiSender sender = api.getApiSender();
        APIAddHostEvent evt = sender.send(msg, APIAddHostEvent.class);
        return evt.getInventory();
    }

    private PrimaryStorageInventory addPrimaryStorage() throws ApiSenderException {
        APIAddNfsPrimaryStorageMsg msg = new APIAddNfsPrimaryStorageMsg();
        msg.setName("KVM-NFS");
        msg.setUrl("10.1.1.22:/opt/zstack/");
        msg.setSession(session);
        ZoneInventory zone = deployer.zones.get("Zone1");
        msg.setZoneUuid(zone.getUuid());
        ApiSender sender = api.getApiSender();
        APIAddPrimaryStorageEvent evt = sender.send(msg, APIAddPrimaryStorageEvent.class);
        PrimaryStorageInventory inv = evt.getInventory();
        api.attachPrimaryStorage(cinv.getUuid(), inv.getUuid());
        return inv;
    }

    @Test
    public void testAttach() throws ApiSenderException, InterruptedException {
        addHost();
        nfsConfig.totalCapacity = SizeUnit.GIGABYTE.toByte(100);
        nfsConfig.availableCapacity = SizeUnit.GIGABYTE.toByte(50);
        PrimaryStorageInventory pinv = addPrimaryStorage();
        TimeUnit.SECONDS.sleep(2);
        PrimaryStorageVO pvo = dbf.findByUuid(pinv.getUuid(), PrimaryStorageVO.class);
        Assert.assertEquals(nfsConfig.totalCapacity, pvo.getCapacity().getTotalCapacity());
        Assert.assertEquals(nfsConfig.availableCapacity, pvo.getCapacity().getAvailableCapacity());
        String mountBase = NfsPrimaryStorageGlobalConfig.MOUNT_BASE.value(String.class);
        String mountPath = PathUtil.join(mountBase, "prim-" + pinv.getUuid());
        Assert.assertEquals(mountPath, pvo.getMountPath());
    }
}