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/test/test_suite.h" | 5 #include "base/test/test_suite.h" |
6 | 6 |
7 #include "base/at_exit.h" | 7 #include "base/at_exit.h" |
8 #include "base/base_paths.h" | 8 #include "base/base_paths.h" |
9 #include "base/base_switches.h" | 9 #include "base/base_switches.h" |
10 #include "base/bind.h" | 10 #include "base/bind.h" |
(...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
83 | 83 |
84 int RunUnitTestsUsingBaseTestSuite(int argc, char **argv) { | 84 int RunUnitTestsUsingBaseTestSuite(int argc, char **argv) { |
85 TestSuite test_suite(argc, argv); | 85 TestSuite test_suite(argc, argv); |
86 return base::LaunchUnitTests( | 86 return base::LaunchUnitTests( |
87 argc, argv, Bind(&TestSuite::Run, Unretained(&test_suite))); | 87 argc, argv, Bind(&TestSuite::Run, Unretained(&test_suite))); |
88 } | 88 } |
89 | 89 |
90 } // namespace base | 90 } // namespace base |
91 | 91 |
92 TestSuite::TestSuite(int argc, char** argv) : initialized_command_line_(false) { | 92 TestSuite::TestSuite(int argc, char** argv) : initialized_command_line_(false) { |
93 PreInitialize(true); | 93 PreInitialize(argc, argv, true); |
94 InitializeFromCommandLine(argc, argv); | |
95 } | 94 } |
96 | 95 |
97 #if defined(OS_WIN) | |
98 TestSuite::TestSuite(int argc, wchar_t** argv) | |
99 : initialized_command_line_(false) { | |
100 PreInitialize(true); | |
101 InitializeFromCommandLine(argc, argv); | |
102 } | |
103 #endif // defined(OS_WIN) | |
104 | |
105 TestSuite::TestSuite(int argc, char** argv, bool create_at_exit_manager) | 96 TestSuite::TestSuite(int argc, char** argv, bool create_at_exit_manager) |
106 : initialized_command_line_(false) { | 97 : initialized_command_line_(false) { |
107 PreInitialize(create_at_exit_manager); | 98 PreInitialize(argc, argv, create_at_exit_manager); |
108 InitializeFromCommandLine(argc, argv); | |
109 } | 99 } |
110 | 100 |
111 TestSuite::~TestSuite() { | 101 TestSuite::~TestSuite() { |
112 if (initialized_command_line_) | 102 if (initialized_command_line_) |
113 CommandLine::Reset(); | 103 CommandLine::Reset(); |
114 } | 104 } |
115 | 105 |
116 void TestSuite::InitializeFromCommandLine(int argc, char** argv) { | 106 void TestSuite::PreInitialize(int argc, char** argv, |
117 initialized_command_line_ = CommandLine::Init(argc, argv); | 107 bool create_at_exit_manager) { |
118 testing::InitGoogleTest(&argc, argv); | |
119 testing::InitGoogleMock(&argc, argv); | |
120 | |
121 #if defined(OS_IOS) | |
122 InitIOSRunHook(this, argc, argv); | |
123 #endif | |
124 } | |
125 | |
126 #if defined(OS_WIN) | |
127 void TestSuite::InitializeFromCommandLine(int argc, wchar_t** argv) { | |
128 // Windows CommandLine::Init ignores argv anyway. | |
129 initialized_command_line_ = CommandLine::Init(argc, NULL); | |
130 testing::InitGoogleTest(&argc, argv); | |
131 testing::InitGoogleMock(&argc, argv); | |
132 } | |
133 #endif // defined(OS_WIN) | |
134 | |
135 void TestSuite::PreInitialize(bool create_at_exit_manager) { | |
136 #if defined(OS_WIN) | 108 #if defined(OS_WIN) |
137 testing::GTEST_FLAG(catch_exceptions) = false; | 109 testing::GTEST_FLAG(catch_exceptions) = false; |
138 base::TimeTicks::SetNowIsHighResNowIfSupported(); | 110 base::TimeTicks::SetNowIsHighResNowIfSupported(); |
139 #endif | 111 #endif |
140 base::EnableTerminationOnHeapCorruption(); | 112 base::EnableTerminationOnHeapCorruption(); |
| 113 initialized_command_line_ = CommandLine::Init(argc, argv); |
| 114 testing::InitGoogleTest(&argc, argv); |
| 115 testing::InitGoogleMock(&argc, argv); |
141 #if defined(OS_LINUX) && defined(USE_AURA) | 116 #if defined(OS_LINUX) && defined(USE_AURA) |
142 // When calling native char conversion functions (e.g wrctomb) we need to | 117 // When calling native char conversion functions (e.g wrctomb) we need to |
143 // have the locale set. In the absence of such a call the "C" locale is the | 118 // have the locale set. In the absence of such a call the "C" locale is the |
144 // default. In the gtk code (below) gtk_init() implicitly sets a locale. | 119 // default. In the gtk code (below) gtk_init() implicitly sets a locale. |
145 setlocale(LC_ALL, ""); | 120 setlocale(LC_ALL, ""); |
146 #endif // defined(OS_LINUX) && defined(USE_AURA) | 121 #endif // defined(OS_LINUX) && defined(USE_AURA) |
147 | 122 |
148 // On Android, AtExitManager is created in | 123 // On Android, AtExitManager is created in |
149 // testing/android/native_test_wrapper.cc before main() is called. | 124 // testing/android/native_test_wrapper.cc before main() is called. |
150 #if !defined(OS_ANDROID) | 125 #if !defined(OS_ANDROID) |
151 if (create_at_exit_manager) | 126 if (create_at_exit_manager) |
152 at_exit_manager_.reset(new base::AtExitManager); | 127 at_exit_manager_.reset(new base::AtExitManager); |
153 #endif | 128 #endif |
154 | 129 |
| 130 #if defined(OS_IOS) |
| 131 InitIOSRunHook(this, argc, argv); |
| 132 #endif |
| 133 |
155 // Don't add additional code to this function. Instead add it to | 134 // Don't add additional code to this function. Instead add it to |
156 // Initialize(). See bug 6436. | 135 // Initialize(). See bug 6436. |
157 } | 136 } |
158 | 137 |
159 | 138 |
160 // static | 139 // static |
161 bool TestSuite::IsMarkedMaybe(const testing::TestInfo& test) { | 140 bool TestSuite::IsMarkedMaybe(const testing::TestInfo& test) { |
162 return strncmp(test.name(), "MAYBE_", 6) == 0; | 141 return strncmp(test.name(), "MAYBE_", 6) == 0; |
163 } | 142 } |
164 | 143 |
(...skipping 163 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
328 ResetCommandLine(); | 307 ResetCommandLine(); |
329 #if !defined(OS_IOS) | 308 #if !defined(OS_IOS) |
330 AddTestLauncherResultPrinter(); | 309 AddTestLauncherResultPrinter(); |
331 #endif // !defined(OS_IOS) | 310 #endif // !defined(OS_IOS) |
332 | 311 |
333 TestTimeouts::Initialize(); | 312 TestTimeouts::Initialize(); |
334 } | 313 } |
335 | 314 |
336 void TestSuite::Shutdown() { | 315 void TestSuite::Shutdown() { |
337 } | 316 } |
OLD | NEW |