|
|
@@ -0,0 +1,141 @@
|
|
|
+/**
|
|
|
+ * 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.serveraction.upgrades;
|
|
|
+
|
|
|
+import static org.easymock.EasyMock.anyObject;
|
|
|
+import static org.easymock.EasyMock.expect;
|
|
|
+import static org.easymock.EasyMock.replay;
|
|
|
+import static org.junit.Assert.assertEquals;
|
|
|
+import static org.junit.Assert.assertNotNull;
|
|
|
+import static org.junit.Assert.assertTrue;
|
|
|
+
|
|
|
+import java.lang.reflect.Field;
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+import org.apache.ambari.server.actionmanager.ExecutionCommandWrapper;
|
|
|
+import org.apache.ambari.server.actionmanager.HostRoleCommand;
|
|
|
+import org.apache.ambari.server.agent.CommandReport;
|
|
|
+import org.apache.ambari.server.agent.ExecutionCommand;
|
|
|
+import org.apache.ambari.server.state.Cluster;
|
|
|
+import org.apache.ambari.server.state.Clusters;
|
|
|
+import org.apache.ambari.server.state.SecurityType;
|
|
|
+import org.apache.ambari.server.state.Config;
|
|
|
+import org.apache.ambari.server.state.ConfigImpl;
|
|
|
+import org.easymock.EasyMock;
|
|
|
+import org.junit.Before;
|
|
|
+import org.junit.Test;
|
|
|
+
|
|
|
+import com.google.inject.Injector;
|
|
|
+
|
|
|
+
|
|
|
+public class RangerKmsProxyConfigTest {
|
|
|
+ private Injector m_injector;
|
|
|
+ private Clusters m_clusters;
|
|
|
+ private Field m_clusterField;
|
|
|
+
|
|
|
+ @Before
|
|
|
+ public void setup() throws Exception {
|
|
|
+ m_injector = EasyMock.createMock(Injector.class);
|
|
|
+ m_clusters = EasyMock.createMock(Clusters.class);
|
|
|
+ Cluster cluster = EasyMock.createMock(Cluster.class);
|
|
|
+
|
|
|
+ Config rangerEnv = new ConfigImpl("ranger-env") {
|
|
|
+ Map<String, String> mockProperties = new HashMap<String, String>() {{
|
|
|
+ put("ranger_user", "ranger");
|
|
|
+ }};
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public Map<String, String> getProperties() {
|
|
|
+ return mockProperties;
|
|
|
+ }
|
|
|
+ };
|
|
|
+
|
|
|
+ Config kmsSite = new ConfigImpl("kms-site") {
|
|
|
+ Map<String, String> mockProperties = new HashMap<String, String>();
|
|
|
+ @Override
|
|
|
+ public Map<String, String> getProperties() {
|
|
|
+ return mockProperties;
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void setProperties(Map<String, String> properties) {
|
|
|
+ mockProperties.putAll(properties);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void persist(boolean newConfig) {
|
|
|
+ // no-op
|
|
|
+ }
|
|
|
+ };
|
|
|
+
|
|
|
+ expect(cluster.getDesiredConfigByType("ranger-env")).andReturn(rangerEnv).atLeastOnce();
|
|
|
+ expect(cluster.getDesiredConfigByType("kms-site")).andReturn(kmsSite).atLeastOnce();
|
|
|
+ expect(m_clusters.getCluster((String) anyObject())).andReturn(cluster).anyTimes();
|
|
|
+ expect(m_injector.getInstance(Clusters.class)).andReturn(m_clusters).atLeastOnce();
|
|
|
+ expect(cluster.getSecurityType()).andReturn(SecurityType.KERBEROS).anyTimes();
|
|
|
+
|
|
|
+ replay(m_injector, m_clusters, cluster);
|
|
|
+
|
|
|
+ m_clusterField = RangerKmsProxyConfig.class.getDeclaredField("m_clusters");
|
|
|
+ m_clusterField.setAccessible(true);
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testAction() throws Exception {
|
|
|
+
|
|
|
+ Map<String, String> commandParams = new HashMap<String, String>();
|
|
|
+ commandParams.put("clusterName", "c1");
|
|
|
+
|
|
|
+ ExecutionCommand executionCommand = new ExecutionCommand();
|
|
|
+ executionCommand.setCommandParams(commandParams);
|
|
|
+ executionCommand.setClusterName("c1");
|
|
|
+
|
|
|
+ HostRoleCommand hrc = EasyMock.createMock(HostRoleCommand.class);
|
|
|
+ expect(hrc.getRequestId()).andReturn(1L).anyTimes();
|
|
|
+ expect(hrc.getStageId()).andReturn(2L).anyTimes();
|
|
|
+ expect(hrc.getExecutionCommandWrapper()).andReturn(new ExecutionCommandWrapper(executionCommand)).anyTimes();
|
|
|
+ replay(hrc);
|
|
|
+
|
|
|
+ RangerKmsProxyConfig action = new RangerKmsProxyConfig();
|
|
|
+ m_clusterField.set(action, m_clusters);
|
|
|
+
|
|
|
+ action.setExecutionCommand(executionCommand);
|
|
|
+ action.setHostRoleCommand(hrc);
|
|
|
+
|
|
|
+ CommandReport report = action.execute(null);
|
|
|
+ assertNotNull(report);
|
|
|
+
|
|
|
+ Cluster c = m_clusters.getCluster("c1");
|
|
|
+ Config config = c.getDesiredConfigByType("kms-site");
|
|
|
+ Map<String, String> map = config.getProperties();
|
|
|
+
|
|
|
+ assertTrue(map.containsKey("hadoop.kms.proxyuser.ranger.users"));
|
|
|
+ assertTrue(map.containsKey("hadoop.kms.proxyuser.ranger.groups"));
|
|
|
+ assertTrue(map.containsKey("hadoop.kms.proxyuser.ranger.hosts"));
|
|
|
+
|
|
|
+
|
|
|
+ assertEquals("*", map.get("hadoop.kms.proxyuser.ranger.users"));
|
|
|
+ assertEquals("*", map.get("hadoop.kms.proxyuser.ranger.groups"));
|
|
|
+ assertEquals("*", map.get("hadoop.kms.proxyuser.ranger.hosts"));
|
|
|
+
|
|
|
+ report = action.execute(null);
|
|
|
+ assertNotNull(report);
|
|
|
+
|
|
|
+ }
|
|
|
+}
|
Mugdha Varadkar