|
@@ -89,10 +89,8 @@ import org.apache.hadoop.yarn.logaggregation.filecontroller.LogAggregationFileCo
|
|
|
import org.apache.hadoop.yarn.logaggregation.filecontroller.LogAggregationFileControllerContext;
|
|
|
import org.apache.hadoop.yarn.logaggregation.filecontroller.LogAggregationFileControllerFactory;
|
|
|
import org.apache.hadoop.yarn.logaggregation.filecontroller.ifile.LogAggregationIndexedFileController;
|
|
|
-import org.apache.hadoop.yarn.server.webapp.WebServiceClient;
|
|
|
import org.codehaus.jettison.json.JSONArray;
|
|
|
import org.codehaus.jettison.json.JSONObject;
|
|
|
-import org.junit.After;
|
|
|
import org.junit.Assert;
|
|
|
import org.junit.Before;
|
|
|
import org.junit.Test;
|
|
@@ -111,10 +109,8 @@ public class TestLogsCLI {
|
|
|
ByteArrayOutputStream sysErrStream;
|
|
|
private PrintStream sysErr;
|
|
|
|
|
|
- private Configuration conf;
|
|
|
-
|
|
|
@Before
|
|
|
- public void setUp() throws Exception {
|
|
|
+ public void setUp() {
|
|
|
sysOutStream = new ByteArrayOutputStream();
|
|
|
sysOut = new PrintStream(sysOutStream);
|
|
|
System.setOut(sysOut);
|
|
@@ -122,17 +118,11 @@ public class TestLogsCLI {
|
|
|
sysErrStream = new ByteArrayOutputStream();
|
|
|
sysErr = new PrintStream(sysErrStream);
|
|
|
System.setErr(sysErr);
|
|
|
- conf = new YarnConfiguration();
|
|
|
- WebServiceClient.initialize(conf);
|
|
|
- }
|
|
|
-
|
|
|
- @After
|
|
|
- public void tearDown() {
|
|
|
- WebServiceClient.destroy();
|
|
|
}
|
|
|
|
|
|
@Test(timeout = 5000l)
|
|
|
public void testFailResultCodes() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
conf.setClass("fs.file.impl", LocalFileSystem.class, FileSystem.class);
|
|
|
LogCLIHelpers cliHelper = new LogCLIHelpers();
|
|
|
cliHelper.setConf(conf);
|
|
@@ -155,6 +145,7 @@ public class TestLogsCLI {
|
|
|
|
|
|
@Test(timeout = 10000l)
|
|
|
public void testInvalidOpts() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
YarnClient mockYarnClient = createMockYarnClient(
|
|
|
YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
@@ -169,6 +160,7 @@ public class TestLogsCLI {
|
|
|
|
|
|
@Test(timeout = 5000l)
|
|
|
public void testInvalidApplicationId() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
YarnClient mockYarnClient = createMockYarnClient(
|
|
|
YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
@@ -182,6 +174,7 @@ public class TestLogsCLI {
|
|
|
|
|
|
@Test(timeout = 5000L)
|
|
|
public void testInvalidAMContainerId() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
conf.setBoolean(YarnConfiguration.APPLICATION_HISTORY_ENABLED, true);
|
|
|
YarnClient mockYarnClient =
|
|
|
createMockYarnClient(YarnApplicationState.FINISHED,
|
|
@@ -203,6 +196,7 @@ public class TestLogsCLI {
|
|
|
|
|
|
@Test
|
|
|
public void testAMContainerInfoFetchFromTimelineReader() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
conf.setBoolean(YarnConfiguration.TIMELINE_SERVICE_ENABLED, true);
|
|
|
conf.set(YarnConfiguration.TIMELINE_SERVICE_VERSIONS, "2.0f");
|
|
|
YarnClient mockYarnClient =
|
|
@@ -238,6 +232,7 @@ public class TestLogsCLI {
|
|
|
|
|
|
@Test(timeout = 5000l)
|
|
|
public void testUnknownApplicationId() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
YarnClient mockYarnClient = createMockYarnClientUnknownApp();
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
cli.setConf(conf);
|
|
@@ -253,6 +248,7 @@ public class TestLogsCLI {
|
|
|
|
|
|
@Test (timeout = 10000)
|
|
|
public void testHelpMessage() throws Exception {
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
YarnClient mockYarnClient = createMockYarnClient(
|
|
|
YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
@@ -376,12 +372,13 @@ public class TestLogsCLI {
|
|
|
@Test (timeout = 15000)
|
|
|
public void testFetchFinishedApplictionLogs() throws Exception {
|
|
|
String remoteLogRootDir = "target/logs/";
|
|
|
- conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
- conf
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
+ configuration
|
|
|
.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
- conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
- conf.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
+ configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
|
|
|
UserGroupInformation ugi = UserGroupInformation.getCurrentUser();
|
|
|
ApplicationId appId = ApplicationId.newInstance(0, 1);
|
|
@@ -434,15 +431,15 @@ public class TestLogsCLI {
|
|
|
// the first two logs is empty. When we try to read first two logs,
|
|
|
// we will meet EOF exception, but it will not impact other logs.
|
|
|
// Other logs should be read successfully.
|
|
|
- uploadEmptyContainerLogIntoRemoteDir(ugi, conf, rootLogDirs, nodeId,
|
|
|
+ uploadEmptyContainerLogIntoRemoteDir(ugi, configuration, rootLogDirs, nodeId,
|
|
|
containerId0, path, fs);
|
|
|
- uploadEmptyContainerLogIntoRemoteDir(ugi, conf, rootLogDirs, nodeId,
|
|
|
+ uploadEmptyContainerLogIntoRemoteDir(ugi, configuration, rootLogDirs, nodeId,
|
|
|
containerId1, path, fs);
|
|
|
- uploadContainerLogIntoRemoteDir(ugi, conf, rootLogDirs, nodeId,
|
|
|
+ uploadContainerLogIntoRemoteDir(ugi, configuration, rootLogDirs, nodeId,
|
|
|
containerId1, path, fs);
|
|
|
- uploadContainerLogIntoRemoteDir(ugi, conf, rootLogDirs, nodeId,
|
|
|
+ uploadContainerLogIntoRemoteDir(ugi, configuration, rootLogDirs, nodeId,
|
|
|
containerId2, path, fs);
|
|
|
- uploadContainerLogIntoRemoteDir(ugi, conf, rootLogDirs, nodeId,
|
|
|
+ uploadContainerLogIntoRemoteDir(ugi, configuration, rootLogDirs, nodeId,
|
|
|
containerId3, path, fs);
|
|
|
|
|
|
YarnClient mockYarnClient =
|
|
@@ -458,7 +455,7 @@ public class TestLogsCLI {
|
|
|
return mockReport;
|
|
|
}
|
|
|
};
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
|
|
|
int exitCode = cli.run(new String[] { "-applicationId", appId.toString() });
|
|
|
LOG.info(sysOutStream.toString());
|
|
@@ -684,7 +681,7 @@ public class TestLogsCLI {
|
|
|
YarnClient mockYarnClientWithException =
|
|
|
createMockYarnClientWithException();
|
|
|
cli = new LogsCLIForTest(mockYarnClientWithException);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
|
|
|
exitCode =
|
|
|
cli.run(new String[] { "-applicationId", appId.toString(),
|
|
@@ -807,7 +804,8 @@ public class TestLogsCLI {
|
|
|
any(ContainerId.class));
|
|
|
|
|
|
// create local logs
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
String rootLogDir = "target/LocalLogs";
|
|
|
Path rootLogDirPath = new Path(rootLogDir);
|
|
|
if (fs.exists(rootLogDirPath)) {
|
|
@@ -966,12 +964,13 @@ public class TestLogsCLI {
|
|
|
UserGroupInformation testUgi = UserGroupInformation
|
|
|
.createRemoteUser(testUser);
|
|
|
|
|
|
- conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
- conf
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
+ configuration
|
|
|
.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
- conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
- conf.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
+ configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
|
|
|
ApplicationId appId = ApplicationId.newInstance(0, 1);
|
|
|
ApplicationAttemptId appAttemptId =
|
|
@@ -1011,13 +1010,13 @@ public class TestLogsCLI {
|
|
|
assertTrue(fs.mkdirs(path));
|
|
|
|
|
|
// upload container logs for app into remote dir
|
|
|
- uploadContainerLogIntoRemoteDir(testUgi, conf, rootLogDirs,
|
|
|
+ uploadContainerLogIntoRemoteDir(testUgi, configuration, rootLogDirs,
|
|
|
nodeId, containerId, path, fs);
|
|
|
|
|
|
YarnClient mockYarnClient = createMockYarnClient(
|
|
|
YarnApplicationState.FINISHED, testUgi.getShortUserName());
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
|
|
|
// Verify that we can get the application logs by specifying
|
|
|
// a correct appOwner
|
|
@@ -1044,7 +1043,7 @@ public class TestLogsCLI {
|
|
|
// and can get app logs successfully.
|
|
|
YarnClient mockYarnClient2 = createMockYarnClientUnknownApp();
|
|
|
cli = new LogsCLIForTest(mockYarnClient2);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
exitCode = cli.run(new String[] {
|
|
|
"-applicationId", appId.toString()});
|
|
|
assertTrue(exitCode == 0);
|
|
@@ -1058,7 +1057,7 @@ public class TestLogsCLI {
|
|
|
fs.delete(path, true);
|
|
|
}
|
|
|
assertTrue(fs.mkdirs(path));
|
|
|
- uploadContainerLogIntoRemoteDir(testUgi, conf, rootLogDirs,
|
|
|
+ uploadContainerLogIntoRemoteDir(testUgi, configuration, rootLogDirs,
|
|
|
nodeId, containerId, path, fs);
|
|
|
|
|
|
exitCode = cli.run(new String[] {
|
|
@@ -1113,13 +1112,14 @@ public class TestLogsCLI {
|
|
|
public void testLogsCLIWithInvalidArgs() throws Exception {
|
|
|
String localDir = "target/SaveLogs";
|
|
|
Path localPath = new Path(localDir);
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
ApplicationId appId = ApplicationId.newInstance(0, 1);
|
|
|
YarnClient mockYarnClient =
|
|
|
createMockYarnClient(YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
|
|
|
// Specify an invalid applicationId
|
|
|
int exitCode = cli.run(new String[] {"-applicationId",
|
|
@@ -1180,6 +1180,7 @@ public class TestLogsCLI {
|
|
|
String remoteLogRootDir = "target/logs/";
|
|
|
String jobUser = "user1";
|
|
|
String loggedUser = "user2";
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
conf.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
@@ -1225,6 +1226,7 @@ public class TestLogsCLI {
|
|
|
String remoteLogRootDir1 = "target/logs1/";
|
|
|
String jobUser = "user1";
|
|
|
String loggedUser = "user2";
|
|
|
+ Configuration conf = new YarnConfiguration();
|
|
|
conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
conf.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
String controllerName = "indexed";
|
|
@@ -1259,13 +1261,14 @@ public class TestLogsCLI {
|
|
|
String localDir = "target/SaveLogs";
|
|
|
Path localPath = new Path(localDir);
|
|
|
|
|
|
- conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
- conf
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
+ configuration
|
|
|
.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
- conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
- conf.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
+ configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
+ configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
ApplicationId appId = ApplicationId.newInstance(0, 1);
|
|
|
ApplicationAttemptId appAttemptId =
|
|
|
ApplicationAttemptId.newInstance(appId, 1);
|
|
@@ -1284,14 +1287,14 @@ public class TestLogsCLI {
|
|
|
nodeIds.add(nodeId2);
|
|
|
|
|
|
try {
|
|
|
- createContainerLogs(conf, remoteLogRootDir, rootLogDir, fs,
|
|
|
+ createContainerLogs(configuration, remoteLogRootDir, rootLogDir, fs,
|
|
|
appId, containerIds, nodeIds);
|
|
|
|
|
|
YarnClient mockYarnClient =
|
|
|
createMockYarnClient(YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
int exitCode = cli.run(new String[] {"-applicationId",
|
|
|
appId.toString(),
|
|
|
"-out" , localPath.toString()});
|
|
@@ -1349,12 +1352,13 @@ public class TestLogsCLI {
|
|
|
@Test (timeout = 15000)
|
|
|
public void testPrintContainerLogMetadata() throws Exception {
|
|
|
String remoteLogRootDir = "target/logs/";
|
|
|
- conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
- conf
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
+ configuration
|
|
|
.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
- conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
- conf.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
+ configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
String rootLogDir = "target/LocalLogs";
|
|
|
|
|
|
ApplicationId appId = ApplicationId.newInstance(0, 1);
|
|
@@ -1373,14 +1377,14 @@ public class TestLogsCLI {
|
|
|
nodeIds.add(nodeId);
|
|
|
nodeIds.add(nodeId);
|
|
|
|
|
|
- createContainerLogs(conf, remoteLogRootDir, rootLogDir, fs,
|
|
|
+ createContainerLogs(configuration, remoteLogRootDir, rootLogDir, fs,
|
|
|
appId, containerIds, nodeIds);
|
|
|
|
|
|
YarnClient mockYarnClient =
|
|
|
createMockYarnClient(YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
|
|
|
cli.run(new String[] {"-applicationId", appId.toString(),
|
|
|
"-show_container_log_info"});
|
|
@@ -1453,11 +1457,12 @@ public class TestLogsCLI {
|
|
|
@Test (timeout = 15000)
|
|
|
public void testListNodeInfo() throws Exception {
|
|
|
String remoteLogRootDir = "target/logs/";
|
|
|
- conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
- conf
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
+ configuration
|
|
|
.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
- conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
- conf.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
+ configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
+ configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
|
|
|
ApplicationId appId = ApplicationId.newInstance(0, 1);
|
|
|
ApplicationAttemptId appAttemptId =
|
|
@@ -1477,16 +1482,16 @@ public class TestLogsCLI {
|
|
|
nodeIds.add(nodeId2);
|
|
|
|
|
|
String rootLogDir = "target/LocalLogs";
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
|
|
|
- createContainerLogs(conf, remoteLogRootDir, rootLogDir, fs,
|
|
|
+ createContainerLogs(configuration, remoteLogRootDir, rootLogDir, fs,
|
|
|
appId, containerIds, nodeIds);
|
|
|
|
|
|
YarnClient mockYarnClient =
|
|
|
createMockYarnClient(YarnApplicationState.FINISHED,
|
|
|
UserGroupInformation.getCurrentUser().getShortUserName());
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
|
|
|
cli.run(new String[] { "-applicationId", appId.toString(),
|
|
|
"-list_nodes" });
|
|
@@ -1503,12 +1508,13 @@ public class TestLogsCLI {
|
|
|
@Test (timeout = 15000)
|
|
|
public void testFetchApplictionLogsHar() throws Exception {
|
|
|
String remoteLogRootDir = "target/logs/";
|
|
|
- conf.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
- conf
|
|
|
+ Configuration configuration = new YarnConfiguration();
|
|
|
+ configuration.setBoolean(YarnConfiguration.LOG_AGGREGATION_ENABLED, true);
|
|
|
+ configuration
|
|
|
.set(YarnConfiguration.NM_REMOTE_APP_LOG_DIR, remoteLogRootDir);
|
|
|
- conf.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
- conf.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
- FileSystem fs = FileSystem.get(conf);
|
|
|
+ configuration.setBoolean(YarnConfiguration.YARN_ACL_ENABLE, true);
|
|
|
+ configuration.set(YarnConfiguration.YARN_ADMIN_ACL, "admin");
|
|
|
+ FileSystem fs = FileSystem.get(configuration);
|
|
|
UserGroupInformation ugi = UserGroupInformation.getCurrentUser();
|
|
|
URL harUrl = ClassLoader.getSystemClassLoader()
|
|
|
.getResource("application_1440536969523_0001.har");
|
|
@@ -1528,7 +1534,7 @@ public class TestLogsCLI {
|
|
|
createMockYarnClient(YarnApplicationState.FINISHED,
|
|
|
ugi.getShortUserName());
|
|
|
LogsCLI cli = new LogsCLIForTest(mockYarnClient);
|
|
|
- cli.setConf(conf);
|
|
|
+ cli.setConf(configuration);
|
|
|
int exitCode = cli.run(new String[]{"-applicationId",
|
|
|
"application_1440536969523_0001"});
|
|
|
assertTrue(exitCode == 0);
|