Java WebAppTests.flushOutput方法代码示例

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testNodesBlockRenderForLostNodes() {
  NodesBlock nodesBlock = injector.getInstance(NodesBlock.class);
  nodesBlock.set("node.state", "lost");
  nodesBlock.render();
  PrintWriter writer = injector.getInstance(PrintWriter.class);
  WebAppTests.flushOutput(injector);

  Mockito.verify(writer,
      Mockito.times(numberOfActualTableHeaders + numberOfThInMetricsTable))
      .print("<th");
  Mockito.verify(
      writer,
      Mockito.times(numberOfRacks * numberOfLostNodesPerRack
          * numberOfActualTableHeaders + numberOfThInMetricsTable)).print(
      "<td");
}
 

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 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 testNodesBlockRender() throws Exception {
  injector.getInstance(NodesBlock.class).render();
  PrintWriter writer = injector.getInstance(PrintWriter.class);
  WebAppTests.flushOutput(injector);

  Mockito.verify(writer,
      Mockito.times(numberOfActualTableHeaders + numberOfThInMetricsTable))
      .print("<th");
  Mockito.verify(
      writer,
      Mockito.times(numberOfRacks * numberOfNodesPerRack
          * numberOfActualTableHeaders + numberOfThInMetricsTable)).print(
      "<td");
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testNodesBlockRenderForNodeLabelFilterWithNonEmptyLabel() {
  NodesBlock nodesBlock = injector.getInstance(NodesBlock.class);
  nodesBlock.set("node.label", "x");
  nodesBlock.render();
  PrintWriter writer = injector.getInstance(PrintWriter.class);
  WebAppTests.flushOutput(injector);

  Mockito.verify(
      writer,
      Mockito.times(numberOfRacks
          * numberOfActualTableHeaders + numberOfThInMetricsTable)).print(
      "<td");
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testNodesBlockRenderForNodeLabelFilterWithEmptyLabel() {
  NodesBlock nodesBlock = injector.getInstance(NodesBlock.class);
  nodesBlock.set("node.label", "");
  nodesBlock.render();
  PrintWriter writer = injector.getInstance(PrintWriter.class);
  WebAppTests.flushOutput(injector);

  Mockito.verify(
      writer,
      Mockito.times(numberOfRacks * (numberOfNodesPerRack - 1)
          * numberOfActualTableHeaders + numberOfThInMetricsTable)).print(
      "<td");
}
 

import org.apache.hadoop.yarn.webapp.test.WebAppTests; //导入方法依赖的package包/类
@Test
public void testNodesBlockRenderForNodeLabelFilterWithAnyLabel() {
  NodesBlock nodesBlock = injector.getInstance(NodesBlock.class);
  nodesBlock.set("node.label", "*");
  nodesBlock.render();
  PrintWriter writer = injector.getInstance(PrintWriter.class);
  WebAppTests.flushOutput(injector);

  Mockito.verify(
      writer,
      Mockito.times(numberOfRacks * numberOfNodesPerRack
          * numberOfActualTableHeaders + numberOfThInMetricsTable)).print(
      "<td");
}
 

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 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 testAPPViewNaturalSortType() throws Exception {
  Injector injector =
      WebAppTests.createMockInjector(ApplicationBaseProtocol.class,
          mockApplicationHistoryClientService(5, 1, 1));
  AHSView ahsViewInstance = injector.getInstance(AHSView.class);

  ahsViewInstance.render();
  WebAppTests.flushOutput(injector);
  Map<String, String> moreParams =
      ahsViewInstance.context().requestContext().moreParams();
  String appTableColumnsMeta = moreParams.get("ui.dataTables.apps.init");
  Assert.assertTrue(appTableColumnsMeta.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);
}