@Test
public void testCountVmsOnHost() throws Throwable {
HostEntity hostEntity = new HostEntity();
hostEntity.setAddress(vm.host);
hostEntity.setId(hostId);
int countOfVmsOnHost = vmDcpBackend.countVmsOnHost(hostEntity);
assertThat(countOfVmsOnHost, is(1));
}
@BeforeMethod
public void setUp() {
taskCommand = mock(TaskCommand.class);
stepBackend = mock(StepBackend.class);
hostBackend = mock(HostBackend.class);
deployerClient = mock(DeployerClient.class);
hostEntity = new HostEntity();
hostEntity.setId("host-1");
hostEntity.setUsername("username");
hostEntity.setPassword("password");
hostEntity.setAddress("192.168.0.1");
hostEntity.setUsageTags(UsageTag.MGMT.toString());
step = new StepEntity();
step.setId("step-1");
step.addResource(hostEntity);
command = new HostEnterSuspendedModeStepCmd(taskCommand, stepBackend, step, hostBackend);
when(taskCommand.getDeployerClient()).thenReturn(deployerClient);
}
