|
@@ -21,6 +21,7 @@ package org.apache.hadoop.yarn.server.resourcemanager;
|
|
import static org.mockito.Matchers.isA;
|
|
import static org.mockito.Matchers.isA;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.spy;
|
|
import static org.mockito.Mockito.spy;
|
|
|
|
+import static org.mockito.Mockito.timeout;
|
|
import static org.mockito.Mockito.times;
|
|
import static org.mockito.Mockito.times;
|
|
import static org.mockito.Mockito.verify;
|
|
import static org.mockito.Mockito.verify;
|
|
import static org.mockito.Mockito.when;
|
|
import static org.mockito.Mockito.when;
|
|
@@ -976,9 +977,10 @@ public class TestRMRestart extends ParameterizedSchedulerTestBase {
|
|
List<ApplicationReport> appList2 = response2.getApplicationList();
|
|
List<ApplicationReport> appList2 = response2.getApplicationList();
|
|
Assert.assertTrue(3 == appList2.size());
|
|
Assert.assertTrue(3 == appList2.size());
|
|
|
|
|
|
- // check application summary is logged for the completed apps after RM restart.
|
|
|
|
- verify(rm2.getRMAppManager(), times(3)).logApplicationSummary(
|
|
|
|
- isA(ApplicationId.class));
|
|
|
|
|
|
+ // check application summary is logged for the completed apps with timeout
|
|
|
|
+ // to make sure APP_COMPLETED events are processed, after RM restart.
|
|
|
|
+ verify(rm2.getRMAppManager(), timeout(1000).times(3)).
|
|
|
|
+ logApplicationSummary(isA(ApplicationId.class));
|
|
}
|
|
}
|
|
|
|
|
|
private MockAM launchAM(RMApp app, MockRM rm, MockNM nm)
|
|
private MockAM launchAM(RMApp app, MockRM rm, MockNM nm)
|