| Index: chrome/browser/download/save_page_browsertest.cc
|
| diff --git a/chrome/browser/download/save_page_browsertest.cc b/chrome/browser/download/save_page_browsertest.cc
|
| index eff3068dd3844c76aac59f28a659a5dfbc9f3cee..33d1de6c7a61af2e1caf8c56e2cd06b4607daf28 100644
|
| --- a/chrome/browser/download/save_page_browsertest.cc
|
| +++ b/chrome/browser/download/save_page_browsertest.cc
|
| @@ -18,6 +18,7 @@
|
| #include "chrome/browser/download/download_prefs.h"
|
| #include "chrome/browser/download/download_service.h"
|
| #include "chrome/browser/download/download_service_factory.h"
|
| +#include "chrome/browser/download/save_package_file_picker.h"
|
| #include "chrome/browser/history/download_row.h"
|
| #include "chrome/browser/net/url_request_mock_util.h"
|
| #include "chrome/browser/profiles/profile.h"
|
| @@ -40,12 +41,6 @@
|
| #include "content/test/net/url_request_mock_http_job.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| -#if defined(OS_CHROMEOS)
|
| -#include "chrome/browser/download/save_package_file_picker_chromeos.h"
|
| -#else
|
| -#include "chrome/browser/download/save_package_file_picker.h"
|
| -#endif
|
| -
|
| using content::BrowserContext;
|
| using content::BrowserThread;
|
| using content::DownloadItem;
|
| @@ -65,7 +60,9 @@ class DownloadPersistedObserver : public DownloadHistory::Observer {
|
|
|
| DownloadPersistedObserver(Profile* profile, const PersistedFilter& filter)
|
| : profile_(profile),
|
| - filter_(filter) {
|
| + filter_(filter),
|
| + waiting_(false),
|
| + persisted_(false) {
|
| DownloadServiceFactory::GetForProfile(profile_)->
|
| GetDownloadHistory()->AddObserver(this);
|
| }
|
| @@ -87,7 +84,7 @@ class DownloadPersistedObserver : public DownloadHistory::Observer {
|
|
|
| virtual void OnDownloadStored(DownloadItem* item,
|
| const history::DownloadRow& info) OVERRIDE {
|
| - persisted_ = filter_.Run(item, info);
|
| + persisted_ = persisted_ || filter_.Run(item, info);
|
| if (persisted_ && waiting_)
|
| base::MessageLoopForUI::current()->Quit();
|
| }
|
| @@ -293,6 +290,8 @@ class SavePageBrowserTest : public InProcessBrowserTest {
|
| virtual void SetUpOnMainThread() OVERRIDE {
|
| browser()->profile()->GetPrefs()->SetFilePath(
|
| prefs::kDownloadDefaultDirectory, save_dir_.path());
|
| + browser()->profile()->GetPrefs()->SetFilePath(
|
| + prefs::kSaveFileDefaultDirectory, save_dir_.path());
|
| BrowserThread::PostTask(
|
| BrowserThread::IO, FROM_HERE,
|
| base::Bind(&chrome_browser_net::SetUrlRequestMocksEnabled, true));
|
| @@ -754,11 +753,7 @@ IN_PROC_BROWSER_TEST_F(SavePageAsMHTMLBrowserTest, SavePageAsMHTML) {
|
| GetDownloadManager())->DownloadPath();
|
| base::FilePath full_file_name = download_dir.AppendASCII(std::string(
|
| "Test page for saving page feature.mhtml"));
|
| -#if defined(OS_CHROMEOS)
|
| - SavePackageFilePickerChromeOS::SetShouldPromptUser(false);
|
| -#else
|
| SavePackageFilePicker::SetShouldPromptUser(false);
|
| -#endif
|
| DownloadPersistedObserver persisted(browser()->profile(), base::Bind(
|
| &DownloadStoredProperly, url, full_file_name, -1,
|
| DownloadItem::COMPLETE));
|
| @@ -772,18 +767,14 @@ IN_PROC_BROWSER_TEST_F(SavePageAsMHTMLBrowserTest, SavePageAsMHTML) {
|
| ASSERT_TRUE(VerifySavePackageExpectations(browser(), url));
|
| persisted.WaitForPersisted();
|
|
|
| - EXPECT_TRUE(file_util::PathExists(full_file_name));
|
| + ASSERT_TRUE(file_util::PathExists(full_file_name));
|
| int64 actual_file_size = -1;
|
| EXPECT_TRUE(file_util::GetFileSize(full_file_name, &actual_file_size));
|
| EXPECT_LE(kFileSizeMin, actual_file_size);
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(SavePageBrowserTest, SavePageBrowserTest_NonMHTML) {
|
| -#if defined(OS_CHROMEOS)
|
| - SavePackageFilePickerChromeOS::SetShouldPromptUser(false);
|
| -#else
|
| SavePackageFilePicker::SetShouldPromptUser(false);
|
| -#endif
|
| GURL url("data:text/plain,foo");
|
| ui_test_utils::NavigateToURL(browser(), url);
|
| scoped_refptr<content::MessageLoopRunner> loop_runner(
|
| @@ -796,7 +787,7 @@ IN_PROC_BROWSER_TEST_F(SavePageBrowserTest, SavePageBrowserTest_NonMHTML) {
|
| base::FilePath download_dir = DownloadPrefs::FromDownloadManager(
|
| GetDownloadManager())->DownloadPath();
|
| base::FilePath filename = download_dir.AppendASCII("dataurl.txt");
|
| - EXPECT_TRUE(file_util::PathExists(filename));
|
| + ASSERT_TRUE(file_util::PathExists(filename));
|
| std::string contents;
|
| EXPECT_TRUE(file_util::ReadFileToString(filename, &contents));
|
| EXPECT_EQ("foo", contents);
|
|
|