| 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/command_line.h" | 5 #include "base/command_line.h" |
| 6 #include "base/logging.h" | 6 #include "base/logging.h" |
| 7 #include "base/path_service.h" | 7 #include "base/path_service.h" |
| 8 #include "base/prefs/pref_service.h" | 8 #include "base/prefs/pref_service.h" |
| 9 #include "base/stringprintf.h" | 9 #include "base/stringprintf.h" |
| 10 #include "chrome/browser/browser_process.h" | 10 #include "chrome/browser/browser_process.h" |
| 11 #include "chrome/browser/download/download_service.h" | 11 #include "chrome/browser/download/download_service.h" |
| 12 #include "chrome/browser/download/download_service_factory.h" | 12 #include "chrome/browser/download/download_service_factory.h" |
| 13 #include "chrome/browser/download/download_test_file_activity_observer.h" | 13 #include "chrome/browser/download/download_test_file_activity_observer.h" |
| 14 #include "chrome/browser/net/url_request_mock_util.h" | 14 #include "chrome/browser/net/url_request_mock_util.h" |
| 15 #include "chrome/browser/profiles/profile.h" | 15 #include "chrome/browser/profiles/profile.h" |
| 16 #include "chrome/browser/profiles/profile_manager.h" | 16 #include "chrome/browser/profiles/profile_manager.h" |
| 17 #include "chrome/browser/ui/browser.h" | 17 #include "chrome/browser/ui/browser.h" |
| 18 #include "chrome/browser/ui/browser_list.h" | 18 #include "chrome/browser/ui/browser_iterator.h" |
| 19 #include "chrome/browser/ui/browser_tabstrip.h" | 19 #include "chrome/browser/ui/browser_tabstrip.h" |
| 20 #include "chrome/browser/ui/browser_window.h" | 20 #include "chrome/browser/ui/browser_window.h" |
| 21 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 21 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| 22 #include "chrome/common/chrome_paths.h" | 22 #include "chrome/common/chrome_paths.h" |
| 23 #include "chrome/common/pref_names.h" | 23 #include "chrome/common/pref_names.h" |
| 24 #include "chrome/common/url_constants.h" | 24 #include "chrome/common/url_constants.h" |
| 25 #include "chrome/test/base/in_process_browser_test.h" | 25 #include "chrome/test/base/in_process_browser_test.h" |
| 26 #include "chrome/test/base/ui_test_utils.h" | 26 #include "chrome/test/base/ui_test_utils.h" |
| 27 #include "content/public/browser/download_item.h" | 27 #include "content/public/browser/download_item.h" |
| 28 #include "content/public/common/page_transition_types.h" | 28 #include "content/public/common/page_transition_types.h" |
| (...skipping 175 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 204 // num_windows > 0 | 204 // num_windows > 0 |
| 205 Profile* profile((*base_browser)->profile()); | 205 Profile* profile((*base_browser)->profile()); |
| 206 for (int w = 1; w < num_windows; ++w) { | 206 for (int w = 1; w < num_windows; ++w) { |
| 207 CreateBrowserOnProfile(profile); | 207 CreateBrowserOnProfile(profile); |
| 208 } | 208 } |
| 209 return true; | 209 return true; |
| 210 } | 210 } |
| 211 | 211 |
| 212 int TotalUnclosedBrowsers() { | 212 int TotalUnclosedBrowsers() { |
| 213 int count = 0; | 213 int count = 0; |
| 214 for (BrowserList::const_iterator iter = BrowserList::begin(); | 214 for (chrome::BrowserIterator it; !it.done(); it.Next()) { |
| 215 iter != BrowserList::end(); ++iter) | 215 if (!it->IsAttemptingToCloseBrowser()) |
| 216 if (!(*iter)->IsAttemptingToCloseBrowser()) { | |
| 217 count++; | 216 count++; |
| 218 } | 217 } |
| 219 return count; | 218 return count; |
| 220 } | 219 } |
| 221 | 220 |
| 222 // Note that this is invalid to call if TotalUnclosedBrowsers() == 0. | 221 // Note that this is invalid to call if TotalUnclosedBrowsers() == 0. |
| 223 Browser* FirstUnclosedBrowser() { | 222 Browser* FirstUnclosedBrowser() { |
| 224 for (BrowserList::const_iterator iter = BrowserList::begin(); | 223 for (chrome::BrowserIterator it; !it.done(); it.Next()) { |
| 225 iter != BrowserList::end(); ++iter) | 224 if (!it->IsAttemptingToCloseBrowser()) |
| 226 if (!(*iter)->IsAttemptingToCloseBrowser()) | 225 return *it; |
| 227 return (*iter); | 226 } |
| 228 return NULL; | 227 return NULL; |
| 229 } | 228 } |
| 230 | 229 |
| 231 bool SetupForDownloadCloseCheck() { | 230 bool SetupForDownloadCloseCheck() { |
| 232 first_profile_ = browser()->profile(); | 231 first_profile_ = browser()->profile(); |
| 233 | 232 |
| 234 bool result = first_profile_downloads_dir_.CreateUniqueTempDir(); | 233 bool result = first_profile_downloads_dir_.CreateUniqueTempDir(); |
| 235 EXPECT_TRUE(result); | 234 EXPECT_TRUE(result); |
| 236 if (!result) return false; | 235 if (!result) return false; |
| 237 first_profile_->GetPrefs()->SetFilePath( | 236 first_profile_->GetPrefs()->SetFilePath( |
| (...skipping 115 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 353 if (type != Browser::DOWNLOAD_CLOSE_OK) | 352 if (type != Browser::DOWNLOAD_CLOSE_OK) |
| 354 EXPECT_EQ(check_case.num_blocking, num_downloads_blocking) | 353 EXPECT_EQ(check_case.num_blocking, num_downloads_blocking) |
| 355 << "Case " << i | 354 << "Case " << i |
| 356 << ": " << check_case.DebugString(); | 355 << ": " << check_case.DebugString(); |
| 357 | 356 |
| 358 // Release all the downloads. | 357 // Release all the downloads. |
| 359 CompleteAllDownloads(browser_to_probe); | 358 CompleteAllDownloads(browser_to_probe); |
| 360 | 359 |
| 361 // Create a new main window and kill everything else. | 360 // Create a new main window and kill everything else. |
| 362 entry_browser = CreateBrowserOnProfile(first_profile_); | 361 entry_browser = CreateBrowserOnProfile(first_profile_); |
| 363 for (BrowserList::const_iterator bit = BrowserList::begin(); | 362 for (chrome::BrowserIterator it; !it.done(); it.Next()) { |
| 364 bit != BrowserList::end(); ++bit) { | 363 if ((*it) != entry_browser) { |
| 365 if ((*bit) != entry_browser) { | 364 if (!it->window()) { |
| 366 EXPECT_TRUE((*bit)->window()); | 365 ADD_FAILURE(); |
| 367 if (!(*bit)->window()) | |
| 368 return false; | 366 return false; |
| 369 (*bit)->window()->Close(); | 367 } |
| 368 it->window()->Close(); |
| 370 } | 369 } |
| 371 } | 370 } |
| 372 content::RunAllPendingInMessageLoop(); | 371 content::RunAllPendingInMessageLoop(); |
| 373 | 372 |
| 374 return true; | 373 return true; |
| 375 } | 374 } |
| 376 | 375 |
| 377 static const DownloadsCloseCheckCase download_close_check_cases[]; | 376 static const DownloadsCloseCheckCase download_close_check_cases[]; |
| 378 | 377 |
| 379 // DownloadCloseCheck variables. | 378 // DownloadCloseCheck variables. |
| (...skipping 177 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 557 } | 556 } |
| 558 } | 557 } |
| 559 | 558 |
| 560 IN_PROC_BROWSER_TEST_F(BrowserCloseTest, MAYBE_DownloadsCloseCheck_5) { | 559 IN_PROC_BROWSER_TEST_F(BrowserCloseTest, MAYBE_DownloadsCloseCheck_5) { |
| 561 ASSERT_TRUE(SetupForDownloadCloseCheck()); | 560 ASSERT_TRUE(SetupForDownloadCloseCheck()); |
| 562 for (size_t i = 5 * arraysize(download_close_check_cases) / 6; | 561 for (size_t i = 5 * arraysize(download_close_check_cases) / 6; |
| 563 i < 6 * arraysize(download_close_check_cases) / 6; ++i) { | 562 i < 6 * arraysize(download_close_check_cases) / 6; ++i) { |
| 564 ExecuteDownloadCloseCheckCase(i); | 563 ExecuteDownloadCloseCheckCase(i); |
| 565 } | 564 } |
| 566 } | 565 } |
| OLD | NEW |