| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "chrome/test/in_process_browser_test.h" | 5 #include "chrome/test/in_process_browser_test.h" |
| 6 | 6 |
| 7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
| 8 #include "base/file_path.h" | 8 #include "base/file_path.h" |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/mac/scoped_nsautorelease_pool.h" | 10 #include "base/mac/scoped_nsautorelease_pool.h" |
| (...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 67 | 67 |
| 68 const char kUnitTestShowWindows[] = "show-windows"; | 68 const char kUnitTestShowWindows[] = "show-windows"; |
| 69 | 69 |
| 70 // Passed as value of kTestType. | 70 // Passed as value of kTestType. |
| 71 static const char kBrowserTestType[] = "browser"; | 71 static const char kBrowserTestType[] = "browser"; |
| 72 | 72 |
| 73 InProcessBrowserTest::InProcessBrowserTest() | 73 InProcessBrowserTest::InProcessBrowserTest() |
| 74 : browser_(NULL), | 74 : browser_(NULL), |
| 75 show_window_(false), | 75 show_window_(false), |
| 76 dom_automation_enabled_(false), | 76 dom_automation_enabled_(false), |
| 77 tab_closeable_state_watcher_enabled_(false), | 77 tab_closeable_state_watcher_enabled_(false) { |
| 78 original_single_process_(false) { | |
| 79 #if defined(OS_MACOSX) | 78 #if defined(OS_MACOSX) |
| 80 base::mac::SetOverrideAmIBundled(true); | 79 base::mac::SetOverrideAmIBundled(true); |
| 81 #endif | 80 #endif |
| 82 | 81 |
| 83 // Before we run the browser, we have to hack the path to the exe to match | 82 // Before we run the browser, we have to hack the path to the exe to match |
| 84 // what it would be if Chrome was running, because it is used to fork renderer | 83 // what it would be if Chrome was running, because it is used to fork renderer |
| 85 // processes, on Linux at least (failure to do so will cause a browser_test to | 84 // processes, on Linux at least (failure to do so will cause a browser_test to |
| 86 // be run instead of a renderer). | 85 // be run instead of a renderer). |
| 87 FilePath chrome_path; | 86 FilePath chrome_path; |
| 88 CHECK(PathService::Get(base::FILE_EXE, &chrome_path)); | 87 CHECK(PathService::Get(base::FILE_EXE, &chrome_path)); |
| 89 chrome_path = chrome_path.DirName(); | 88 chrome_path = chrome_path.DirName(); |
| 90 chrome_path = chrome_path.Append(chrome::kBrowserProcessExecutablePath); | 89 chrome_path = chrome_path.Append(chrome::kBrowserProcessExecutablePath); |
| 91 CHECK(PathService::Override(base::FILE_EXE, chrome_path)); | 90 CHECK(PathService::Override(base::FILE_EXE, chrome_path)); |
| 92 | 91 |
| 93 test_server_.reset(new net::TestServer( | 92 test_server_.reset(new net::TestServer( |
| 94 net::TestServer::TYPE_HTTP, | 93 net::TestServer::TYPE_HTTP, |
| 95 FilePath(FILE_PATH_LITERAL("chrome/test/data")))); | 94 FilePath(FILE_PATH_LITERAL("chrome/test/data")))); |
| 96 } | 95 } |
| 97 | 96 |
| 98 InProcessBrowserTest::~InProcessBrowserTest() { | 97 InProcessBrowserTest::~InProcessBrowserTest() { |
| 99 } | 98 } |
| 100 | 99 |
| 101 void InProcessBrowserTest::SetUp() { | 100 void InProcessBrowserTest::SetUp() { |
| 102 // Remember the command line. Normally this doesn't matter, because the test | |
| 103 // harness creates a new process for each test, but when the test harness is | |
| 104 // running in single process mode, we can't let one test's command-line | |
| 105 // changes (e.g. enabling DOM automation) affect other tests. | |
| 106 // TODO(phajdan.jr): This save/restore logic is unnecessary. Remove it. | |
| 107 CommandLine* command_line = CommandLine::ForCurrentProcessMutable(); | |
| 108 original_command_line_.reset(new CommandLine(*command_line)); | |
| 109 | |
| 110 // Create a temporary user data directory if required. | 101 // Create a temporary user data directory if required. |
| 111 ASSERT_TRUE(CreateUserDataDirectory()) | 102 ASSERT_TRUE(CreateUserDataDirectory()) |
| 112 << "Could not create user data directory."; | 103 << "Could not create user data directory."; |
| 113 | 104 |
| 114 // The unit test suite creates a testingbrowser, but we want the real thing. | 105 // The unit test suite creates a testingbrowser, but we want the real thing. |
| 115 // Delete the current one. We'll install the testing one in TearDown. | 106 // Delete the current one. We'll install the testing one in TearDown. |
| 116 delete g_browser_process; | 107 delete g_browser_process; |
| 117 g_browser_process = NULL; | 108 g_browser_process = NULL; |
| 118 | 109 |
| 119 // Allow subclasses the opportunity to make changes to the default user data | 110 // Allow subclasses the opportunity to make changes to the default user data |
| 120 // dir before running any tests. | 111 // dir before running any tests. |
| 121 ASSERT_TRUE(SetUpUserDataDirectory()) | 112 ASSERT_TRUE(SetUpUserDataDirectory()) |
| 122 << "Could not set up user data directory."; | 113 << "Could not set up user data directory."; |
| 123 | 114 |
| 124 // Don't delete the resources when BrowserMain returns. Many ui classes | 115 // Don't delete the resources when BrowserMain returns. Many ui classes |
| 125 // cache SkBitmaps in a static field so that if we delete the resource | 116 // cache SkBitmaps in a static field so that if we delete the resource |
| 126 // bundle we'll crash. | 117 // bundle we'll crash. |
| 127 browser_shutdown::delete_resources_on_shutdown = false; | 118 browser_shutdown::delete_resources_on_shutdown = false; |
| 128 | 119 |
| 129 // Allow subclasses the opportunity to make changes to the command line before | 120 CommandLine* command_line = CommandLine::ForCurrentProcess(); |
| 130 // running any tests. | 121 // Allow subclasses to change the command line before running any tests. |
| 131 SetUpCommandLine(command_line); | 122 SetUpCommandLine(command_line); |
| 132 // Add command line arguments that are used by all InProcessBrowserTests. | 123 // Add command line arguments that are used by all InProcessBrowserTests. |
| 133 PrepareTestCommandLine(command_line); | 124 PrepareTestCommandLine(command_line); |
| 134 | 125 |
| 135 // Save the single process mode state before it was reset in this test. This | 126 // Single-process mode is not set in BrowserMain, so process it explicitly. |
| 136 // state will be recovered in TearDown(). Single-process mode is not set in | |
| 137 // BrowserMain so it needs to be processed explicitly. | |
| 138 original_single_process_ = RenderProcessHost::run_renderer_in_process(); | |
| 139 if (command_line->HasSwitch(switches::kSingleProcess)) | 127 if (command_line->HasSwitch(switches::kSingleProcess)) |
| 140 RenderProcessHost::set_run_renderer_in_process(true); | 128 RenderProcessHost::set_run_renderer_in_process(true); |
| 141 | 129 |
| 142 #if defined(OS_CHROMEOS) | 130 #if defined(OS_CHROMEOS) |
| 143 // Make sure that the log directory exists. | 131 // Make sure that the log directory exists. |
| 144 FilePath log_dir = logging::GetSessionLogFile(*command_line).DirName(); | 132 FilePath log_dir = logging::GetSessionLogFile(*command_line).DirName(); |
| 145 file_util::CreateDirectory(log_dir); | 133 file_util::CreateDirectory(log_dir); |
| 146 #endif // defined(OS_CHROMEOS) | 134 #endif // defined(OS_CHROMEOS) |
| 147 | 135 |
| 148 SandboxInitWrapper sandbox_wrapper; | 136 SandboxInitWrapper sandbox_wrapper; |
| (...skipping 14 matching lines...) Expand all Loading... |
| 163 | 151 |
| 164 net::ScopedDefaultHostResolverProc scoped_host_resolver_proc( | 152 net::ScopedDefaultHostResolverProc scoped_host_resolver_proc( |
| 165 host_resolver_.get()); | 153 host_resolver_.get()); |
| 166 | 154 |
| 167 SetUpInProcessBrowserTestFixture(); | 155 SetUpInProcessBrowserTestFixture(); |
| 168 | 156 |
| 169 BrowserMain(params); | 157 BrowserMain(params); |
| 170 TearDownInProcessBrowserTestFixture(); | 158 TearDownInProcessBrowserTestFixture(); |
| 171 } | 159 } |
| 172 | 160 |
| 173 void InProcessBrowserTest::PrepareTestCommandLine( | 161 void InProcessBrowserTest::PrepareTestCommandLine(CommandLine* command_line) { |
| 174 CommandLine* command_line) { | |
| 175 // Propagate commandline settings from test_launcher_utils. | 162 // Propagate commandline settings from test_launcher_utils. |
| 176 test_launcher_utils::PrepareBrowserCommandLineForTests(command_line); | 163 test_launcher_utils::PrepareBrowserCommandLineForTests(command_line); |
| 177 | 164 |
| 178 #if defined(OS_WIN) | 165 #if defined(OS_WIN) |
| 179 // Hide windows on show. | 166 // Hide windows on show. |
| 180 if (!command_line->HasSwitch(kUnitTestShowWindows) && !show_window_) | 167 if (!command_line->HasSwitch(kUnitTestShowWindows) && !show_window_) |
| 181 BrowserView::SetShowState(SW_HIDE); | 168 BrowserView::SetShowState(SW_HIDE); |
| 182 #endif | 169 #endif |
| 183 | 170 |
| 184 if (dom_automation_enabled_) | 171 if (dom_automation_enabled_) |
| (...skipping 24 matching lines...) Expand all Loading... |
| 209 command_line->AppendSwitchPath(switches::kBrowserSubprocessPath, | 196 command_line->AppendSwitchPath(switches::kBrowserSubprocessPath, |
| 210 subprocess_path); | 197 subprocess_path); |
| 211 #endif | 198 #endif |
| 212 | 199 |
| 213 // If neccessary, disable TabCloseableStateWatcher. | 200 // If neccessary, disable TabCloseableStateWatcher. |
| 214 if (!tab_closeable_state_watcher_enabled_) | 201 if (!tab_closeable_state_watcher_enabled_) |
| 215 command_line->AppendSwitch(switches::kDisableTabCloseableStateWatcher); | 202 command_line->AppendSwitch(switches::kDisableTabCloseableStateWatcher); |
| 216 } | 203 } |
| 217 | 204 |
| 218 bool InProcessBrowserTest::CreateUserDataDirectory() { | 205 bool InProcessBrowserTest::CreateUserDataDirectory() { |
| 219 CommandLine* command_line = CommandLine::ForCurrentProcessMutable(); | 206 CommandLine* command_line = CommandLine::ForCurrentProcess(); |
| 220 FilePath user_data_dir = | 207 FilePath user_data_dir = |
| 221 command_line->GetSwitchValuePath(switches::kUserDataDir); | 208 command_line->GetSwitchValuePath(switches::kUserDataDir); |
| 222 if (user_data_dir.empty()) { | 209 if (user_data_dir.empty()) { |
| 223 if (temp_user_data_dir_.CreateUniqueTempDir() && | 210 if (temp_user_data_dir_.CreateUniqueTempDir() && |
| 224 temp_user_data_dir_.IsValid()) { | 211 temp_user_data_dir_.IsValid()) { |
| 225 user_data_dir = temp_user_data_dir_.path(); | 212 user_data_dir = temp_user_data_dir_.path(); |
| 226 } else { | 213 } else { |
| 227 LOG(ERROR) << "Could not create temporary user data directory \"" | 214 LOG(ERROR) << "Could not create temporary user data directory \"" |
| 228 << temp_user_data_dir_.path().value() << "\"."; | 215 << temp_user_data_dir_.path().value() << "\"."; |
| 229 return false; | 216 return false; |
| 230 } | 217 } |
| 231 } | 218 } |
| 232 return test_launcher_utils::OverrideUserDataDir(user_data_dir); | 219 return test_launcher_utils::OverrideUserDataDir(user_data_dir); |
| 233 } | 220 } |
| 234 | 221 |
| 235 void InProcessBrowserTest::TearDown() { | 222 void InProcessBrowserTest::TearDown() { |
| 236 // Reinstall testing browser process. | 223 // Reinstall testing browser process. |
| 237 delete g_browser_process; | 224 delete g_browser_process; |
| 238 g_browser_process = new TestingBrowserProcess(); | 225 g_browser_process = new TestingBrowserProcess(); |
| 239 | 226 |
| 240 browser_shutdown::delete_resources_on_shutdown = true; | 227 browser_shutdown::delete_resources_on_shutdown = true; |
| 241 | 228 |
| 242 #if defined(OS_WIN) | 229 #if defined(OS_WIN) |
| 243 BrowserView::SetShowState(-1); | 230 BrowserView::SetShowState(-1); |
| 244 #endif | 231 #endif |
| 245 | |
| 246 *CommandLine::ForCurrentProcessMutable() = *original_command_line_; | |
| 247 RenderProcessHost::set_run_renderer_in_process(original_single_process_); | |
| 248 } | 232 } |
| 249 | 233 |
| 250 void InProcessBrowserTest::AddTabAtIndexToBrowser( | 234 void InProcessBrowserTest::AddTabAtIndexToBrowser( |
| 251 Browser* browser, | 235 Browser* browser, |
| 252 int index, | 236 int index, |
| 253 const GURL& url, | 237 const GURL& url, |
| 254 PageTransition::Type transition) { | 238 PageTransition::Type transition) { |
| 255 browser::NavigateParams params(browser, url, transition); | 239 browser::NavigateParams params(browser, url, transition); |
| 256 params.tabstrip_index = index; | 240 params.tabstrip_index = index; |
| 257 params.disposition = NEW_FOREGROUND_TAB; | 241 params.disposition = NEW_FOREGROUND_TAB; |
| (...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 352 return; | 336 return; |
| 353 | 337 |
| 354 // Invoke CloseAllBrowsersAndExit on a running message loop. | 338 // Invoke CloseAllBrowsersAndExit on a running message loop. |
| 355 // CloseAllBrowsersAndExit exits the message loop after everything has been | 339 // CloseAllBrowsersAndExit exits the message loop after everything has been |
| 356 // shut down properly. | 340 // shut down properly. |
| 357 MessageLoopForUI::current()->PostTask( | 341 MessageLoopForUI::current()->PostTask( |
| 358 FROM_HERE, | 342 FROM_HERE, |
| 359 NewRunnableFunction(&BrowserList::CloseAllBrowsersAndExit)); | 343 NewRunnableFunction(&BrowserList::CloseAllBrowsersAndExit)); |
| 360 ui_test_utils::RunMessageLoop(); | 344 ui_test_utils::RunMessageLoop(); |
| 361 } | 345 } |
| OLD | NEW |