|
45 | 45 | import org.apache.hadoop.yarn.api.records.LocalResourceType; |
46 | 46 | import org.apache.hadoop.yarn.api.records.LocalResourceVisibility; |
47 | 47 | import org.apache.hadoop.yarn.api.records.LogAggregationContext; |
| 48 | +import org.apache.hadoop.yarn.api.records.NodeId; |
48 | 49 | import org.apache.hadoop.yarn.api.records.Priority; |
49 | 50 | import org.apache.hadoop.yarn.api.records.ResourceRequest; |
50 | 51 | import org.apache.hadoop.yarn.api.records.URL; |
@@ -569,13 +570,16 @@ protected RMApp testCreateAppSubmittedRecovery( |
569 | 570 | protected RMApp testCreateAppAccepted( |
570 | 571 | ApplicationSubmissionContext submissionContext) throws IOException { |
571 | 572 | RMApp application = testCreateAppSubmittedNoRecovery(submissionContext); |
| 573 | + NodeId nodeId = NodeId.newInstance("host", 1234); |
572 | 574 | // SUBMITTED => ACCEPTED event RMAppEventType.APP_ACCEPTED |
573 | 575 | RMAppEvent event = |
574 | 576 | new RMAppEvent(application.getApplicationId(), |
575 | 577 | RMAppEventType.APP_ACCEPTED); |
576 | 578 | application.handle(event); |
577 | 579 | assertStartTimeSet(application); |
578 | 580 | assertAppState(RMAppState.ACCEPTED, application); |
| 581 | + application.handle( |
| 582 | + new RMAppRunningOnNodeEvent(application.getApplicationId(), nodeId)); |
579 | 583 | return application; |
580 | 584 | } |
581 | 585 |
|
@@ -1089,6 +1093,7 @@ public void testAppFinishedFinished() throws Exception { |
1089 | 1093 | rmDispatcher.await(); |
1090 | 1094 | assertTimesAtFinish(application); |
1091 | 1095 | assertAppState(RMAppState.FINISHED, application); |
| 1096 | + Assert.assertEquals(0, application.getRanNodes().size()); |
1092 | 1097 | StringBuilder diag = application.getDiagnostics(); |
1093 | 1098 | Assert.assertEquals("application diagnostics is not correct", |
1094 | 1099 | "", diag.toString()); |
|
0 commit comments