tools/gn/exec_process_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/command_line.h"
   6 #include "base/files/scoped_temp_dir.h"
   7 #include "base/strings/string_util.h"
   8 #include "testing/gtest/include/gtest/gtest.h"
   9 #include "tools/gn/exec_process.h"
  10
  11 #if defined(OS_WIN)
  12 #include "base/strings/utf_string_conversions.h"
  13 #endif
  14
  15 namespace internal {
  16
  17 // TODO(cjhopman): Enable these tests when windows ExecProcess handles stderr.
  18 // 'python' is not runnable on Windows. Adding ["cmd", "/c"] fails because
  19 // CommandLine does unusual reordering of args.
  20 #if !defined(OS_WIN)
  21 namespace {
  22 bool ExecPython(const std::string& command,
  23                 std::string* std_out,
  24                 std::string* std_err,
  25                 int* exit_code) {
  26   base::ScopedTempDir temp_dir;
  27   base::CommandLine::StringVector args;
  28 #if defined(OS_WIN)
  29   args.push_back(L"python");
  30   args.push_back(L"-c");
  31   args.push_back(base::UTF8ToUTF16(command));
  32 #else
  33   args.push_back("python");
  34   args.push_back("-c");
  35   args.push_back(command);
  36 #endif
  37   return ExecProcess(
  38       base::CommandLine(args), temp_dir.path(), std_out, std_err, exit_code);
  39 }
  40 }  // namespace
  41
  42 TEST(ExecProcessTest, TestExitCode) {
  43   std::string std_out, std_err;
  44   int exit_code;
  45
  46   ASSERT_TRUE(
  47       ExecPython("import sys; sys.exit(0)", &std_out, &std_err, &exit_code));
  48   EXPECT_EQ(0, exit_code);
  49
  50   ASSERT_TRUE(
  51       ExecPython("import sys; sys.exit(1)", &std_out, &std_err, &exit_code));
  52   EXPECT_EQ(1, exit_code);
  53
  54   ASSERT_TRUE(
  55       ExecPython("import sys; sys.exit(253)", &std_out, &std_err, &exit_code));
  56   EXPECT_EQ(253, exit_code);
  57
  58   ASSERT_TRUE(
  59       ExecPython("throw Exception()", &std_out, &std_err, &exit_code));
  60   EXPECT_EQ(1, exit_code);
  61 }
  62
  63 // Test that large output is handled correctly. There are various ways that this
  64 // could potentially fail. For example, non-blocking Linux pipes have a 65536
  65 // byte buffer and, if stdout is non-blocking, python will throw an IOError when
  66 // a write exceeds the buffer size.
  67 TEST(ExecProcessTest, TestLargeOutput) {
  68   base::ScopedTempDir temp_dir;
  69   std::string std_out, std_err;
  70   int exit_code;
  71
  72   ASSERT_TRUE(ExecPython(
  73       "import sys; print 'o' * 1000000", &std_out, &std_err, &exit_code));
  74   EXPECT_EQ(0, exit_code);
  75   EXPECT_EQ(1000001u, std_out.size());
  76 }
  77
  78 TEST(ExecProcessTest, TestStdoutAndStderrOutput) {
  79   base::ScopedTempDir temp_dir;
  80   std::string std_out, std_err;
  81   int exit_code;
  82
  83   ASSERT_TRUE(ExecPython(
  84       "import sys; print 'o' * 10000; print >>sys.stderr, 'e' * 10000",
  85       &std_out,
  86       &std_err,
  87       &exit_code));
  88   EXPECT_EQ(0, exit_code);
  89   EXPECT_EQ(10001u, std_out.size());
  90   EXPECT_EQ(10001u, std_err.size());
  91
  92   std_out.clear();
  93   std_err.clear();
  94   ASSERT_TRUE(ExecPython(
  95       "import sys; print >>sys.stderr, 'e' * 10000; print 'o' * 10000",
  96       &std_out,
  97       &std_err,
  98       &exit_code));
  99   EXPECT_EQ(0, exit_code);
 100   EXPECT_EQ(10001u, std_out.size());
 101   EXPECT_EQ(10001u, std_err.size());
 102 }
 103
 104 TEST(ExecProcessTest, TestOneOutputClosed) {
 105   std::string std_out, std_err;
 106   int exit_code;
 107
 108   ASSERT_TRUE(ExecPython("import sys; sys.stderr.close(); print 'o' * 10000",
 109                          &std_out,
 110                          &std_err,
 111                          &exit_code));
 112   EXPECT_EQ(0, exit_code);
 113   EXPECT_EQ(10001u, std_out.size());
 114   EXPECT_EQ(std_err.size(), 0u);
 115
 116   std_out.clear();
 117   std_err.clear();
 118   ASSERT_TRUE(ExecPython(
 119       "import sys; sys.stdout.close(); print >>sys.stderr, 'e' * 10000",
 120       &std_out,
 121       &std_err,
 122       &exit_code));
 123   EXPECT_EQ(0, exit_code);
 124   EXPECT_EQ(0u, std_out.size());
 125   EXPECT_EQ(10001u, std_err.size());
 126 }
 127 #endif
 128 }  // namespace internal