Browse Source

YARN-10350. TestUserGroupMappingPlacementRule fails

Signed-off-by: Akira Ajisaka <aajisaka@apache.org>
Bilwa S T 4 years ago
parent
commit
8980c23fc2

+ 8 - 2
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/fair/SimpleGroupsMapping.java

@@ -50,7 +50,13 @@ public class SimpleGroupsMapping implements GroupMappingServiceProvider {
 
 
   @Override
   @Override
   public Set<String> getGroupsSet(String user) throws IOException {
   public Set<String> getGroupsSet(String user) throws IOException {
-    return ImmutableSet.of(user + "group", user + "subgroup1",
-        user + "subgroup2");
+    if ("admins".equals(user)) {
+      return ImmutableSet.of("root");
+    } else if ("nosecondarygroupuser".equals(user)) {
+      return ImmutableSet.of("primarygrouponly");
+    } else {
+      return ImmutableSet.of(user + "group", user + "subgroup1",
+          user + "subgroup2");
+    }
   }
   }
 }
 }