Chromium Code Reviews| Index: tools/gn/exec_process_unittest.cc |
| diff --git a/tools/gn/exec_process_unittest.cc b/tools/gn/exec_process_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..35f8dd4f3ee4a11969d7eb971aa74d54a9263465 |
| --- /dev/null |
| +++ b/tools/gn/exec_process_unittest.cc |
| @@ -0,0 +1,119 @@ |
| +// Copyright 2014 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "base/command_line.h" |
| +#include "base/files/scoped_temp_dir.h" |
| +#include "base/strings/string_util.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| +#include "tools/gn/exec_process.h" |
| + |
| +namespace internal { |
| + |
| +namespace { |
| +bool ExecPython(const std::string& command, |
| + std::string* std_out, |
| + std::string* std_err, |
| + int* exit_code) { |
| + base::ScopedTempDir temp_dir; |
| + base::CommandLine::StringVector args; |
| +#if defined(OS_WIN) |
| + args.push_back(L"python"); |
| + args.push_back(L"-c"); |
| + args.push_back(base::UTF8ToUTF16(command)); |
| +#else |
| + args.push_back("python"); |
| + args.push_back("-c"); |
| + args.push_back(command); |
| +#endif |
| + return ExecProcess( |
| + base::CommandLine(args), temp_dir.path(), std_out, std_err, exit_code); |
| +} |
| +} // namespace |
| + |
| +TEST(ExecProcessTest, TestExitCode) { |
| + std::string std_out, std_err; |
| + int exit_code; |
| + |
| + ASSERT_TRUE( |
| + ExecPython("import sys; sys.exit(0)", &std_out, &std_err, &exit_code)); |
| + EXPECT_EQ(0, exit_code); |
| + |
| + ASSERT_TRUE( |
| + ExecPython("import sys; sys.exit(1)", &std_out, &std_err, &exit_code)); |
| + EXPECT_EQ(1, exit_code); |
| + |
| + ASSERT_TRUE( |
| + ExecPython("import sys; sys.exit(253)", &std_out, &std_err, &exit_code)); |
| + EXPECT_EQ(253, exit_code); |
| + |
| + ASSERT_TRUE( |
| + ExecPython("throw Exception()", &std_out, &std_err, &exit_code)); |
| + EXPECT_EQ(1, exit_code); |
| +} |
| + |
| +// Test that large output is handled correctly. There are various ways that this |
| +// could potentially fail. For example, non-blocking Linux pipes have a 65536 |
| +// byte buffer and, if stdout is non-blocking, python will throw an IOError when |
| +// a write exceeds the buffer size. |
| +TEST(ExecProcessTest, TestLargeOutput) { |
| + base::ScopedTempDir temp_dir; |
| + std::string std_out, std_err; |
| + int exit_code; |
| + |
| + ASSERT_TRUE(ExecPython( |
| + "import sys; print 'o' * 1000000", &std_out, &std_err, &exit_code)); |
| + EXPECT_EQ(0, exit_code); |
| + EXPECT_EQ(1000001u, std_out.size()); |
| +} |
| + |
| +TEST(ExecProcessTest, TestStdoutAndStderrOutput) { |
|
cjhopman
2014/10/13 18:07:55
Windows should fail these two tests, so they still
|
| + base::ScopedTempDir temp_dir; |
| + std::string std_out, std_err; |
| + int exit_code; |
| + |
| + ASSERT_TRUE(ExecPython( |
| + "import sys; print 'o' * 10000; print >>sys.stderr, 'e' * 10000", |
| + &std_out, |
| + &std_err, |
| + &exit_code)); |
| + EXPECT_EQ(0, exit_code); |
| + EXPECT_EQ(10001u, std_out.size()); |
| + EXPECT_EQ(10001u, std_err.size()); |
| + |
| + std_out.clear(); |
| + std_err.clear(); |
| + ASSERT_TRUE(ExecPython( |
| + "import sys; print >>sys.stderr, 'e' * 10000; print 'o' * 10000", |
| + &std_out, |
| + &std_err, |
| + &exit_code)); |
| + EXPECT_EQ(0, exit_code); |
| + EXPECT_EQ(10001u, std_out.size()); |
| + EXPECT_EQ(10001u, std_err.size()); |
| +} |
| + |
| +TEST(ExecProcessTest, TestOneOutputClosed) { |
| + std::string std_out, std_err; |
| + int exit_code; |
| + |
| + ASSERT_TRUE(ExecPython("import sys; sys.stderr.close(); print 'o' * 10000", |
| + &std_out, |
| + &std_err, |
| + &exit_code)); |
| + EXPECT_EQ(0, exit_code); |
| + EXPECT_EQ(10001u, std_out.size()); |
| + EXPECT_EQ(std_err.size(), 0u); |
| + |
| + std_out.clear(); |
| + std_err.clear(); |
| + ASSERT_TRUE(ExecPython( |
| + "import sys; sys.stdout.close(); print >>sys.stderr, 'e' * 10000", |
| + &std_out, |
| + &std_err, |
| + &exit_code)); |
| + EXPECT_EQ(0, exit_code); |
| + EXPECT_EQ(0u, std_out.size()); |
| + EXPECT_EQ(10001u, std_err.size()); |
| +} |
| +} // namespace internal |