|
36 | 36 | import javax.ws.rs.core.Response.Status;
|
37 | 37 |
|
38 | 38 | import org.apache.commons.lang3.EnumUtils;
|
| 39 | +import org.apache.hadoop.conf.Configuration; |
39 | 40 | import org.apache.hadoop.security.authorize.AuthorizationException;
|
40 | 41 | import org.apache.hadoop.util.Sets;
|
41 | 42 | import org.apache.hadoop.util.StringUtils;
|
|
57 | 58 | import org.apache.hadoop.yarn.exceptions.ApplicationNotFoundException;
|
58 | 59 | import org.apache.hadoop.yarn.exceptions.YarnException;
|
59 | 60 | import org.apache.hadoop.yarn.server.federation.store.records.SubClusterId;
|
| 61 | +import org.apache.hadoop.yarn.server.resourcemanager.MockNM; |
| 62 | +import org.apache.hadoop.yarn.server.resourcemanager.MockRM; |
| 63 | +import org.apache.hadoop.yarn.server.resourcemanager.MockRMAppSubmitter; |
| 64 | +import org.apache.hadoop.yarn.server.resourcemanager.ResourceManager; |
| 65 | +import org.apache.hadoop.yarn.server.resourcemanager.rmapp.RMApp; |
| 66 | +import org.apache.hadoop.yarn.server.resourcemanager.rmapp.attempt.RMAppAttempt; |
60 | 67 | import org.apache.hadoop.yarn.server.resourcemanager.webapp.NodeIDsInfo;
|
61 | 68 | import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.NodeLabelsInfo;
|
62 | 69 | import org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.LabelsToNodesInfo;
|
@@ -459,9 +466,23 @@ public AppAttemptsInfo getAppAttempts(HttpServletRequest hsr, String appId) {
|
459 | 466 | }
|
460 | 467 |
|
461 | 468 | AppAttemptsInfo infos = new AppAttemptsInfo();
|
462 |
| - org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptInfo |
463 |
| - appAttemptInfo1 = new org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptInfo(); |
464 |
| - infos.add(appAttemptInfo1); |
| 469 | + |
| 470 | + try { |
| 471 | + Configuration conf = new Configuration(); |
| 472 | + MockRM rm = new MockRM(conf); |
| 473 | + rm.start(); |
| 474 | + MockNM amNodeManager = rm.registerNode("127.0.0.1:1234", 2048); |
| 475 | + RMApp app1 = MockRMAppSubmitter.submitWithMemory(1024, rm); |
| 476 | + amNodeManager.nodeHeartbeat(true); |
| 477 | + RMAppAttempt appAttempt = app1.getCurrentAppAttempt(); |
| 478 | + org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptInfo |
| 479 | + appAttemptInfo1 = new org.apache.hadoop.yarn.server.resourcemanager.webapp.dao.AppAttemptInfo(rm, |
| 480 | + appAttempt, Boolean.TRUE, "user", "http://"); |
| 481 | + infos.add(appAttemptInfo1); |
| 482 | + |
| 483 | + } catch (Exception e) { |
| 484 | + throw new RuntimeException("Test getAppAttempts Error.", e); |
| 485 | + } |
465 | 486 |
|
466 | 487 | return infos;
|
467 | 488 | }
|
|
0 commit comments