@Test
public void testWarningWhenHostWithVersionMismatchExists() throws Exception {

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