When the download folder does not exist, change the download folder to a user's "Downloads"

chrome/browser/ui/download/download_tab_helper.h

 // Copyright (c) 2011 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.
 
 #ifndef CHROME_BROWSER_UI_DOWNLOAD_DOWNLOAD_TAB_HELPER_H_
 #define CHROME_BROWSER_UI_DOWNLOAD_DOWNLOAD_TAB_HELPER_H_
 #pragma once
 
 #include "base/basictypes.h"
+#include "base/gtest_prod_util.h"
 #include "chrome/browser/download/save_package.h"
 #include "content/browser/tab_contents/tab_contents_observer.h"
 
 class DownloadItem;
 class DownloadTabHelperDelegate;
 class TabContentsWrapper;
 
 // Per-tab download controller. Handles dealing with various per-tab download
 // duties.
 class DownloadTabHelper : public TabContentsObserver {
@@ skipping 20 matching lines @@
   // Returns the SavePackage which manages the page saving job. May be NULL.
   SavePackage* save_package() const { return save_package_.get(); }
 
   // Notifies the delegate that a download is about to be started.
   // This notification is fired before a local temporary file has been created.
   bool CanDownload(int request_id);
 
   // Notifies the delegate that a download started.
   void OnStartDownload(DownloadItem* download);
 
+  // Used in automated testing to bypass prompting the user for file names.
+  // If we want to change the default folder prefs used for tests,
+  // call ChangeSaveDirectoryPrefs() before this method
+  // and call RestoreSaveDirectoryPrefs() after this method.
+  // Returns the title of the current tab.
+  string16 SavePageWithoutDialog();

[Randy Smith (Not in Mondays), 2011/06/02 19:13:57]

I still don't see a comment describing the difference in functionality between
this and SavePage...?

[haraken1, 2011/06/03 06:50:27]

I wrote the difference here and download_tab_helper.cc (about implementation
details). Also, I changed the method name to SavePageBasedOnDefaultPrefs(). 

+
+  // Changes the default folder prefs. This method saves the current folder
+  // for saving HTML, the current folder for saving downloaded files,
+  // the current user's "Downloads" folder and a save type (HTML only or
+  // complete HTML files), and then changes them to |website_save_dir|,
+  // |download_save_dir|, |default_downloads_dir| and |save_type|, respectively.
+  // If we call this method, we must call RestoreSaveDirectoryPrefs()
+  // after the test to restore the default folder prefs.
+  void ChangeSaveDirectoryPrefs(
+      const FilePath& website_save_dir,
+      const FilePath& download_save_dir,
+      const FilePath& default_downloads_dir,
+      const SavePackage::SavePackageType save_type);
+
+  // Restores the default folder prefs.
+  void RestoreSaveDirectoryPrefs();
+
  private:
   // TabContentsObserver overrides.
   virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
   virtual void DidGetUserGesture() OVERRIDE;
 
   // SavePackage, lazily created.
   scoped_refptr<SavePackage> save_package_;
 
   // Owning TabContentsWrapper.
   TabContentsWrapper* tab_contents_wrapper_;
 
   // Delegate for notifying our owner (usually Browser) about stuff. Not owned
   // by us.
   DownloadTabHelperDelegate* delegate_;
 
+  // Temporarily stores the default folder prefs.
+  FilePath prev_website_save_dir_;
+  FilePath prev_download_save_dir_;
+  FilePath prev_default_downloads_dir_;
+  SavePackage::SavePackageType prev_save_type_;
+
   DISALLOW_COPY_AND_ASSIGN(DownloadTabHelper);
 };
 
 #endif  // CHROME_BROWSER_UI_DOWNLOAD_DOWNLOAD_TAB_HELPER_H_