Java BlockForTest类代码示例

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * Bad user. User 'owner' is trying to read logs without access
 */
@Test
public void testAccessDenied() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  writeLogs("target/logs/logs/application_0_0001/container_0_0001_01_000001");

  writeLog(configuration, "owner");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "owner", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out
      .contains("User [owner] is not authorized to view the logs for entity"));

}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * try to read bad logs
 * 
 * @throws Exception
 */
@Test
public void testBadLogs() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  writeLogs("target/logs/logs/application_0_0001/container_0_0001_01_000001");

  writeLog(configuration, "owner");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "admin", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out
      .contains("Logs not available for entity. Aggregation may not be complete, Check back later or try the nodemanager at localhost:1234"));

}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * All ok and the AggregatedLogsBlockFor should aggregate logs and show it.
 * 
 * @throws Exception
 */
@Test
public void testAggregatedLogsBlock() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  writeLogs("target/logs/logs/application_0_0001/container_0_0001_01_000001");

  writeLog(configuration, "admin");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "admin", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out.contains("test log1"));
  assertTrue(out.contains("test log2"));
  assertTrue(out.contains("test log3"));

}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * Log files was deleted.
 * @throws Exception
 */
@Test
public void testNoLogs() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  File f = new File("target/logs/logs/application_0_0001/container_0_0001_01_000001");
  if (!f.exists()) {
    assertTrue(f.mkdirs());
  }
  writeLog(configuration, "admin");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "admin", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out.contains("No logs available for container container_0_0001_01_000001"));

}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * test HsJobsBlock's rendering.
 */
@Test
public void testHsJobsBlock() {
  AppContext ctx = mock(AppContext.class);
  Map<JobId, Job> jobs = new HashMap<JobId, Job>();
  Job job = getJob();
  jobs.put(job.getID(), job);
  when(ctx.getAllJobs()).thenReturn(jobs);

  HsJobsBlock block = new HsJobsBlockForTest(ctx);
  PrintWriter pWriter = new PrintWriter(data);
  Block html = new BlockForTest(new HtmlBlockForTest(), pWriter, 0, false);
  block.render(html);

  pWriter.flush();
  assertTrue(data.toString().contains("JobName"));
  assertTrue(data.toString().contains("UserName"));
  assertTrue(data.toString().contains("QueueName"));
  assertTrue(data.toString().contains("SUCCEEDED"));
}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * Reading from logs should succeed and they should be shown in the
 * AggregatedLogsBlock html.
 * 
 * @throws Exception
 */
@Test
public void testAggregatedLogsBlock() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  writeLogs("target/logs/logs/application_0_0001/container_0_0001_01_000001");

  writeLog(configuration, "admin");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "admin", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out.contains("test log1"));
  assertTrue(out.contains("test log2"));
  assertTrue(out.contains("test log3"));
}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * Reading from logs should succeed and they should be shown in the
 * AggregatedLogsBlock html.
 * 
 * @throws Exception
 */
@Test
public void testAggregatedLogsBlock() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  String user = UserGroupInformation.getCurrentUser().getShortUserName();
  writeLogs("target/logs/logs/" + user + "/application_0_0001/container_0_0001_01_000001");

  writeLog(configuration, "admin");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "admin", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out.contains("test log1"));
  assertTrue(out.contains("test log2"));
  assertTrue(out.contains("test log3"));
}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * Log files was deleted.
 * 
 * @throws Exception
 */
@Test
public void testNoLogs() throws Exception {

  FileUtil.fullyDelete(new File("target/logs"));
  Configuration configuration = getConfiguration();

  File f = new File("target/logs/logs/application_0_0001/container_0_0001_01_000001");
  if (!f.exists()) {
    assertTrue(f.mkdirs());
  }
  writeLog(configuration, "admin");

  AggregatedLogsBlockForTest aggregatedBlock = getAggregatedLogsBlockForTest(
      configuration, "admin", "container_0_0001_01_000001");
  ByteArrayOutputStream data = new ByteArrayOutputStream();
  PrintWriter printWriter = new PrintWriter(data);
  HtmlBlock html = new HtmlBlockForTest();
  HtmlBlock.Block block = new BlockForTest(html, printWriter, 10, false);
  aggregatedBlock.render(block);

  block.getWriter().flush();
  String out = data.toString();
  assertTrue(out.contains("No logs available for container container_0_0001_01_000001"));

}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * Test rendering for ConfBlock
 */
@Test
public void testConfigurationBlock() throws Exception {
  AppContext ctx = mock(AppContext.class);
  Job job = mock(Job.class);
  Path path = new Path("conf");
  Configuration configuration = new Configuration();
  configuration.set("Key for test", "Value for test");
  when(job.getConfFile()).thenReturn(path);
  when(job.loadConfFile()).thenReturn(configuration);

  when(ctx.getJob(any(JobId.class))).thenReturn(job);


  ConfBlockForTest configurationBlock = new ConfBlockForTest(ctx);
  PrintWriter pWriter = new PrintWriter(data);
  Block html = new BlockForTest(new HtmlBlockForTest(), pWriter, 0, false);

  configurationBlock.render(html);
  pWriter.flush();
  assertTrue(data.toString().contains(
          "Sorry, can't do anything without a JobID"));

  configurationBlock.addParameter(AMParams.JOB_ID, "job_01_01");
  data.reset();
  configurationBlock.render(html);
  pWriter.flush();
  assertTrue(data.toString().contains("Key for test"));

  assertTrue(data.toString().contains("Value for test"));

}
 

import org.apache.hadoop.yarn.webapp.view.BlockForTest; //导入依赖的package包/类
/**
 * test HsTasksBlock's rendering.
 */
@Test
public void testHsTasksBlock() {

  Task task = getTask(0);

  Map<TaskId, Task> tasks = new HashMap<TaskId, Task>();
  tasks.put(task.getID(), task);

  AppContext ctx = mock(AppContext.class);
  AppForTest app = new AppForTest(ctx);
  Job job = mock(Job.class);
  when(job.getTasks()).thenReturn(tasks);

  app.setJob(job);

  HsTasksBlockForTest block = new HsTasksBlockForTest(app);

  block.addParameter(AMParams.TASK_TYPE, "r");

  PrintWriter pWriter = new PrintWriter(data);
  Block html = new BlockForTest(new HtmlBlockForTest(), pWriter, 0, false);

  block.render(html);
  pWriter.flush();
  // should be printed information about task
  assertTrue(data.toString().contains("task_0_0001_r_000000"));
  assertTrue(data.toString().contains("SUCCEEDED"));
  assertTrue(data.toString().contains("100001"));
  assertTrue(data.toString().contains("100011"));
  assertTrue(data.toString().contains(""));
}