Index: chrome/browser/ui/download/download_tab_helper.h |
diff --git a/chrome/browser/ui/download/download_tab_helper.h b/chrome/browser/ui/download/download_tab_helper.h |
index ac559136eb7b74d2056acbd2bdd50d8c95cb87db..9c9250988c49c59d51e6541ee97689585486557d 100644 |
--- a/chrome/browser/ui/download/download_tab_helper.h |
+++ b/chrome/browser/ui/download/download_tab_helper.h |
@@ -7,6 +7,7 @@ |
#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" |
@@ -34,6 +35,9 @@ class DownloadTabHelper : public TabContentsObserver { |
// Save page with the main HTML file path, the directory for saving resources, |
// and the save type: HTML only or complete web page. Returns true if the |
// saving process has been initiated successfully. |
+ // This method is used in automated testing to bypass prompting the user for |
+ // file names. Instead, the names and paths are hard coded rather than |
+ // running them through file name sanitation and extension / mime checking. |
bool SavePage(const FilePath& main_file, const FilePath& dir_path, |
SavePackage::SavePackageType save_type); |
@@ -48,6 +52,21 @@ class DownloadTabHelper : public TabContentsObserver { |
void OnStartDownload(DownloadItem* download); |
private: |
+ FRIEND_TEST_ALL_PREFIXES(SavePageBrowserTest, SavedFolder1); |
+ FRIEND_TEST_ALL_PREFIXES(SavePageBrowserTest, SavedFolder2); |
+ FRIEND_TEST_ALL_PREFIXES(SavePageBrowserTest, SavedFolder3); |
+ FRIEND_TEST_ALL_PREFIXES(SavePageBrowserTest, SavedFolder4); |
+ |
+ // Used in automated testing to bypass prompting the user for file names. |
+ // The difference between SavePageBasedOnDefaultPrefs() and SavePage() |
+ // is whether the default folder prefs are used. In case of SavePage(), |
+ // we need to give it the file path to which the file is saved. |
+ // On the other hand, in case of SavePageBasedOnDefaultPrefs(), |
+ // we need not to give the file path since the file path is determined |
+ // based on the default folder prefs. This method returns the title |
+ // of the current tab. |
+ string16 SavePageBasedOnDefaultPrefs(); |
+ |
// TabContentsObserver overrides. |
virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; |
virtual void DidGetUserGesture() OVERRIDE; |