HostMaintenanceModeCheck extends AbstractCheckDescriptor { @Override public void perform(PrerequisiteCheck prerequisiteCheck, PrereqCheckRequest request) throws AmbariException { final String clusterName = request.getClusterName(); final Cluster cluster = clustersProvider.get().getCluster(clusterName); Collection<Host> hosts = cluster.getHosts(); for (Host host : hosts) { MaintenanceState maintenanceState = host.getMaintenanceState(cluster.getClusterId()); if (maintenanceState != MaintenanceState.OFF) { prerequisiteCheck.getFailedOn().add(host.getHostName()); } } if (!prerequisiteCheck.getFailedOn().isEmpty()) { PrereqCheckStatus status = request.getUpgradeType() == UpgradeType.HOST_ORDERED ? PrereqCheckStatus.FAIL : PrereqCheckStatus.WARNING; prerequisiteCheck.setStatus(status); String failReason = request.getUpgradeType() == UpgradeType.HOST_ORDERED ? getFailReason(KEY_CANNOT_START_HOST_ORDERED, prerequisiteCheck, request) : getFailReason(prerequisiteCheck, request); prerequisiteCheck.setFailReason(failReason); } } HostMaintenanceModeCheck(); @Override void perform(PrerequisiteCheck prerequisiteCheck, PrereqCheckRequest request); static final String KEY_CANNOT_START_HOST_ORDERED; }
@Test public void testPerform() throws Exception { final HostMaintenanceModeCheck hostMaintenanceModeCheck = new HostMaintenanceModeCheck(); hostMaintenanceModeCheck.clustersProvider = new Provider<Clusters>() { @Override public Clusters get() { return clusters; } }; final Cluster cluster = Mockito.mock(Cluster.class); Mockito.when(cluster.getClusterId()).thenReturn(1L); Mockito.when(cluster.getCurrentStackVersion()).thenReturn(new StackId("HDP", "2.2")); Mockito.when(clusters.getCluster("cluster")).thenReturn(cluster); final List<Host> hosts = new ArrayList<>(); final Host host1 = Mockito.mock(Host.class); final Host host2 = Mockito.mock(Host.class); final Host host3 = Mockito.mock(Host.class); Mockito.when(host1.getMaintenanceState(1L)).thenReturn(MaintenanceState.OFF); Mockito.when(host2.getMaintenanceState(1L)).thenReturn(MaintenanceState.OFF); Mockito.when(host3.getMaintenanceState(1L)).thenReturn(MaintenanceState.OFF); hosts.add(host1); hosts.add(host2); hosts.add(host3); Mockito.when(cluster.getHosts()).thenReturn(hosts); PrerequisiteCheck check = new PrerequisiteCheck(null, null); hostMaintenanceModeCheck.perform(check, new PrereqCheckRequest("cluster")); Assert.assertEquals(PrereqCheckStatus.PASS, check.getStatus()); check = new PrerequisiteCheck(null, null); Mockito.when(host3.getMaintenanceState(1L)).thenReturn(MaintenanceState.ON); hostMaintenanceModeCheck.perform(check, new PrereqCheckRequest("cluster")); Assert.assertEquals(PrereqCheckStatus.WARNING, check.getStatus()); }
@Test public void testPerformHostOrdered() throws Exception { final HostMaintenanceModeCheck hostMaintenanceModeCheck = new HostMaintenanceModeCheck(); hostMaintenanceModeCheck.clustersProvider = new Provider<Clusters>() { @Override public Clusters get() { return clusters; } }; final Cluster cluster = Mockito.mock(Cluster.class); Mockito.when(cluster.getClusterId()).thenReturn(1L); Mockito.when(cluster.getCurrentStackVersion()).thenReturn(new StackId("HDP", "2.2")); Mockito.when(clusters.getCluster("cluster")).thenReturn(cluster); final List<Host> hosts = new ArrayList<>(); final Host host1 = Mockito.mock(Host.class); final Host host2 = Mockito.mock(Host.class); final Host host3 = Mockito.mock(Host.class); Mockito.when(host1.getMaintenanceState(1L)).thenReturn(MaintenanceState.OFF); Mockito.when(host2.getMaintenanceState(1L)).thenReturn(MaintenanceState.OFF); Mockito.when(host3.getMaintenanceState(1L)).thenReturn(MaintenanceState.OFF); Mockito.when(host1.getHostName()).thenReturn("h1"); Mockito.when(host2.getHostName()).thenReturn("h2"); Mockito.when(host3.getHostName()).thenReturn("h3"); hosts.add(host1); hosts.add(host2); hosts.add(host3); Mockito.when(cluster.getHosts()).thenReturn(hosts); PrerequisiteCheck check = new PrerequisiteCheck(null, null); hostMaintenanceModeCheck.perform(check, new PrereqCheckRequest("cluster")); Assert.assertEquals(PrereqCheckStatus.PASS, check.getStatus()); check = new PrerequisiteCheck(null, null); Mockito.when(host3.getMaintenanceState(1L)).thenReturn(MaintenanceState.ON); hostMaintenanceModeCheck.perform(check, new PrereqCheckRequest("cluster", UpgradeType.HOST_ORDERED)); Assert.assertEquals(PrereqCheckStatus.FAIL, check.getStatus()); Assert.assertEquals("The following hosts cannot be in Maintenance Mode: h3.", check.getFailReason()); }