@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);
  }