OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/at_exit.h" | 5 #include "base/at_exit.h" |
6 #include "base/base_switches.h" | 6 #include "base/base_switches.h" |
| 7 #include "base/bind.h" |
7 #include "base/command_line.h" | 8 #include "base/command_line.h" |
8 #include "base/files/file_path.h" | 9 #include "base/files/file_path.h" |
9 #include "base/files/file_util.h" | 10 #include "base/files/file_util.h" |
10 #include "base/logging.h" | 11 #include "base/logging.h" |
11 #include "base/test/test_suite.h" | 12 #include "base/test/test_suite.h" |
12 #include "build/build_config.h" | 13 #include "build/build_config.h" |
13 #include "sandbox/linux/tests/test_utils.h" | 14 #include "sandbox/linux/tests/test_utils.h" |
14 #include "sandbox/linux/tests/unit_tests.h" | 15 #include "sandbox/linux/tests/unit_tests.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
16 #include "testing/multiprocess_func_list.h" | 17 #include "testing/multiprocess_func_list.h" |
(...skipping 13 matching lines...) Expand all Loading... |
30 if (orig_cwd != cwd) { | 31 if (orig_cwd != cwd) { |
31 LOG(FATAL) << "One of the tests changed the current working directory. " | 32 LOG(FATAL) << "One of the tests changed the current working directory. " |
32 << "Please, clean up after your tests!"; | 33 << "Please, clean up after your tests!"; |
33 } | 34 } |
34 } | 35 } |
35 | 36 |
36 } // namespace | 37 } // namespace |
37 } // namespace sandbox | 38 } // namespace sandbox |
38 | 39 |
39 #if !defined(SANDBOX_USES_BASE_TEST_SUITE) | 40 #if !defined(SANDBOX_USES_BASE_TEST_SUITE) |
40 void UnitTestAssertHandler(const std::string& str) { | 41 void UnitTestAssertHandler(const char* file, |
| 42 int line, |
| 43 const base::StringPiece message, |
| 44 const base::StringPiece stack_trace) { |
41 _exit(1); | 45 _exit(1); |
42 } | 46 } |
43 #endif | 47 #endif |
44 | 48 |
45 int main(int argc, char* argv[]) { | 49 int main(int argc, char* argv[]) { |
46 base::CommandLine::Init(argc, argv); | 50 base::CommandLine::Init(argc, argv); |
47 std::string client_func; | 51 std::string client_func; |
48 #if defined(SANDBOX_USES_BASE_TEST_SUITE) | 52 #if defined(SANDBOX_USES_BASE_TEST_SUITE) |
49 client_func = base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII( | 53 client_func = base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII( |
50 switches::kTestChildProcess); | 54 switches::kTestChildProcess); |
51 #endif | 55 #endif |
52 if (!client_func.empty()) { | 56 if (!client_func.empty()) { |
53 base::AtExitManager exit_manager; | 57 base::AtExitManager exit_manager; |
54 return multi_process_function_list::InvokeChildProcessTest(client_func); | 58 return multi_process_function_list::InvokeChildProcessTest(client_func); |
55 } | 59 } |
56 | 60 |
57 base::FilePath orig_cwd; | 61 base::FilePath orig_cwd; |
58 CHECK(GetCurrentDirectory(&orig_cwd)); | 62 CHECK(GetCurrentDirectory(&orig_cwd)); |
59 | 63 |
60 #if !defined(SANDBOX_USES_BASE_TEST_SUITE) | 64 #if !defined(SANDBOX_USES_BASE_TEST_SUITE) |
61 // The use of Callbacks requires an AtExitManager. | 65 // The use of Callbacks requires an AtExitManager. |
62 base::AtExitManager exit_manager; | 66 base::AtExitManager exit_manager; |
63 testing::InitGoogleTest(&argc, argv); | 67 testing::InitGoogleTest(&argc, argv); |
64 // Death tests rely on LOG(FATAL) triggering an exit (the default behavior is | 68 // Death tests rely on LOG(FATAL) triggering an exit (the default behavior is |
65 // SIGABRT). The normal test launcher does this at initialization, but since | 69 // SIGABRT). The normal test launcher does this at initialization, but since |
66 // we still do not use this on Android, we must install the handler ourselves. | 70 // we still do not use this on Android, we must install the handler ourselves. |
67 logging::SetLogAssertHandler(UnitTestAssertHandler); | 71 logging::ScopedLogAssertHandler scoped_assert_handler( |
| 72 base::Bind(UnitTestAssertHandler)); |
68 #endif | 73 #endif |
69 // Always go through re-execution for death tests. | 74 // Always go through re-execution for death tests. |
70 // This makes gtest only marginally slower for us and has the | 75 // This makes gtest only marginally slower for us and has the |
71 // additional side effect of getting rid of gtest warnings about fork() | 76 // additional side effect of getting rid of gtest warnings about fork() |
72 // safety. | 77 // safety. |
73 ::testing::FLAGS_gtest_death_test_style = "threadsafe"; | 78 ::testing::FLAGS_gtest_death_test_style = "threadsafe"; |
74 #if !defined(SANDBOX_USES_BASE_TEST_SUITE) | 79 #if !defined(SANDBOX_USES_BASE_TEST_SUITE) |
75 int tests_result = RUN_ALL_TESTS(); | 80 int tests_result = RUN_ALL_TESTS(); |
76 #else | 81 #else |
77 int tests_result = base::RunUnitTestsUsingBaseTestSuite(argc, argv); | 82 int tests_result = base::RunUnitTestsUsingBaseTestSuite(argc, argv); |
78 #endif | 83 #endif |
79 | 84 |
80 sandbox::RunPostTestsChecks(orig_cwd); | 85 sandbox::RunPostTestsChecks(orig_cwd); |
81 return tests_result; | 86 return tests_result; |
82 } | 87 } |
OLD | NEW |