Examples with ContainersLauncherEvent org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainersLauncherEvent used on opensource projects

Example 6 with ContainersLauncherEvent

use of org.apache.hadoop.yarn.server.nodemanager.containermanager.launcher.ContainersLauncherEvent in project hadoop by apache.

the class TestContainer method testServiceData.

/**
   * Verify serviceData correctly sent.
   */
@Test
public void testServiceData() throws Exception {
    WrappedContainer wc = null;
    try {
        wc = new WrappedContainer(9, 314159265358979L, 4344, "yak", false, true);
        assertEquals(ContainerState.NEW, wc.c.getContainerState());
        wc.initContainer();
        for (final Map.Entry<String, ByteBuffer> e : wc.serviceData.entrySet()) {
            ArgumentMatcher<AuxServicesEvent> matchesServiceReq = new ArgumentMatcher<AuxServicesEvent>() {

                @Override
                public boolean matches(Object o) {
                    AuxServicesEvent evt = (AuxServicesEvent) o;
                    return e.getKey().equals(evt.getServiceID()) && 0 == e.getValue().compareTo(evt.getServiceData());
                }
            };
            verify(wc.auxBus).handle(argThat(matchesServiceReq));
        }
        final WrappedContainer wcf = wc;
        // verify launch on empty resource request
        ArgumentMatcher<ContainersLauncherEvent> matchesLaunchReq = new ArgumentMatcher<ContainersLauncherEvent>() {

            @Override
            public boolean matches(Object o) {
                ContainersLauncherEvent evt = (ContainersLauncherEvent) o;
                return evt.getType() == ContainersLauncherEventType.LAUNCH_CONTAINER && wcf.cId.equals(evt.getContainer().getContainerId());
            }
        };
        verify(wc.launcherBus).handle(argThat(matchesLaunchReq));
    } finally {
        if (wc != null) {
            wc.finished();
        }
    }
}