YARN-6749. TestAppSchedulingInfo.testPriorityAccounting fails consistently. Contributed by Naganarasimha G R

This commit is contained in:
bibinchundatt 2017-06-30 21:52:16 +05:30
parent 2956c584c3
commit f6fcbf426f
1 changed files with 3 additions and 3 deletions

View File

@ -18,8 +18,8 @@
package org.apache.hadoop.yarn.server.resourcemanager.scheduler;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import java.util.ArrayList;
import java.util.List;
@ -106,7 +106,7 @@ public class TestAppSchedulingInfo {
// iterate to verify no ConcurrentModificationException
for (Priority priority: info.getPriorities()) {
info.allocate(NodeType.OFF_SWITCH, null, priority, req1, null);
info.allocate(NodeType.OFF_SWITCH, mock(SchedulerNode.class), priority, req1, null);
}
Assert.assertEquals(1, info.getPriorities().size());
Assert.assertEquals(req2.getPriority(),
@ -117,7 +117,7 @@ public class TestAppSchedulingInfo {
reqs.clear();
reqs.add(req2);
info.updateResourceRequests(reqs, false);
info.allocate(NodeType.OFF_SWITCH, null, req2.getPriority(), req2, null);
info.allocate(NodeType.OFF_SWITCH, mock(SchedulerNode.class), req2.getPriority(), req2, null);
Assert.assertEquals(0, info.getPriorities().size());
req1 = ResourceRequest.newInstance(pri1,