PrabhuJoseph commented on a change in pull request #3287:
URL: https://github.com/apache/hadoop/pull/3287#discussion_r689765263
##########
File path:
hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/test/java/org/apache/hadoop/yarn/server/resourcemanager/TestResourceTrackerService.java
##########
@@ -460,6 +460,64 @@ public void testGracefulDecommissionWithApp() throws
Exception {
rm.waitForState(id1, NodeState.DECOMMISSIONED);
}
+ /**
+ * Test graceful decommission of node when an AM container is scheduled on a
+ * node just before it is gracefully decommissioned.
+ */
+ @Test
+ public void testGracefulDecommissionAfterAMContainerAlloc() throws Exception
{
+ Configuration conf = new Configuration();
+ conf.set(YarnConfiguration.RM_NODES_EXCLUDE_FILE_PATH, hostFile
+ .getAbsolutePath());
+
+ writeToHostsFile("");
+ rm = new MockRM(conf);
+ rm.start();
+
+ MockNM nm1 = rm.registerNode("host1:1234", 10240);
+ MockNM nm2 = rm.registerNode("host2:5678", 20480);
+ MockNM nm3 = rm.registerNode("host3:4433", 10240);
+
+ NodeId id1 = nm1.getNodeId();
+ NodeId id2 = nm2.getNodeId();
+ NodeId id3 = nm3.getNodeId();
+
+ rm.waitForState(id1, NodeState.RUNNING);
+ rm.waitForState(id2, NodeState.RUNNING);
+ rm.waitForState(id3, NodeState.RUNNING);
+
+ // Create an app and schedule AM on host1.
+ RMApp app = MockRMAppSubmitter.submitWithMemory(2000, rm);
+ MockAM am = MockRM.launchAM(app, rm, nm1);
+
+ // Before sending heartbeat we gracefully decommission the node on which AM
+ // is scheduled to simulate race condition.
+ writeToHostsFile("host1", "host3");
+ rm.getNodesListManager().refreshNodes(conf, true);
+ rm.waitForState(id1, NodeState.DECOMMISSIONING);
+ rm.waitForState(id3, NodeState.DECOMMISSIONING);
+
+ // Heartbeat after the node is in DECOMMISSIONING state. This will be the
+ // first heartbeat containing information about the AM container since the
+ // application was submitted.
+ ApplicationAttemptId aaid = app.getCurrentAppAttempt().getAppAttemptId();
+ nm1.nodeHeartbeat(aaid, 1, ContainerState.RUNNING);
Review comment:
Oh yes, got it. Thanks.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
