blob: a8660cfec39ecfaceda47cc8fe55c012159c4a35 [file] [log] [blame]
/*
* 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()); }
}