VMManagerAppUnitTest.java

package com.neverwinterdp.vm.jvm;


import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import com.beust.jcommander.JCommander;
import com.neverwinterdp.registry.zk.RegistryImpl;
import com.neverwinterdp.vm.VMConfig;
import com.neverwinterdp.vm.VMDescriptor;
import com.neverwinterdp.vm.VMDummyApp;
import com.neverwinterdp.vm.VMStatus;
import com.neverwinterdp.vm.client.VMClient;
import com.neverwinterdp.vm.client.shell.Shell;
import com.neverwinterdp.vm.command.CommandResult;
import com.neverwinterdp.vm.command.VMCommand;
import com.neverwinterdp.vm.event.VMWaitingEventListener;
import com.neverwinterdp.vm.service.VMServiceCommand;
import com.neverwinterdp.vm.tool.VMZKClusterBuilder;

public class VMManagerAppUnitTest  {
  VMZKClusterBuilder  vmCluster ;
  Shell      shell;
  VMClient   vmClient;
  
  @Before
  public void setup() throws Exception {
    vmCluster = new VMZKClusterBuilder() ;
    vmCluster.clean();
    vmCluster.starZookeeper();
    Thread.sleep(5000);
  }
  
  @After
  public void teardown() throws Exception {
    vmCluster.shutdown();
  }
  
  @Test
  public void testMaster() throws Exception {
    try {
      VMWaitingEventListener master1waitingListener = vmCluster.createVMMaster("vm-master-1");
      master1waitingListener.waitForEvents(5000);
      
      vmCluster.createVMMaster("vm-master-2");
      Thread.sleep(2000);
      
      vmClient = vmCluster.getVMClient();
      shell = new Shell(vmClient) ;
      shell.execute("registry dump");
      
      VMWaitingEventListener eventsListener = new VMWaitingEventListener(shell.getVMClient().getRegistry());
      banner("Create VM Dummy 1");
      eventsListener.waitVMStatus("Expect vm-dummy-1 with running status", "vm-dummy-1", VMStatus.RUNNING);
      eventsListener.waitHeartbeat("Expect vm-dummy-1 has connected heartbeat", "vm-dummy-1", true);
      VMDescriptor vmDummy1 = allocate(vmClient, "vm-dummy-1") ;
      eventsListener.waitForEvents(5000);

      shell.execute("registry dump");

      banner("Shutdown VM Master 1");
      //shutdown vm master 1 , the vm-master-2 should pickup the leader role.
      eventsListener.waitVMStatus("Expect vm-master-1 with running status", "vm-master-1", VMStatus.TERMINATED);
      eventsListener.waitHeartbeat("Expect vm-master-1 has connected heartbeat", "vm-master-1", false);
      eventsListener.waitVMMaster("Expect the vm-master-2 will be elected", "vm-master-2");
      vmClient.shutdown(vmClient.getMasterVMDescriptor());
      //vmMaster1.shutdown();
      eventsListener.waitForEvents(5000);
      shell.execute("registry dump");

      banner("Create VM Dummy 2");
      VMDescriptor vmDummy2 = allocate(vmClient, "vm-dummy-2") ;
      shell.execute("registry dump");

      banner("Shutdown VM Dummy 1 and 2");
      eventsListener.waitVMStatus("Expect vm-dummy-1 terminated status", "vm-dummy-1", VMStatus.TERMINATED);
      eventsListener.waitVMStatus("Expect vm-dummy-2 terminated status", "vm-dummy-2", VMStatus.TERMINATED);
      Assert.assertTrue(shutdown(vmClient, vmDummy2));
      Assert.assertTrue(shutdown(vmClient, vmDummy1));
      eventsListener.waitForEvents(10000);
      vmClient.shutdown();
    } finally {
      Thread.sleep(3000);
      shell.execute("registry dump");
      shell.execute("help");
    }
  }

  private void banner(String title) {
    System.out.println("\n\n");
    System.out.println("------------------------------------------------------------------------");
    System.out.println(title);
    System.out.println("------------------------------------------------------------------------");
  }
  
  private VMDescriptor allocate(VMClient vmClient, String name) throws Exception {
    VMDescriptor masterVMDescriptor = vmClient.getMasterVMDescriptor();
    String[] args = {
      "--name", name,
      "--role", "dummy",
      "--registry-connect", "127.0.0.1:2181", 
      "--registry-db-domain", "/NeverwinterDP",
      "--registry-implementation", RegistryImpl.class.getName(),
      "--vm-application", VMDummyApp.class.getName()
    };
    VMConfig vmConfig = new VMConfig() ;
    new JCommander(vmConfig, args);
    CommandResult<?> result = vmClient.execute(masterVMDescriptor, new VMServiceCommand.Allocate(vmConfig));
    Assert.assertNull(result.getErrorStacktrace());
    VMDescriptor vmDescriptor = result.getResultAs(VMDescriptor.class);
    Assert.assertNotNull(vmDescriptor);
    return vmDescriptor;
  }
  
  private boolean shutdown(VMClient vmClient, VMDescriptor vmDescriptor) throws Exception {
    CommandResult<?> result = vmClient.execute(vmDescriptor, new VMCommand.Shutdown());
    Assert.assertNull(result.getErrorStacktrace());
    return result.getResultAs(Boolean.class);
  }
}