Java ExitCodeException.getMessage方法代码示例

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
/**
 * Creates a hardlink 
 * @param file - existing source file
 * @param linkName - desired target link file
 */
public static void createHardLink(File file, File linkName) 
throws IOException {
  if (file == null) {
    throw new IOException(
        "invalid arguments to createHardLink: source file is null");
  }
  if (linkName == null) {
    throw new IOException(
        "invalid arguments to createHardLink: link name is null");
  }
 // construct and execute shell command
  String[] hardLinkCommand = getHardLinkCommand.linkOne(file, linkName);
  ShellCommandExecutor shexec = new ShellCommandExecutor(hardLinkCommand);
  try {
    shexec.execute();
  } catch (ExitCodeException e) {
    throw new IOException("Failed to execute command " +
        Arrays.toString(hardLinkCommand) +
        "; command output: \"" + shexec.getOutput() + "\"" +
        "; WrappedException: \"" + e.getMessage() + "\"");
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidEnvSyntaxDiagnostics() throws IOException  {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile);
    FileUtil.setExecutable(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    // invalid env
    env.put(
        "APPLICATION_WORKFLOW_CONTEXT", "{\"workflowId\":\"609f91c5cd83\"," +
        "\"workflowName\":\"\n\ninsert table " +
        "\npartition (cd_education_status)\nselect cd_demo_sk, cd_gender, " );
    List<String> commands = new ArrayList<String>();
    new DefaultContainerExecutor().writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    // It is supposed that LANG is set as C.
    Map<String, String> cmdEnv = new HashMap<String, String>();
    cmdEnv.put("LANG", "C");
    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()},
      tmpDir, cmdEnv);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertTrue(diagnostics.contains(Shell.WINDOWS ?
        "is not recognized as an internal or external command" :
        "command not found"));
    Assert.assertTrue(shexc.getExitCode() != 0);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testContainerLaunchStdoutAndStderrDiagnostics() throws IOException {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    // echo "hello" to stdout and "error" to stderr and exit code with 2;
    String command = Shell.WINDOWS ?
        "@echo \"hello\" & @echo \"error\" 1>&2 & exit /b 2" :
        "echo \"hello\"; echo \"error\" 1>&2; exit 2;";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(command);
    writer.close();
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    commands.add(command);
    ContainerExecutor exec = new DefaultContainerExecutor();
    exec.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    // test stderr
    Assert.assertTrue(diagnostics.contains("error"));
    // test stdout
    Assert.assertTrue(shexc.getOutput().contains("hello"));
    Assert.assertTrue(shexc.getExitCode() == 2);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidSymlinkDiagnostics() throws IOException  {

  File shellFile = null;
  File tempFile = null;
  String symLink = Shell.WINDOWS ? "test.cmd" :
    "test";
  File symLinkFile = null;

  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    tempFile = Shell.appendScriptExtension(tmpDir, "temp");
    String timeoutCommand = Shell.WINDOWS ? "@echo \"hello\"" :
      "echo \"hello\"";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(timeoutCommand);
    writer.close();

    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    //This is an invalid path and should throw exception because of No such file.
    Path invalidPath = new Path(shellFile.getAbsolutePath()+"randomPath");
    resources.put(invalidPath, Arrays.asList(symLink));
    FileOutputStream fos = new FileOutputStream(tempFile);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    if (Shell.WINDOWS) {
      commands.add("cmd");
      commands.add("/c");
      commands.add("\"" + symLink + "\"");
    } else {
      commands.add("/bin/sh ./\\\"" + symLink + "\\\"");
    }
    new DefaultContainerExecutor()
        .writeLaunchEnv(fos, env, resources, commands,
            new Path(localLogDir.getAbsolutePath()));
    fos.flush();
    fos.close();
    FileUtil.setExecutable(tempFile, true);

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{tempFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertNotNull(diagnostics);
    Assert.assertTrue(shexc.getExitCode() != 0);
    symLinkFile = new File(tmpDir, symLink);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
    if (tempFile != null
        && tempFile.exists()) {
      tempFile.delete();
    }
    if (symLinkFile != null
        && symLinkFile.exists()) {
      symLinkFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidEnvSyntaxDiagnostics() throws IOException  {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile);
    FileUtil.setExecutable(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    // invalid env
    env.put(
        "APPLICATION_WORKFLOW_CONTEXT", "{\"workflowId\":\"609f91c5cd83\"," +
        "\"workflowName\":\"\n\ninsert table " +
        "\npartition (cd_education_status)\nselect cd_demo_sk, cd_gender, " );
    List<String> commands = new ArrayList<String>();
    new DefaultContainerExecutor()
        .writeLaunchEnv(fos, env, resources, commands,
            new Path(localLogDir.getAbsolutePath()));
    fos.flush();
    fos.close();

    // It is supposed that LANG is set as C.
    Map<String, String> cmdEnv = new HashMap<String, String>();
    cmdEnv.put("LANG", "C");
    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()},
      tmpDir, cmdEnv);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertTrue(diagnostics.contains(Shell.WINDOWS ?
        "is not recognized as an internal or external command" :
        "command not found"));
    Assert.assertTrue(shexc.getExitCode() != 0);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testContainerLaunchStdoutAndStderrDiagnostics() throws IOException {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    // echo "hello" to stdout and "error" to stderr and exit code with 2;
    String command = Shell.WINDOWS ?
        "@echo \"hello\" & @echo \"error\" 1>&2 & exit /b 2" :
        "echo \"hello\"; echo \"error\" 1>&2; exit 2;";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(command);
    writer.close();
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    commands.add(command);
    ContainerExecutor exec = new DefaultContainerExecutor();
    exec.writeLaunchEnv(fos, env, resources, commands,
        new Path(localLogDir.getAbsolutePath()));
    fos.flush();
    fos.close();

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    // test stderr
    Assert.assertTrue(diagnostics.contains("error"));
    // test stdout
    Assert.assertTrue(shexc.getOutput().contains("hello"));
    Assert.assertTrue(shexc.getExitCode() == 2);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidSymlinkDiagnostics() throws IOException  {

  File shellFile = null;
  File tempFile = null;
  String symLink = Shell.WINDOWS ? "test.cmd" :
    "test";
  File symLinkFile = null;

  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    tempFile = Shell.appendScriptExtension(tmpDir, "temp");
    String timeoutCommand = Shell.WINDOWS ? "@echo \"hello\"" :
      "echo \"hello\"";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(timeoutCommand);
    writer.close();

    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    //This is an invalid path and should throw exception because of No such file.
    Path invalidPath = new Path(shellFile.getAbsolutePath()+"randomPath");
    resources.put(invalidPath, Arrays.asList(symLink));
    FileOutputStream fos = new FileOutputStream(tempFile);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    if (Shell.WINDOWS) {
      commands.add("cmd");
      commands.add("/c");
      commands.add("\"" + symLink + "\"");
    } else {
      commands.add("/bin/sh ./\\\"" + symLink + "\\\"");
    }
    new DefaultContainerExecutor().writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();
    FileUtil.setExecutable(tempFile, true);

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{tempFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertNotNull(diagnostics);
    Assert.assertTrue(shexc.getExitCode() != 0);
    symLinkFile = new File(tmpDir, symLink);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
    if (tempFile != null
        && tempFile.exists()) {
      tempFile.delete();
    }
    if (symLinkFile != null
        && symLinkFile.exists()) {
      symLinkFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
protected static int createHardLinkMult(File parentDir, 
    String[] fileBaseNames, File linkDir, int maxLength) 
throws IOException {
  if (parentDir == null) {
    throw new IOException(
        "invalid arguments to createHardLinkMult: parent directory is null");
  }
  if (linkDir == null) {
    throw new IOException(
        "invalid arguments to createHardLinkMult: link directory is null");
  }
  if (fileBaseNames == null) {
    throw new IOException(
        "invalid arguments to createHardLinkMult: "
        + "filename list can be empty but not null");
  }
  if (fileBaseNames.length == 0) {
    //the OS cmds can't handle empty list of filenames, 
    //but it's legal, so just return.
    return 0; 
  }
  if (!linkDir.exists()) {
    throw new FileNotFoundException(linkDir + " not found.");
  }

  //if the list is too long, split into multiple invocations
  int callCount = 0;
  if (getLinkMultArgLength(parentDir, fileBaseNames, linkDir) > maxLength
        && fileBaseNames.length > 1) {
    String[] list1 = Arrays.copyOf(fileBaseNames, fileBaseNames.length/2);
    callCount += createHardLinkMult(parentDir, list1, linkDir, maxLength);
    String[] list2 = Arrays.copyOfRange(fileBaseNames, fileBaseNames.length/2,
        fileBaseNames.length);
    callCount += createHardLinkMult(parentDir, list2, linkDir, maxLength);  
    return callCount;
  } else {
    callCount = 1;
  }
  
  // construct and execute shell command
  String[] hardLinkCommand = getHardLinkCommand.linkMult(fileBaseNames, 
      linkDir);
  ShellCommandExecutor shexec = new ShellCommandExecutor(hardLinkCommand,
    parentDir, null, 0L);
  try {
    shexec.execute();
  } catch (ExitCodeException e) {
    throw new IOException(shexec.getOutput() + e.getMessage());
  }
  return callCount;
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidSymlinkDiagnostics() throws IOException  {

  File shellFile = null;
  File tempFile = null;
  String symLink = Shell.WINDOWS ? "test.cmd" :
    "test";
  File symLinkFile = null;

  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    tempFile = Shell.appendScriptExtension(tmpDir, "temp");
    String timeoutCommand = Shell.WINDOWS ? "@echo \"hello\"" :
      "echo \"hello\"";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(timeoutCommand);
    writer.close();

    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    //This is an invalid path and should throw exception because of No such file.
    Path invalidPath = new Path(shellFile.getAbsolutePath()+"randomPath");
    resources.put(invalidPath, Arrays.asList(symLink));
    FileOutputStream fos = new FileOutputStream(tempFile);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    if (Shell.WINDOWS) {
      commands.add("cmd");
      commands.add("/c");
      commands.add("\"" + symLink + "\"");
    } else {
      commands.add("/bin/sh ./\\\"" + symLink + "\\\"");
    }
    ContainerLaunch.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();
    FileUtil.setExecutable(tempFile, true);

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{tempFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertNotNull(diagnostics);
    Assert.assertTrue(shexc.getExitCode() != 0);
    symLinkFile = new File(tmpDir, symLink);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
    if (tempFile != null
        && tempFile.exists()) {
      tempFile.delete();
    }
    if (symLinkFile != null
        && symLinkFile.exists()) {
      symLinkFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidEnvSyntaxDiagnostics() throws IOException  {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    String timeoutCommand = Shell.WINDOWS ? "@echo \"hello\"" :
      "echo \"hello\"";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(timeoutCommand);
    writer.close();
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile);

    Map<String, String> env = new HashMap<String, String>();
    // invalid env
    env.put(
        "APPLICATION_WORKFLOW_CONTEXT", "{\"workflowId\":\"609f91c5cd83\"," +
        "\"workflowName\":\"\n\ninsert table " +
        "\npartition (cd_education_status)\nselect cd_demo_sk, cd_gender, " );
    List<String> commands = new ArrayList<String>();
    ContainerLaunch.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertTrue(diagnostics.contains("command not found"));
    Assert.assertTrue(shexc.getExitCode() != 0);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testContainerLaunchStdoutAndStderrDiagnostics() throws IOException {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    // echo "hello" to stdout and "error" to stderr and exit code with 2;
    String command = Shell.WINDOWS ? "@echo \"hello\"; @echo \"error\" 1>&2; exit 2;" :
      "echo \"hello\"; echo \"error\" 1>&2; exit 2;";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(command);
    writer.close();
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    commands.add(command);
    ContainerLaunch.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    // test stderr
    Assert.assertTrue(diagnostics.contains("error"));
    // test stdout
    Assert.assertTrue(shexc.getOutput().contains("hello"));
    Assert.assertTrue(shexc.getExitCode() == 2);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidEnvSyntaxDiagnostics() throws IOException  {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile);
    FileUtil.setExecutable(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    // invalid env
    env.put(
        "APPLICATION_WORKFLOW_CONTEXT", "{\"workflowId\":\"609f91c5cd83\"," +
        "\"workflowName\":\"\n\ninsert table " +
        "\npartition (cd_education_status)\nselect cd_demo_sk, cd_gender, " );
    List<String> commands = new ArrayList<String>();
    ContainerLaunch.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertTrue(diagnostics.contains(Shell.WINDOWS ?
        "is not recognized as an internal or external command" :
        "command not found"));
    Assert.assertTrue(shexc.getExitCode() != 0);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testContainerLaunchStdoutAndStderrDiagnostics() throws IOException {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    // echo "hello" to stdout and "error" to stderr and exit code with 2;
    String command = Shell.WINDOWS ?
        "@echo \"hello\" & @echo \"error\" 1>&2 & exit /b 2" :
        "echo \"hello\"; echo \"error\" 1>&2; exit 2;";
    PrintWriter writer = new PrintWriter(new FileOutputStream(shellFile));
    FileUtil.setExecutable(shellFile, true);
    writer.println(command);
    writer.close();
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    List<String> commands = new ArrayList<String>();
    commands.add(command);
    ContainerLaunch.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()}, tmpDir);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    // test stderr
    Assert.assertTrue(diagnostics.contains("error"));
    // test stdout
    Assert.assertTrue(shexc.getOutput().contains("hello"));
    Assert.assertTrue(shexc.getExitCode() == 2);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}
 

import org.apache.hadoop.util.Shell.ExitCodeException; //导入方法依赖的package包/类
@Test (timeout = 20000)
public void testInvalidEnvSyntaxDiagnostics() throws IOException  {

  File shellFile = null;
  try {
    shellFile = Shell.appendScriptExtension(tmpDir, "hello");
    Map<Path, List<String>> resources =
        new HashMap<Path, List<String>>();
    FileOutputStream fos = new FileOutputStream(shellFile);
    FileUtil.setExecutable(shellFile, true);

    Map<String, String> env = new HashMap<String, String>();
    // invalid env
    env.put(
        "APPLICATION_WORKFLOW_CONTEXT", "{\"workflowId\":\"609f91c5cd83\"," +
        "\"workflowName\":\"\n\ninsert table " +
        "\npartition (cd_education_status)\nselect cd_demo_sk, cd_gender, " );
    List<String> commands = new ArrayList<String>();
    ContainerLaunch.writeLaunchEnv(fos, env, resources, commands);
    fos.flush();
    fos.close();

    // It is supposed that LANG is set as C.
    Map<String, String> cmdEnv = new HashMap<String, String>();
    cmdEnv.put("LANG", "C");
    Shell.ShellCommandExecutor shexc
    = new Shell.ShellCommandExecutor(new String[]{shellFile.getAbsolutePath()},
      tmpDir, cmdEnv);
    String diagnostics = null;
    try {
      shexc.execute();
      Assert.fail("Should catch exception");
    } catch(ExitCodeException e){
      diagnostics = e.getMessage();
    }
    Assert.assertTrue(diagnostics.contains(Shell.WINDOWS ?
        "is not recognized as an internal or external command" :
        "command not found"));
    Assert.assertTrue(shexc.getExitCode() != 0);
  }
  finally {
    // cleanup
    if (shellFile != null
        && shellFile.exists()) {
      shellFile.delete();
    }
  }
}