| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.ambari.server.checks; |
| |
| import static java.util.Arrays.asList; |
| import static java.util.Collections.singletonList; |
| import static org.mockito.Matchers.any; |
| import static org.mockito.Matchers.eq; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.when; |
| |
| import java.util.Collections; |
| |
| import org.apache.ambari.server.AmbariException; |
| import org.apache.ambari.server.Role; |
| import org.apache.ambari.server.controller.PrereqCheckRequest; |
| import org.apache.ambari.server.metadata.ActionMetadata; |
| import org.apache.ambari.server.orm.dao.HostRoleCommandDAO; |
| import org.apache.ambari.server.orm.dao.HostRoleCommandDAO.LastServiceCheckDTO; |
| import org.apache.ambari.server.orm.dao.ServiceConfigDAO; |
| import org.apache.ambari.server.orm.entities.ServiceConfigEntity; |
| import org.apache.ambari.server.state.Cluster; |
| import org.apache.ambari.server.state.Clusters; |
| import org.apache.ambari.server.state.MaintenanceState; |
| import org.apache.ambari.server.state.Service; |
| import org.apache.ambari.server.state.ServiceComponent; |
| import org.apache.ambari.server.state.stack.PrereqCheckStatus; |
| import org.apache.ambari.server.state.stack.PrerequisiteCheck; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| import com.google.common.collect.ImmutableMap; |
| import com.google.inject.Provider; |
| |
| public class ServiceCheckValidityCheckTest { |
| private static final String CLUSTER_NAME = "cluster1"; |
| private static final long CLUSTER_ID = 1L; |
| private static final String SERVICE_NAME = "HDFS"; |
| private static final long CONFIG_CREATE_TIMESTAMP = 1461518722202L; |
| private static final long SERVICE_CHECK_START_TIME = CONFIG_CREATE_TIMESTAMP - 2000L; |
| private static final String SERVICE_COMPONENT_NAME = "service component"; |
| private ServiceCheckValidityCheck serviceCheckValidityCheck; |
| |
| private ServiceConfigDAO serviceConfigDAO; |
| private HostRoleCommandDAO hostRoleCommandDAO; |
| private Service service; |
| private ActionMetadata actionMetadata; |
| |
| @Before |
| public void setUp() throws Exception { |
| final Clusters clusters = mock(Clusters.class); |
| service = mock(Service.class); |
| serviceConfigDAO = mock(ServiceConfigDAO.class); |
| hostRoleCommandDAO = mock(HostRoleCommandDAO.class); |
| actionMetadata = new ActionMetadata(); |
| |
| serviceCheckValidityCheck = new ServiceCheckValidityCheck(); |
| serviceCheckValidityCheck.hostRoleCommandDAOProvider = new Provider<HostRoleCommandDAO>() { |
| @Override |
| public HostRoleCommandDAO get() { |
| return hostRoleCommandDAO; |
| } |
| }; |
| serviceCheckValidityCheck.serviceConfigDAOProvider = new Provider<ServiceConfigDAO>() { |
| @Override |
| public ServiceConfigDAO get() { |
| return serviceConfigDAO; |
| } |
| }; |
| serviceCheckValidityCheck.clustersProvider = new Provider<Clusters>() { |
| @Override |
| public Clusters get() { |
| return clusters; |
| } |
| }; |
| serviceCheckValidityCheck.actionMetadataProvider = new Provider<ActionMetadata>() { |
| @Override |
| public ActionMetadata get() { |
| return actionMetadata; |
| } |
| }; |
| |
| Cluster cluster = mock(Cluster.class); |
| when(clusters.getCluster(CLUSTER_NAME)).thenReturn(cluster); |
| when(cluster.getClusterId()).thenReturn(CLUSTER_ID); |
| when(cluster.getServices()).thenReturn(ImmutableMap.of(SERVICE_NAME, service)); |
| |
| when(service.getName()).thenReturn(SERVICE_NAME); |
| |
| actionMetadata.addServiceCheckAction("HDFS"); |
| } |
| |
| @Test |
| public void testWithNullCommandDetailAtCommand() throws AmbariException { |
| ServiceComponent serviceComponent = mock(ServiceComponent.class); |
| when(serviceComponent.isVersionAdvertised()).thenReturn(true); |
| |
| when(service.getMaintenanceState()).thenReturn(MaintenanceState.OFF); |
| when(service.getServiceComponents()).thenReturn(ImmutableMap.of(SERVICE_COMPONENT_NAME, serviceComponent)); |
| |
| ServiceConfigEntity serviceConfigEntity = new ServiceConfigEntity(); |
| serviceConfigEntity.setServiceName(SERVICE_NAME); |
| serviceConfigEntity.setCreateTimestamp(CONFIG_CREATE_TIMESTAMP); |
| |
| LastServiceCheckDTO lastServiceCheckDTO1 = new LastServiceCheckDTO(Role.ZOOKEEPER_QUORUM_SERVICE_CHECK.name(), SERVICE_CHECK_START_TIME); |
| LastServiceCheckDTO lastServiceCheckDTO2 = new LastServiceCheckDTO(Role.HDFS_SERVICE_CHECK.name(), SERVICE_CHECK_START_TIME); |
| |
| when(serviceConfigDAO.getLastServiceConfig(eq(CLUSTER_ID), eq(SERVICE_NAME))).thenReturn(serviceConfigEntity); |
| when(hostRoleCommandDAO.getLatestServiceChecksByRole(any(Long.class))).thenReturn(asList(lastServiceCheckDTO1, lastServiceCheckDTO2)); |
| |
| PrerequisiteCheck check = new PrerequisiteCheck(null, CLUSTER_NAME); |
| try { |
| serviceCheckValidityCheck.perform(check, new PrereqCheckRequest(CLUSTER_NAME)); |
| } catch (NullPointerException ex){ |
| Assert.fail("serviceCheckValidityCheck failed due to null at start_time were not handled"); |
| } |
| Assert.assertEquals(PrereqCheckStatus.FAIL, check.getStatus()); |
| } |
| |
| @Test |
| public void testFailWhenServiceWithOutdatedServiceCheckExists() throws AmbariException { |
| ServiceComponent serviceComponent = mock(ServiceComponent.class); |
| when(serviceComponent.isVersionAdvertised()).thenReturn(true); |
| |
| when(service.getMaintenanceState()).thenReturn(MaintenanceState.OFF); |
| when(service.getServiceComponents()).thenReturn(ImmutableMap.of(SERVICE_COMPONENT_NAME, serviceComponent)); |
| |
| ServiceConfigEntity serviceConfigEntity = new ServiceConfigEntity(); |
| serviceConfigEntity.setServiceName(SERVICE_NAME); |
| serviceConfigEntity.setCreateTimestamp(CONFIG_CREATE_TIMESTAMP); |
| |
| LastServiceCheckDTO lastServiceCheckDTO = new LastServiceCheckDTO(Role.HDFS_SERVICE_CHECK.name(), SERVICE_CHECK_START_TIME); |
| |
| when(serviceConfigDAO.getLastServiceConfig(eq(CLUSTER_ID), eq(SERVICE_NAME))).thenReturn(serviceConfigEntity); |
| when(hostRoleCommandDAO.getLatestServiceChecksByRole(any(Long.class))).thenReturn(singletonList(lastServiceCheckDTO)); |
| |
| PrerequisiteCheck check = new PrerequisiteCheck(null, CLUSTER_NAME); |
| serviceCheckValidityCheck.perform(check, new PrereqCheckRequest(CLUSTER_NAME)); |
| Assert.assertEquals(PrereqCheckStatus.FAIL, check.getStatus()); |
| } |
| |
| |
| @Test |
| public void testFailWhenServiceWithNoServiceCheckExists() throws AmbariException { |
| ServiceComponent serviceComponent = mock(ServiceComponent.class); |
| when(serviceComponent.isVersionAdvertised()).thenReturn(true); |
| |
| when(service.getMaintenanceState()).thenReturn(MaintenanceState.OFF); |
| when(service.getServiceComponents()).thenReturn(ImmutableMap.of(SERVICE_COMPONENT_NAME, serviceComponent)); |
| |
| ServiceConfigEntity serviceConfigEntity = new ServiceConfigEntity(); |
| serviceConfigEntity.setServiceName(SERVICE_NAME); |
| serviceConfigEntity.setCreateTimestamp(CONFIG_CREATE_TIMESTAMP); |
| |
| when(serviceConfigDAO.getLastServiceConfig(eq(CLUSTER_ID), eq(SERVICE_NAME))).thenReturn(serviceConfigEntity); |
| when(hostRoleCommandDAO.getLatestServiceChecksByRole(any(Long.class))).thenReturn(Collections.<LastServiceCheckDTO>emptyList()); |
| |
| PrerequisiteCheck check = new PrerequisiteCheck(null, CLUSTER_NAME); |
| serviceCheckValidityCheck.perform(check, new PrereqCheckRequest(CLUSTER_NAME)); |
| Assert.assertEquals(PrereqCheckStatus.FAIL, check.getStatus()); |
| } |
| |
| @Test |
| public void testFailWhenServiceWithOutdatedServiceCheckExistsRepeated() throws AmbariException { |
| ServiceComponent serviceComponent = mock(ServiceComponent.class); |
| when(serviceComponent.isVersionAdvertised()).thenReturn(true); |
| |
| when(service.getMaintenanceState()).thenReturn(MaintenanceState.OFF); |
| when(service.getServiceComponents()).thenReturn(ImmutableMap.of(SERVICE_COMPONENT_NAME, serviceComponent)); |
| |
| ServiceConfigEntity serviceConfigEntity = new ServiceConfigEntity(); |
| serviceConfigEntity.setServiceName(SERVICE_NAME); |
| serviceConfigEntity.setCreateTimestamp(CONFIG_CREATE_TIMESTAMP); |
| |
| LastServiceCheckDTO lastServiceCheckDTO1 = new LastServiceCheckDTO(Role.HDFS_SERVICE_CHECK.name(), SERVICE_CHECK_START_TIME); |
| LastServiceCheckDTO lastServiceCheckDTO2 = new LastServiceCheckDTO(Role.HDFS_SERVICE_CHECK.name(), CONFIG_CREATE_TIMESTAMP - 1L); |
| |
| when(serviceConfigDAO.getLastServiceConfig(eq(CLUSTER_ID), eq(SERVICE_NAME))).thenReturn(serviceConfigEntity); |
| when(hostRoleCommandDAO.getLatestServiceChecksByRole(any(Long.class))).thenReturn(asList(lastServiceCheckDTO1, lastServiceCheckDTO2)); |
| |
| PrerequisiteCheck check = new PrerequisiteCheck(null, CLUSTER_NAME); |
| serviceCheckValidityCheck.perform(check, new PrereqCheckRequest(CLUSTER_NAME)); |
| Assert.assertEquals(PrereqCheckStatus.FAIL, check.getStatus()); |
| } |
| |
| /** |
| * Tests that old, oudated service checks for the FOO2 service doesn't cause |
| * problems when checking values for the FOO service. |
| * <p/> |
| * The specific test case here is that the FOO2 service was added a long time |
| * ago and then removed. We don't want old service checks for FOO2 to match |
| * when querying for FOO. |
| * |
| * @throws AmbariException |
| */ |
| @Test |
| public void testPassWhenSimilarlyNamedServiceIsOutdated() throws AmbariException { |
| ServiceComponent serviceComponent = mock(ServiceComponent.class); |
| when(serviceComponent.isVersionAdvertised()).thenReturn(true); |
| |
| when(service.getMaintenanceState()).thenReturn(MaintenanceState.OFF); |
| when(service.getServiceComponents()).thenReturn(ImmutableMap.of(SERVICE_COMPONENT_NAME, serviceComponent)); |
| |
| ServiceConfigEntity serviceConfigEntity = new ServiceConfigEntity(); |
| serviceConfigEntity.setServiceName(SERVICE_NAME); |
| serviceConfigEntity.setCreateTimestamp(CONFIG_CREATE_TIMESTAMP); |
| |
| String hdfsRole = Role.HDFS_SERVICE_CHECK.name(); |
| String hdfs2Role = hdfsRole.replace("HDFS", "HDFS2"); |
| |
| LastServiceCheckDTO lastServiceCheckDTO1 = new LastServiceCheckDTO(hdfsRole, SERVICE_CHECK_START_TIME); |
| LastServiceCheckDTO lastServiceCheckDTO2 = new LastServiceCheckDTO(hdfs2Role, CONFIG_CREATE_TIMESTAMP - 1L); |
| |
| when(serviceConfigDAO.getLastServiceConfig(eq(CLUSTER_ID), eq(SERVICE_NAME))).thenReturn(serviceConfigEntity); |
| when(hostRoleCommandDAO.getLatestServiceChecksByRole(any(Long.class))).thenReturn(asList(lastServiceCheckDTO1, lastServiceCheckDTO2)); |
| |
| PrerequisiteCheck check = new PrerequisiteCheck(null, CLUSTER_NAME); |
| serviceCheckValidityCheck.perform(check, new PrereqCheckRequest(CLUSTER_NAME)); |
| Assert.assertEquals(PrereqCheckStatus.FAIL, check.getStatus()); } |
| } |