Examples with BlockForTest org.apache.hadoop.yarn.webapp.view.BlockForTest used on opensource projects

Example 11 with BlockForTest

use of org.apache.hadoop.yarn.webapp.view.BlockForTest in project hadoop by apache.

the class TestBlocks method testHsTasksBlock.

/**
   * 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(""));
}

Example 12 with BlockForTest

use of org.apache.hadoop.yarn.webapp.view.BlockForTest in project hadoop by apache.

the class TestHsJobBlock method createBlockToCreateTo.

private static HtmlBlock.Block createBlockToCreateTo(OutputStream outputStream) {
    PrintWriter printWriter = new PrintWriter(outputStream);
    HtmlBlock html = new HtmlBlockForTest();
    return new BlockForTest(html, printWriter, 10, false) {

        @Override
        protected void subView(Class<? extends SubView> cls) {
        }
    };
}

Example 13 with BlockForTest

use of org.apache.hadoop.yarn.webapp.view.BlockForTest in project hadoop by apache.

the class TestAggregatedLogsBlock method testNoLogs.

/**
   * 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"));
}

Example 14 with BlockForTest

use of org.apache.hadoop.yarn.webapp.view.BlockForTest in project hadoop by apache.

the class TestAggregatedLogsBlock method testBadLogs.

/**
   * 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"));
}