|
67 | 67 |
|
68 | 68 | import static org.apache.hadoop.yarn.conf.YarnConfiguration.MEMORY_CONFIGURATION_STORE;
|
69 | 69 | import static org.apache.hadoop.yarn.conf.YarnConfiguration.SCHEDULER_CONFIGURATION_STORE_CLASS;
|
| 70 | +import static org.apache.hadoop.yarn.server.resourcemanager.scheduler.capacity.CapacitySchedulerTestUtilities.GB; |
70 | 71 | import static org.junit.Assert.assertEquals;
|
71 | 72 |
|
72 | 73 | public final class TestWebServiceUtil {
|
@@ -96,14 +97,14 @@ public static void runTest(String template, String name,
|
96 | 97 | final boolean reinitAfterNodeChane = isMutableConfig(rm.getConfig());
|
97 | 98 | try {
|
98 | 99 | assertJsonResponse(sendRequest(resource), String.format(template, name, 0));
|
99 |
| - MockNM nm1 = rm.registerNode("h1:1234", 8192, 8); |
100 |
| - rm.registerNode("h2:1234", 8192, 8); |
| 100 | + MockNM nm1 = rm.registerNode("h1:1234", 8 * GB, 8); |
| 101 | + rm.registerNode("h2:1234", 8 * GB, 8); |
101 | 102 | if (reinitAfterNodeChane) {
|
102 | 103 | reinitialize(rm, rm.getConfig());
|
103 | 104 | }
|
104 | 105 | assertJsonResponse(sendRequest(resource), String.format(template, name, 16));
|
105 |
| - rm.registerNode("h3:1234", 8192, 8); |
106 |
| - MockNM nm4 = rm.registerNode("h4:1234", 8192, 8); |
| 106 | + rm.registerNode("h3:1234", 8 * GB, 8); |
| 107 | + MockNM nm4 = rm.registerNode("h4:1234", 8 * GB, 8); |
107 | 108 | if (reinitAfterNodeChane) {
|
108 | 109 | reinitialize(rm, rm.getConfig());
|
109 | 110 | }
|
|
0 commit comments