Java WebAppTests.createMockInjector方法代码示例

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Before
public void setUp() throws Exception {
  final RMContext mockRMContext =
      TestRMWebApp.mockRMContext(3, numberOfRacks, numberOfNodesPerRack,
        8 * TestRMWebApp.GiB);
  injector =
      WebAppTests.createMockInjector(RMContext.class, mockRMContext,
        new Module() {
          @Override
          public void configure(Binder binder) {
            try {
              binder.bind(ResourceManager.class).toInstance(
                TestRMWebApp.mockRm(mockRMContext));
            } catch (IOException e) {
              throw new IllegalStateException(e);
            }
          }
        });
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testView() throws Exception {
  Injector injector =
      WebAppTests.createMockInjector(ApplicationBaseProtocol.class,
        mockApplicationHistoryClientService(5, 1, 1));
  AHSView ahsViewInstance = injector.getInstance(AHSView.class);

  ahsViewInstance.render();
  WebAppTests.flushOutput(injector);

  ahsViewInstance.set(YarnWebParams.APP_STATE,
    YarnApplicationState.FAILED.toString());
  ahsViewInstance.render();
  WebAppTests.flushOutput(injector);

  ahsViewInstance.set(YarnWebParams.APP_STATE, StringHelper.cjoin(
    YarnApplicationState.FAILED.toString(), YarnApplicationState.KILLED));
  ahsViewInstance.render();
  WebAppTests.flushOutput(injector);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testAppAttemptPage() throws Exception {
  Injector injector =
      WebAppTests.createMockInjector(ApplicationBaseProtocol.class,
        mockApplicationHistoryClientService(1, 1, 5));
  AppAttemptPage appAttemptPageInstance =
      injector.getInstance(AppAttemptPage.class);

  appAttemptPageInstance.render();
  WebAppTests.flushOutput(injector);

  appAttemptPageInstance.set(YarnWebParams.APPLICATION_ATTEMPT_ID,
    ApplicationAttemptId.newInstance(ApplicationId.newInstance(0, 1), 1)
      .toString());
  appAttemptPageInstance.render();
  WebAppTests.flushOutput(injector);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testContainerPage() throws Exception {
  Injector injector =
      WebAppTests.createMockInjector(ApplicationBaseProtocol.class,
        mockApplicationHistoryClientService(1, 1, 1));
  ContainerPage containerPageInstance =
      injector.getInstance(ContainerPage.class);

  containerPageInstance.render();
  WebAppTests.flushOutput(injector);

  containerPageInstance.set(
    YarnWebParams.CONTAINER_ID,
    ContainerId
      .newContainerId(
        ApplicationAttemptId.newInstance(ApplicationId.newInstance(0, 1), 1),
        1).toString());
  containerPageInstance.render();
  WebAppTests.flushOutput(injector);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test public void testSubView() throws Exception {
  Injector injector = WebAppTests.createMockInjector(this);
  injector.getInstance(MainView.class).render();

  PrintWriter out =
      injector.getInstance(HttpServletResponse.class).getWriter();
  out.flush();
  verify(out).print("sub1 text");
  verify(out).print("sub2 text");
  verify(out, times(16)).println(); // test inline transition across views
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testControllerIndex() {
  Injector injector = WebAppTests.createMockInjector(TestRMWebApp.class,
      this, new Module() {

        @Override
        public void configure(Binder binder) {
          binder.bind(ApplicationACLsManager.class).toInstance(
              new ApplicationACLsManager(new Configuration()));
        }
      });
  RmController c = injector.getInstance(RmController.class);
  c.index();
  assertEquals("Applications", c.get(TITLE, "unknown"));
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test public void testView() {
  Injector injector = WebAppTests.createMockInjector(RMContext.class,
      mockRMContext(15, 1, 2, 8*GiB),
      new Module() {
    @Override
    public void configure(Binder binder) {
      try {
        ResourceManager mockRm = mockRm(3, 1, 2, 8*GiB);
        binder.bind(ResourceManager.class).toInstance(mockRm);
        binder.bind(ApplicationBaseProtocol.class)
            .toInstance(mockRm.getClientRMService());
      } catch (IOException e) {
        throw new IllegalStateException(e);
      }
    }
  });
  RmView rmViewInstance = injector.getInstance(RmView.class);
  rmViewInstance.set(YarnWebParams.APP_STATE,
      YarnApplicationState.RUNNING.toString());
  rmViewInstance.render();
  WebAppTests.flushOutput(injector);

  rmViewInstance.set(YarnWebParams.APP_STATE, StringHelper.cjoin(
      YarnApplicationState.ACCEPTED.toString(),
      YarnApplicationState.RUNNING.toString()));
  rmViewInstance.render();
  WebAppTests.flushOutput(injector);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test public void testNodesPage() {
  // 10 nodes. Two of each type.
  final RMContext rmContext = mockRMContext(3, 2, 12, 8*GiB);
  Injector injector = WebAppTests.createMockInjector(RMContext.class,
      rmContext,
      new Module() {
    @Override
    public void configure(Binder binder) {
      try {
        binder.bind(ResourceManager.class).toInstance(mockRm(rmContext));
      } catch (IOException e) {
        throw new IllegalStateException(e);
      }
    }
  });

  // All nodes
  NodesPage instance = injector.getInstance(NodesPage.class);
  instance.render();
  WebAppTests.flushOutput(injector);

  // Unhealthy nodes
  instance.moreParams().put(YarnWebParams.NODE_STATE,
    NodeState.UNHEALTHY.toString());
  instance.render();
  WebAppTests.flushOutput(injector);

  // Lost nodes
  instance.moreParams().put(YarnWebParams.NODE_STATE,
    NodeState.LOST.toString());
  instance.render();
  WebAppTests.flushOutput(injector);

}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testFairSchedulerWebAppPage() {
  List<RMAppState> appStates = Arrays.asList(RMAppState.NEW,
      RMAppState.NEW_SAVING, RMAppState.SUBMITTED);
  final RMContext rmContext = mockRMContext(appStates);
  Injector injector = WebAppTests.createMockInjector(RMContext.class,
      rmContext,
      new Module() {
        @Override
        public void configure(Binder binder) {
          try {
            ResourceManager mockRmWithFairScheduler =
                mockRm(rmContext);
            binder.bind(ResourceManager.class).toInstance
                (mockRmWithFairScheduler);
            binder.bind(ApplicationBaseProtocol.class).toInstance(
              mockRmWithFairScheduler.getClientRMService());
          } catch (IOException e) {
            throw new IllegalStateException(e);
          }
        }
      });
  FairSchedulerPage fsViewInstance = injector.getInstance(FairSchedulerPage
      .class);
  fsViewInstance.render();
  WebAppTests.flushOutput(injector);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testAppControllerIndex() throws Exception {
  ApplicationHistoryManager ahManager = mock(ApplicationHistoryManager.class);
  Injector injector =
      WebAppTests.createMockInjector(ApplicationHistoryManager.class,
        ahManager);
  AHSController controller = injector.getInstance(AHSController.class);
  controller.index();
  Assert
    .assertEquals("Application History", controller.get(TITLE, "unknown"));
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testAppPage() throws Exception {
  Injector injector =
      WebAppTests.createMockInjector(ApplicationBaseProtocol.class,
        mockApplicationHistoryClientService(1, 5, 1));
  AppPage appPageInstance = injector.getInstance(AppPage.class);

  appPageInstance.render();
  WebAppTests.flushOutput(injector);

  appPageInstance.set(YarnWebParams.APPLICATION_ID, ApplicationId
    .newInstance(0, 1).toString());
  appPageInstance.render();
  WebAppTests.flushOutput(injector);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test public void testAppControllerIndex() {
  AppContext ctx = new MockAppContext(0, 1, 1, 1);
  Injector injector = WebAppTests.createMockInjector(AppContext.class, ctx);
  AppController controller = injector.getInstance(AppController.class);
  controller.index();
  assertEquals(ctx.getApplicationID().toString(), controller.get(APP_ID,""));
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test public void testAppControllerIndex() {
  MockAppContext ctx = new MockAppContext(0, 1, 1, 1);
  Injector injector = WebAppTests.createMockInjector(AppContext.class, ctx);
  HsController controller = injector.getInstance(HsController.class);
  controller.index();
  assertEquals(ctx.getApplicationID().toString(), controller.get(APP_ID,""));
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test public void testView() {
  Injector injector = WebAppTests.createMockInjector(RMContext.class,
      mockRMContext(15, 1, 2, 8*GiB),
      new Module() {
    @Override
    public void configure(Binder binder) {
      try {
        ResourceManager mockRm = mockRm(3, 1, 2, 8*GiB);
        binder.bind(ResourceManager.class).toInstance(mockRm);
        binder.bind(ApplicationBaseProtocol.class)
            .toInstance(mockRm.getClientRMService());
      } catch (IOException e) {
        throw new IllegalStateException(e);
      }
    }
  });
  RmView rmViewInstance = injector.getInstance(RmView.class);
  rmViewInstance.set(YarnWebParams.APP_STATE,
      YarnApplicationState.RUNNING.toString());
  rmViewInstance.render();
  WebAppTests.flushOutput(injector);

  rmViewInstance.set(YarnWebParams.APP_STATE, StringHelper.cjoin(
      YarnApplicationState.ACCEPTED.toString(),
      YarnApplicationState.RUNNING.toString()));
  rmViewInstance.render();
  WebAppTests.flushOutput(injector);
  Map<String, String> moreParams =
      rmViewInstance.context().requestContext().moreParams();
  String appsTableColumnsMeta = moreParams.get("ui.dataTables.apps.init");
  Assert.assertTrue(appsTableColumnsMeta.indexOf("natural") != -1);
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testAboutPage() throws Exception {
  Injector injector =
      WebAppTests.createMockInjector(ApplicationBaseProtocol.class,
          mockApplicationHistoryClientService(0, 0, 0));
  AboutPage aboutPageInstance = injector.getInstance(AboutPage.class);

  aboutPageInstance.render();
  WebAppTests.flushOutput(injector);

  aboutPageInstance.render();
  WebAppTests.flushOutput(injector);
}