From f7dad171e1bedda5800cbcf996bfa94ad568b367 Mon Sep 17 00:00:00 2001 From: Prabhu Joseph Date: Fri, 14 Jun 2019 19:02:43 +0530 Subject: [PATCH] YARN-9621. FIX TestDSWithMultipleNodeManager.testDistributedShellWithPlacementConstraint. --- .../distributedshell/TestDSWithMultipleNodeManager.java | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/test/java/org/apache/hadoop/yarn/applications/distributedshell/TestDSWithMultipleNodeManager.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/test/java/org/apache/hadoop/yarn/applications/distributedshell/TestDSWithMultipleNodeManager.java index 698682a..917f860 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/test/java/org/apache/hadoop/yarn/applications/distributedshell/TestDSWithMultipleNodeManager.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-applications/hadoop-yarn-applications-distributedshell/src/test/java/org/apache/hadoop/yarn/applications/distributedshell/TestDSWithMultipleNodeManager.java @@ -31,6 +31,7 @@ import org.apache.hadoop.yarn.server.resourcemanager.RMContext; import org.apache.hadoop.yarn.server.resourcemanager.nodelabels.RMNodeLabelsManager; import static org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacitySchedulerConfiguration.PREFIX; +import org.junit.After; import org.junit.Assert; import org.junit.Before; import org.junit.Test; @@ -52,6 +53,11 @@ public void setup() throws Exception { distShellTest.setupInternal(NUM_NMS); } + @After + public void tearDown() throws Exception { + distShellTest.tearDown(); + } + private void initializeNodeLabels() throws IOException { RMContext rmContext = distShellTest.yarnCluster.getResourceManager(0).getRMContext(); @@ -145,6 +151,14 @@ public void testDistributedShellWithPlacementConstraint() "1", "--shell_command", distShellTest.getSleepCommand(15), + "--master_memory", + "512", + "--master_vcores", + "2", + "--container_memory", + "128", + "--container_vcores", + "1", "--placement_spec", "zk=1,NOTIN,NODE,zk:spark=1,NOTIN,NODE,zk" }; -- 2.7.4 (Apple Git-66)