Examples with PlacementManager org.apache.hadoop.yarn.server.resourcemanager.placement.PlacementManager used on opensource projects

Example 1 with PlacementManager

use of org.apache.hadoop.yarn.server.resourcemanager.placement.PlacementManager in project hadoop by apache.

the class TestAppManager method testRMAppSubmitWithQueueChanged.

@Test
public void testRMAppSubmitWithQueueChanged() throws Exception {
    // Setup a PlacementManager returns a new queue
    PlacementManager placementMgr = mock(PlacementManager.class);
    doAnswer(new Answer<Void>() {

        @Override
        public Void answer(InvocationOnMock invocation) throws Throwable {
            ApplicationSubmissionContext ctx = (ApplicationSubmissionContext) invocation.getArguments()[0];
            ctx.setQueue("newQueue");
            return null;
        }
    }).when(placementMgr).placeApplication(any(ApplicationSubmissionContext.class), any(String.class));
    rmContext.setQueuePlacementManager(placementMgr);
    asContext.setQueue("oldQueue");
    appMonitor.submitApplication(asContext, "test");
    RMApp app = rmContext.getRMApps().get(appId);
    Assert.assertNotNull("app is null", app);
    Assert.assertEquals("newQueue", asContext.getQueue());
    // wait for event to be processed
    int timeoutSecs = 0;
    while ((getAppEventType() == RMAppEventType.KILL) && timeoutSecs++ < 20) {
        Thread.sleep(1000);
    }
    Assert.assertEquals("app event type sent is wrong", RMAppEventType.START, getAppEventType());
}