Save Page As MHTML

chrome/browser/download/save_page_browsertest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/scoped_temp_dir.h"
 #include "base/test/test_file_util.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/download/chrome_download_manager_delegate.h"
 #include "chrome/browser/download/download_history.h"
 #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/net/url_request_mock_util.h"
+#include "chrome/browser/prefs/pref_member.h"
+#include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/webui/chromeos/active_downloads_ui.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/download_item.h"
 #include "content/public/browser/download_manager.h"
 #include "content/public/browser/download_persistent_store_info.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/web_contents.h"
 #include "content/test/net/url_request_mock_http_job.h"
@@ skipping 20 matching lines @@
 
 class SavePageBrowserTest : public InProcessBrowserTest {
  protected:
   void SetUp() OVERRIDE {
     ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_dir_));
     ASSERT_TRUE(save_dir_.CreateUniqueTempDir());
     InProcessBrowserTest::SetUp();
   }
 
   void SetUpOnMainThread() OVERRIDE {
+    browser()->profile()->GetPrefs()->SetFilePath(
+        prefs::kDownloadDefaultDirectory, save_dir_.path());

[Randy Smith (Not in Mondays), 2012/04/16 17:45:47]

What's the context for this change?

[benjhayden, 2012/04/16 19:08:06]

I noticed it was leaving files in my ~/Downloads/.

     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
         base::Bind(&chrome_browser_net::SetUrlRequestMocksEnabled, true));
   }
 
   GURL NavigateToMockURL(const std::string& prefix) {
     GURL url = URLRequestMockHTTPJob::GetMockUrl(
         FilePath(kTestDir).AppendASCII(prefix + ".htm"));
     ui_test_utils::NavigateToURL(browser(), url);
     return url;
@@ skipping 102 matching lines @@
         info.state == DownloadItem::COMPLETE;
     }
 
     GURL url_;
     FilePath path_;
     int64 num_files_;
   };
 
   void CheckDownloadHistory(const GURL& url,
                             const FilePath& path,
-                            int64 num_files_) {
+                            int64 num_files) {
     QueryDownloadHistory();
 
     EXPECT_NE(std::find_if(history_entries_.begin(), history_entries_.end(),
-        DownloadPersistentStoreInfoMatch(url, path, num_files_)),
+        DownloadPersistentStoreInfoMatch(url, path, num_files)),
         history_entries_.end());
   }
 
   std::vector<DownloadPersistentStoreInfo> history_entries_;
 
   // Path to directory containing test data.
   FilePath test_dir_;
 
   // Temporary directory we will save pages to.
   ScopedTempDir save_dir_;
@@ skipping 160 matching lines @@
         content::NotificationService::AllSources());
   browser()->SavePage();
   observer.Wait();
 
   EXPECT_TRUE(file_util::PathExists(full_file_name));
 
   EXPECT_TRUE(file_util::DieFileDie(full_file_name, false));
   EXPECT_TRUE(file_util::DieFileDie(dir, true));
 }
 #endif
+
+IN_PROC_BROWSER_TEST_F(SavePageBrowserTest, SavePageAsMHTML) {
+  GURL url = NavigateToMockURL("b");
+  FilePath download_dir = DownloadPrefs::FromDownloadManager(
+      GetDownloadManager())->download_path();
+  FilePath full_file_name = download_dir.AppendASCII(std::string(
+      "Test page for saving page feature.mht"));
+  SavePackageFilePicker::SetShouldPromptUser(false);
+  browser()->profile()->GetPrefs()->SetInteger(
+      prefs::kSaveFileType, content::SAVE_PAGE_TYPE_AS_MHTML);
+  ui_test_utils::WindowedNotificationObserver observer(
+        content::NOTIFICATION_SAVE_PACKAGE_SUCCESSFULLY_FINISHED,
+        content::NotificationService::AllSources());
+  browser()->SavePage();
+  observer.Wait();
+  CheckDownloadUI(full_file_name);
+  CheckDownloadHistory(url, full_file_name, 2760);
+
+  EXPECT_TRUE(file_util::PathExists(full_file_name));
+  int64 actual_file_size = -1;
+  EXPECT_TRUE(file_util::GetFileSize(full_file_name, &actual_file_size));
+  EXPECT_EQ(2760, actual_file_size);
+}