From 90fd863cb8fac7d105ca3874264d4c0ac557abf8 Mon Sep 17 00:00:00 2001 From: Jonathan Turner Eagles Date: Thu, 20 Jun 2013 18:56:12 +0000 Subject: [PATCH] YARN-478. fix coverage org.apache.hadoop.yarn.webapp.log (Aleksey Gorshkov via jeagles) git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/trunk@1495129 13f79535-47bb-0310-9956-ffa450edef68 --- hadoop-yarn-project/CHANGES.txt | 3 + .../TestAggregatedLogsBlock.java | 254 ++++++++++++++++++ .../log/AggregatedLogsBlockForTest.java | 52 ++++ .../hadoop/yarn/webapp/view/BlockForTest.java | 30 +++ .../yarn/webapp/view/HtmlBlockForTest.java | 27 ++ 5 files changed, 366 insertions(+) create mode 100644 hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/logaggregation/TestAggregatedLogsBlock.java create mode 100644 hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/log/AggregatedLogsBlockForTest.java create mode 100644 hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/BlockForTest.java create mode 100644 hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/HtmlBlockForTest.java diff --git a/hadoop-yarn-project/CHANGES.txt b/hadoop-yarn-project/CHANGES.txt index 6095ecfe93..ccb1818cd9 100644 --- a/hadoop-yarn-project/CHANGES.txt +++ b/hadoop-yarn-project/CHANGES.txt @@ -989,6 +989,9 @@ Release 0.23.9 - UNRELEASED YARN-427. Coverage fix for org.apache.hadoop.yarn.server.api.* (Aleksey Gorshkov via jeagles) + + YARN-478. fix coverage org.apache.hadoop.yarn.webapp.log (Aleksey Gorshkov + via jeagles) OPTIMIZATIONS diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/logaggregation/TestAggregatedLogsBlock.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/logaggregation/TestAggregatedLogsBlock.java new file mode 100644 index 0000000000..3c720c4888 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/logaggregation/TestAggregatedLogsBlock.java @@ -0,0 +1,254 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.hadoop.yarn.logaggregation; + +import java.io.ByteArrayOutputStream; +import java.io.File; +import java.io.FileWriter; +import java.io.PrintWriter; +import java.io.Writer; +import java.util.Arrays; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +import javax.servlet.http.HttpServletRequest; + +import org.apache.hadoop.conf.Configuration; +import org.apache.hadoop.fs.FileUtil; +import org.apache.hadoop.fs.Path; +import org.apache.hadoop.security.UserGroupInformation; +import org.apache.hadoop.yarn.api.records.ApplicationAccessType; +import org.apache.hadoop.yarn.api.records.ApplicationAttemptId; +import org.apache.hadoop.yarn.api.records.ApplicationId; +import org.apache.hadoop.yarn.api.records.ContainerId; +import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationAttemptIdPBImpl; +import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationIdPBImpl; +import org.apache.hadoop.yarn.api.records.impl.pb.ContainerIdPBImpl; +import org.apache.hadoop.yarn.conf.YarnConfiguration; +import org.apache.hadoop.yarn.webapp.YarnWebParams; +import org.apache.hadoop.yarn.webapp.log.AggregatedLogsBlockForTest; +import org.apache.hadoop.yarn.webapp.view.BlockForTest; +import org.apache.hadoop.yarn.webapp.view.HtmlBlock; +import org.apache.hadoop.yarn.webapp.view.HtmlBlockForTest; +import org.junit.Test; + +import static org.mockito.Mockito.*; +import static org.junit.Assert.*; + +/** + * Test AggregatedLogsBlock. AggregatedLogsBlock should check user, aggregate a + * logs into one file and show this logs or errors into html code + * + */ +public class TestAggregatedLogsBlock { + /** + * 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")); + + } + + /** + * 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")); + + } + + /** + * 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")); + + } + /** + * 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")); + + } + + + private Configuration getConfiguration() { + Configuration configuration = new Configuration(); + configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true); + configuration.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, "target/logs"); + configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true); + configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin"); + return configuration; + } + + private AggregatedLogsBlockForTest getAggregatedLogsBlockForTest( + Configuration configuration, String user, String containerId) { + HttpServletRequest request = mock(HttpServletRequest.class); + when(request.getRemoteUser()).thenReturn(user); + AggregatedLogsBlockForTest aggregatedBlock = new AggregatedLogsBlockForTest( + configuration); + aggregatedBlock.setRequest(request); + aggregatedBlock.moreParams().put(YarnWebParams.CONTAINER_ID, containerId); + aggregatedBlock.moreParams().put(YarnWebParams.NM_NODENAME, + "localhost:1234"); + aggregatedBlock.moreParams().put(YarnWebParams.APP_OWNER, user); + aggregatedBlock.moreParams().put("start", ""); + aggregatedBlock.moreParams().put("end", ""); + aggregatedBlock.moreParams().put(YarnWebParams.ENTITY_STRING, "entity"); + return aggregatedBlock; + } + + private void writeLog(Configuration configuration, String user) + throws Exception { + ApplicationId appId = ApplicationIdPBImpl.newInstance(0, 1); + ApplicationAttemptId appAttemptId = ApplicationAttemptIdPBImpl.newInstance(appId, 1); + ContainerId containerId = ContainerIdPBImpl.newInstance(appAttemptId, 1); + + String path = "target/logs/" + user + + "/logs/application_0_0001/localhost_1234"; + File f = new File(path); + if (!f.getParentFile().exists()) { + assertTrue(f.getParentFile().mkdirs()); + } + List rootLogDirs = Arrays.asList("target/logs/logs"); + UserGroupInformation ugi = UserGroupInformation.getCurrentUser(); + + AggregatedLogFormat.LogWriter writer = new AggregatedLogFormat.LogWriter( + configuration, new Path(path), ugi); + writer.writeApplicationOwner(ugi.getUserName()); + + Map appAcls = new HashMap(); + appAcls.put(ApplicationAccessType.VIEW_APP, ugi.getUserName()); + writer.writeApplicationACLs(appAcls); + + writer.append(new AggregatedLogFormat.LogKey("container_0_0001_01_000001"), + new AggregatedLogFormat.LogValue(rootLogDirs, containerId,UserGroupInformation.getCurrentUser().getShortUserName())); + writer.closeWriter(); + } + + private void writeLogs(String dirName) throws Exception { + File f = new File(dirName + File.separator + "log1"); + if (!f.getParentFile().exists()) { + assertTrue(f.getParentFile().mkdirs()); + } + + writeLog(dirName + File.separator + "log1", "test log1"); + writeLog(dirName + File.separator + "log2", "test log2"); + writeLog(dirName + File.separator + "log3", "test log3"); + } + + private void writeLog(String fileName, String text) throws Exception { + File f = new File(fileName); + Writer writer = new FileWriter(f); + writer.write(text); + writer.flush(); + writer.close(); + } + +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/log/AggregatedLogsBlockForTest.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/log/AggregatedLogsBlockForTest.java new file mode 100644 index 0000000000..57e6c81659 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/log/AggregatedLogsBlockForTest.java @@ -0,0 +1,52 @@ +/** +* Licensed to the Apache Software Foundation (ASF) under one +* or more contributor license agreements. See the NOTICE file +* distributed with this work for additional information +* regarding copyright ownership. The ASF licenses this file +* to you under the Apache License, Version 2.0 (the +* "License"); you may not use this file except in compliance +* with the License. You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +package org.apache.hadoop.yarn.webapp.log; + +import java.util.HashMap; +import java.util.Map; + +import javax.servlet.http.HttpServletRequest; + +import org.apache.hadoop.conf.Configuration; + +public class AggregatedLogsBlockForTest extends AggregatedLogsBlock { + + final private Map params = new HashMap(); + private HttpServletRequest request; + public AggregatedLogsBlockForTest(Configuration conf) { + super(conf); + } + + @Override + public void render(Block html) { + super.render(html); + } + + public Map moreParams() { + return params; + } + + public HttpServletRequest request() { + return request; + } + public void setRequest(HttpServletRequest request) { + this.request = request; + } + +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/BlockForTest.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/BlockForTest.java new file mode 100644 index 0000000000..36143566f1 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/BlockForTest.java @@ -0,0 +1,30 @@ +/** +* Licensed to the Apache Software Foundation (ASF) under one +* or more contributor license agreements. See the NOTICE file +* distributed with this work for additional information +* regarding copyright ownership. The ASF licenses this file +* to you under the Apache License, Version 2.0 (the +* "License"); you may not use this file except in compliance +* with the License. You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +package org.apache.hadoop.yarn.webapp.view; + +import java.io.PrintWriter; + +public class BlockForTest extends HtmlBlock.Block { + + public BlockForTest(HtmlBlock htmlBlock, PrintWriter out, int level, + boolean wasInline) { + htmlBlock.super(out, level, wasInline); + } + +} diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/HtmlBlockForTest.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/HtmlBlockForTest.java new file mode 100644 index 0000000000..b9497a5282 --- /dev/null +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-common/src/test/java/org/apache/hadoop/yarn/webapp/view/HtmlBlockForTest.java @@ -0,0 +1,27 @@ +/** +* Licensed to the Apache Software Foundation (ASF) under one +* or more contributor license agreements. See the NOTICE file +* distributed with this work for additional information +* regarding copyright ownership. The ASF licenses this file +* to you under the Apache License, Version 2.0 (the +* "License"); you may not use this file except in compliance +* with the License. You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +*/ + +package org.apache.hadoop.yarn.webapp.view; + +public class HtmlBlockForTest extends HtmlBlock { + + @Override + protected void render(Block html) { + info("test!"); + } +}