@Test
public void testWarningWhenHostWithVersionMismatchDoesNotExist() throws Exception {

  when(firstServiceComponentHosts.get(FIRST_SERVICE_COMPONENT_HOST_NAME).getUpgradeState()).thenReturn(IN_PROGRESS);
  PrerequisiteCheck check = new PrerequisiteCheck(null, CLUSTER_NAME);
  versionMismatchCheck.perform(check, new PrereqCheckRequest(CLUSTER_NAME));
  Assert.assertEquals(PrereqCheckStatus.PASS, check.getStatus());
}