YARN-8344. Missing nm.stop() in TestNodeManagerResync to fix testKillContainersOnResync. Contributed by Giovanni Matteo Fumarola.

(cherry picked from commit e99e5bf104)
This commit is contained in:
Inigo Goiri 2018-05-23 14:15:26 -07:00
parent c5ff9553fb
commit 8f43ade46a
1 changed files with 47 additions and 38 deletions

View File

@ -150,7 +150,6 @@ public class TestNodeManagerResync {
testContainerPreservationOnResyncImpl(nm, true);
}
@SuppressWarnings("unchecked")
protected void testContainerPreservationOnResyncImpl(TestNodeManager1 nm,
boolean isWorkPreservingRestartEnabled)
throws IOException, YarnException, InterruptedException {
@ -186,12 +185,13 @@ public class TestNodeManagerResync {
}
}
@SuppressWarnings("unchecked")
@SuppressWarnings("resource")
@Test(timeout=10000)
public void testNMshutdownWhenResyncThrowException() throws IOException,
InterruptedException, YarnException {
NodeManager nm = new TestNodeManager3();
YarnConfiguration conf = createNMConfig();
try {
nm.init(conf);
nm.start();
Assert.assertEquals(1, ((TestNodeManager3) nm).getNMRegistrationCount());
@ -199,7 +199,7 @@ public class TestNodeManagerResync {
.handle(new NodeManagerEvent(NodeManagerEventType.RESYNC));
synchronized (isNMShutdownCalled) {
while (isNMShutdownCalled.get() == false) {
while (!isNMShutdownCalled.get()) {
try {
isNMShutdownCalled.wait();
} catch (InterruptedException e) {
@ -208,10 +208,12 @@ public class TestNodeManagerResync {
}
Assert.assertTrue("NM shutdown not called.", isNMShutdownCalled.get());
} finally {
nm.stop();
}
}
@SuppressWarnings("unchecked")
@SuppressWarnings("resource")
@Test(timeout=60000)
public void testContainerResourceIncreaseIsSynchronizedWithRMResync()
throws IOException, InterruptedException, YarnException {
@ -219,6 +221,7 @@ public class TestNodeManagerResync {
YarnConfiguration conf = createNMConfig();
conf.setBoolean(
YarnConfiguration.RM_WORK_PRESERVING_RECOVERY_ENABLED, true);
try {
nm.init(conf);
nm.start();
// Start a container and make sure it is in RUNNING state
@ -227,20 +230,23 @@ public class TestNodeManagerResync {
((TestNodeManager4) nm).updateContainerResource();
// Simulate RM restart by sending a RESYNC event
LOG.info("Sending out RESYNC event");
nm.getNMDispatcher().getEventHandler().handle(
new NodeManagerEvent(NodeManagerEventType.RESYNC));
nm.getNMDispatcher().getEventHandler()
.handle(new NodeManagerEvent(NodeManagerEventType.RESYNC));
try {
syncBarrier.await();
} catch (BrokenBarrierException e) {
e.printStackTrace();
}
Assert.assertFalse(assertionFailedInThread.get());
} finally {
nm.stop();
}
}
// This is to test when NM gets the resync response from last heart beat, it
// should be able to send the already-sent-via-last-heart-beat container
// statuses again when it re-register with RM.
@SuppressWarnings("resource")
@Test
public void testNMSentContainerStatusOnResync() throws Exception {
final ContainerStatus testCompleteContainer =
@ -323,6 +329,7 @@ public class TestNodeManagerResync {
}
};
YarnConfiguration conf = createNMConfig();
try {
nm.init(conf);
nm.start();
@ -331,8 +338,10 @@ public class TestNodeManagerResync {
} catch (BrokenBarrierException e) {
}
Assert.assertFalse(assertionFailedInThread.get());
} finally {
nm.stop();
}
}
// This can be used as a common base class for testing NM resync behavior.
class TestNodeStatusUpdaterResync extends MockNodeStatusUpdater {