| Index: chrome/browser/download/download_danger_prompt_browsertest.cc
|
| diff --git a/chrome/browser/download/download_danger_prompt_browsertest.cc b/chrome/browser/download/download_danger_prompt_browsertest.cc
|
| index cdd17d02dc81162b3df6f5126b4d6e77c0127d4c..731075d6e9f9cea2116d7c2bea9fb89881e05bf7 100644
|
| --- a/chrome/browser/download/download_danger_prompt_browsertest.cc
|
| +++ b/chrome/browser/download/download_danger_prompt_browsertest.cc
|
| @@ -24,8 +24,7 @@ using ::testing::SaveArg;
|
| class DownloadDangerPromptTest : public InProcessBrowserTest {
|
| public:
|
| DownloadDangerPromptTest()
|
| - : download_observer_(NULL),
|
| - prompt_(NULL),
|
| + : prompt_(NULL),
|
| expected_action_(DownloadDangerPrompt::CANCEL),
|
| did_receive_callback_(false) {
|
| }
|
| @@ -66,19 +65,12 @@ class DownloadDangerPromptTest : public InProcessBrowserTest {
|
|
|
| content::MockDownloadItem& download() { return download_; }
|
|
|
| - content::DownloadItem::Observer* download_observer() {
|
| - return download_observer_;
|
| - }
|
| -
|
| DownloadDangerPrompt* prompt() { return prompt_; }
|
|
|
| private:
|
| void SetUpDownloadItemExpectations() {
|
| EXPECT_CALL(download_, GetFileNameToReportUser()).WillRepeatedly(Return(
|
| base::FilePath(FILE_PATH_LITERAL("evil.exe"))));
|
| - EXPECT_CALL(download_, AddObserver(_))
|
| - .WillOnce(SaveArg<0>(&download_observer_));
|
| - EXPECT_CALL(download_, RemoveObserver(Eq(ByRef(download_observer_))));
|
| EXPECT_CALL(download_, GetDangerType())
|
| .WillRepeatedly(Return(content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL));
|
| }
|
| @@ -100,7 +92,6 @@ class DownloadDangerPromptTest : public InProcessBrowserTest {
|
| }
|
|
|
| content::MockDownloadItem download_;
|
| - content::DownloadItem::Observer* download_observer_;
|
| DownloadDangerPrompt* prompt_;
|
| DownloadDangerPrompt::Action expected_action_;
|
| bool did_receive_callback_;
|
| @@ -125,7 +116,7 @@ IN_PROC_BROWSER_TEST_F(DownloadDangerPromptTest, TestAll) {
|
| // dialog should DISMISS itself.
|
| SetUpExpectations(DownloadDangerPrompt::DISMISS);
|
| EXPECT_CALL(download(), IsDangerous()).WillOnce(Return(false));
|
| - download_observer()->OnDownloadUpdated(&download());
|
| + download().NotifyObserversDownloadUpdated();
|
| VerifyExpectations();
|
|
|
| // If the download is in a terminal state then the dialog should DISMISS
|
| @@ -133,7 +124,7 @@ IN_PROC_BROWSER_TEST_F(DownloadDangerPromptTest, TestAll) {
|
| SetUpExpectations(DownloadDangerPrompt::DISMISS);
|
| EXPECT_CALL(download(), IsDangerous()).WillOnce(Return(true));
|
| EXPECT_CALL(download(), IsDone()).WillOnce(Return(true));
|
| - download_observer()->OnDownloadUpdated(&download());
|
| + download().NotifyObserversDownloadUpdated();
|
| VerifyExpectations();
|
|
|
| // If the download is dangerous and is not in a terminal state, don't dismiss
|
| @@ -141,7 +132,7 @@ IN_PROC_BROWSER_TEST_F(DownloadDangerPromptTest, TestAll) {
|
| SetUpExpectations(DownloadDangerPrompt::ACCEPT);
|
| EXPECT_CALL(download(), IsDangerous()).WillOnce(Return(true));
|
| EXPECT_CALL(download(), IsDone()).WillOnce(Return(false));
|
| - download_observer()->OnDownloadUpdated(&download());
|
| + download().NotifyObserversDownloadUpdated();
|
| SimulatePromptAction(DownloadDangerPrompt::ACCEPT);
|
| VerifyExpectations();
|
|
|
|
|