diff options
Diffstat (limited to 'chrome/browser/download/download_danger_prompt_browsertest.cc')
| -rw-r--r-- | chrome/browser/download/download_danger_prompt_browsertest.cc | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/chrome/browser/download/download_danger_prompt_browsertest.cc b/chrome/browser/download/download_danger_prompt_browsertest.cc index 731075d..2b2b57c 100644 --- a/chrome/browser/download/download_danger_prompt_browsertest.cc +++ b/chrome/browser/download/download_danger_prompt_browsertest.cc @@ -5,6 +5,7 @@ #include "base/bind.h" #include "base/files/file_path.h" #include "chrome/browser/download/download_danger_prompt.h" +#include "chrome/browser/profiles/profile.h" #include "chrome/browser/ui/browser.h" #include "chrome/browser/ui/browser_commands.h" #include "chrome/browser/ui/browser_tabstrip.h" @@ -14,11 +15,13 @@ #include "content/public/test/mock_download_item.h" #include "testing/gmock/include/gmock/gmock.h" #include "testing/gtest/include/gtest/gtest.h" +#include "url/gurl.h" using ::testing::_; using ::testing::ByRef; using ::testing::Eq; using ::testing::Return; +using ::testing::ReturnRef; using ::testing::SaveArg; class DownloadDangerPromptTest : public InProcessBrowserTest { @@ -100,6 +103,13 @@ class DownloadDangerPromptTest : public InProcessBrowserTest { }; IN_PROC_BROWSER_TEST_F(DownloadDangerPromptTest, TestAll) { + // ExperienceSampling: Set default actions for DownloadItem methods we need. + ON_CALL(download(), GetURL()).WillByDefault(ReturnRef(GURL::EmptyGURL())); + ON_CALL(download(), GetReferrerUrl()) + .WillByDefault(ReturnRef(GURL::EmptyGURL())); + ON_CALL(download(), GetBrowserContext()) + .WillByDefault(Return(browser()->profile())); + OpenNewTab(); // Clicking the Accept button should invoke the ACCEPT action. |