/**
* 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.hadoop.gateway.security.principal;
import org.apache.hadoop.test.category.FastTests;
import org.apache.hadoop.test.category.UnitTests;
import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
/**
*
*/
@Category( { UnitTests.class, FastTests.class } )
public class PrincipalMapperTest {
PrincipalMapper mapper;
@Before
public void setup() {
mapper = new SimplePrincipalMapper();
}
@Test
public void testSimplePrincipalMappingWithWildcardGroups() {
String principalMapping = "";
String groupMapping = "*=users";
try {
mapper.loadMappingTable(principalMapping, groupMapping);
}
catch (PrincipalMappingException pme) {
pme.printStackTrace();
fail();
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("lmccay"));
assertTrue(mapper.mapGroupPrincipal("hdfs")[0].equals("users"));
assertTrue(mapper.mapGroupPrincipal("lmccay")[0].equals("users"));
}
@Test
public void testSimplePrincipalMappingWithWildcardAndExplicitGroups() {
String principalMapping = "";
String groupMapping = "*=users;lmccay=mrgroup";
try {
mapper.loadMappingTable(principalMapping, groupMapping);
}
catch (PrincipalMappingException pme) {
pme.printStackTrace();
fail();
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("lmccay"));
assertTrue(mapper.mapGroupPrincipal("hdfs")[0].equals("users"));
String group = mapper.mapGroupPrincipal("lmccay")[0];
assertTrue(group.equals("users") || group.equals("mrgroup"));
group = mapper.mapGroupPrincipal("lmccay")[1];
assertTrue(group.equals("users") || group.equals("mrgroup"));
}
@Test
public void testSimplePrincipalMappingWithUserAndWildcardAndExplicitGroups() {
String principalMapping = "guest=lmccay";
String groupMapping = "*=users;lmccay=mrgroup";
try {
mapper.loadMappingTable(principalMapping, groupMapping);
}
catch (PrincipalMappingException pme) {
pme.printStackTrace();
fail();
}
assertTrue(mapper.mapUserPrincipal("guest").equals("lmccay"));
assertTrue(mapper.mapGroupPrincipal("hdfs").length == 1);
assertTrue(mapper.mapGroupPrincipal("hdfs")[0].equals("users"));
assertTrue(mapper.mapGroupPrincipal("lmccay").length == 2);
String group = mapper.mapGroupPrincipal("lmccay")[0];
assertTrue(group.equals("users") || group.equals("mrgroup"));
group = mapper.mapGroupPrincipal("lmccay")[1];
assertTrue(group.equals("users") || group.equals("mrgroup"));
}
@Test
public void testNonNullSimplePrincipalMappingWithGroups() {
String principalMapping = "lmccay,kminder=hdfs;newuser=mapred";
String groupMapping = "hdfs=group1;mapred=mrgroup,mrducks";
try {
mapper.loadMappingTable(principalMapping, groupMapping);
}
catch (PrincipalMappingException pme) {
pme.printStackTrace();
fail();
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("hdfs"));
assertTrue(mapper.mapGroupPrincipal("hdfs")[0].equals("group1"));
assertTrue(mapper.mapUserPrincipal("kminder").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("newuser").equals("mapred"));
assertTrue(mapper.mapGroupPrincipal("mapred")[0].equals("mrgroup"));
assertTrue(mapper.mapGroupPrincipal("mapred")[1].equals("mrducks"));
assertTrue(mapper.mapUserPrincipal("hdfs").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("mapred").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("stink").equals("stink"));
}
@Test
public void testNonNullSimplePrincipalMapping() {
String principalMapping = "lmccay,kminder=hdfs;newuser=mapred";
try {
mapper.loadMappingTable(principalMapping, null);
}
catch (PrincipalMappingException pme) {
pme.printStackTrace();
fail();
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("kminder").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("newuser").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("hdfs").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("mapred").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("stink").equals("stink"));
}
@Test
public void testNonNullEndingSemiColonSimplePrincipalMapping() {
String principalMapping = "lmccay,kminder=hdfs;newuser=mapred;";
try {
mapper.loadMappingTable(principalMapping, null);
}
catch (PrincipalMappingException pme) {
fail();
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("kminder").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("newuser").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("hdfs").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("mapred").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("stink").equals("stink"));
}
@Test
public void testNullSimplePrincipalMapping() {
String principalMapping = null;
try {
mapper.loadMappingTable(principalMapping, null);
}
catch (PrincipalMappingException pme) {
fail();
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("lmccay"));
assertTrue(mapper.mapUserPrincipal("kminder").equals("kminder"));
assertTrue(mapper.mapUserPrincipal("newuser").equals("newuser"));
assertTrue(mapper.mapUserPrincipal("hdfs").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("mapred").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("stink").equals("stink"));
}
@Test
public void testInvalidSimplePrincipalMapping() {
String principalMapping = "ksdlhfjkdshf;kjdshf";
try {
mapper.loadMappingTable(principalMapping, null);
}
catch (PrincipalMappingException pme) {
// expected
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("lmccay"));
assertTrue(mapper.mapUserPrincipal("kminder").equals("kminder"));
assertTrue(mapper.mapUserPrincipal("newuser").equals("newuser"));
assertTrue(mapper.mapUserPrincipal("hdfs").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("mapred").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("stink").equals("stink"));
}
@Test
public void testPartiallyInvalidSimplePrincipalMapping() {
String principalMapping = "lmccay=hdfs;kjdshf";
try {
mapper.loadMappingTable(principalMapping, null);
}
catch (PrincipalMappingException pme) {
// expected
}
assertTrue(mapper.mapUserPrincipal("lmccay").equals("lmccay"));
assertTrue(mapper.mapUserPrincipal("kminder").equals("kminder"));
assertTrue(mapper.mapUserPrincipal("newuser").equals("newuser"));
assertTrue(mapper.mapUserPrincipal("hdfs").equals("hdfs"));
assertTrue(mapper.mapUserPrincipal("mapred").equals("mapred"));
assertTrue(mapper.mapUserPrincipal("stink").equals("stink"));
}
}