| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/file_path.h" | 6 #include "base/file_path.h" |
| 7 #include "chrome/browser/download/download_danger_prompt.h" | 7 #include "chrome/browser/download/download_danger_prompt.h" |
| 8 #include "chrome/browser/ui/browser.h" | 8 #include "chrome/browser/ui/browser.h" |
| 9 #include "chrome/browser/ui/browser_commands.h" | 9 #include "chrome/browser/ui/browser_commands.h" |
| 10 #include "chrome/browser/ui/browser_tabstrip.h" | 10 #include "chrome/browser/ui/browser_tabstrip.h" |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 45 } | 45 } |
| 46 | 46 |
| 47 void SetUpExpectations(DownloadDangerPrompt::Action expected_action) { | 47 void SetUpExpectations(DownloadDangerPrompt::Action expected_action) { |
| 48 did_receive_callback_ = false; | 48 did_receive_callback_ = false; |
| 49 expected_action_ = expected_action; | 49 expected_action_ = expected_action; |
| 50 SetUpDownloadItemExpectations(); | 50 SetUpDownloadItemExpectations(); |
| 51 CreatePrompt(); | 51 CreatePrompt(); |
| 52 } | 52 } |
| 53 | 53 |
| 54 void VerifyExpectations() { | 54 void VerifyExpectations() { |
| 55 ui_test_utils::RunAllPendingInMessageLoop(); | 55 content::RunAllPendingInMessageLoop(); |
| 56 // At the end of each test, we expect no more activity from the prompt. The | 56 // At the end of each test, we expect no more activity from the prompt. The |
| 57 // prompt shouldn't exist anymore either. | 57 // prompt shouldn't exist anymore either. |
| 58 EXPECT_TRUE(did_receive_callback_); | 58 EXPECT_TRUE(did_receive_callback_); |
| 59 EXPECT_FALSE(prompt_); | 59 EXPECT_FALSE(prompt_); |
| 60 testing::Mock::VerifyAndClearExpectations(&download_); | 60 testing::Mock::VerifyAndClearExpectations(&download_); |
| 61 } | 61 } |
| 62 | 62 |
| 63 void SimulatePromptAction(DownloadDangerPrompt::Action action) { | 63 void SimulatePromptAction(DownloadDangerPrompt::Action action) { |
| 64 prompt_->InvokeActionForTesting(action); | 64 prompt_->InvokeActionForTesting(action); |
| 65 } | 65 } |
| (...skipping 16 matching lines...) Expand all Loading... |
| 82 } | 82 } |
| 83 | 83 |
| 84 void CreatePrompt() { | 84 void CreatePrompt() { |
| 85 prompt_ = DownloadDangerPrompt::Create( | 85 prompt_ = DownloadDangerPrompt::Create( |
| 86 &download_, | 86 &download_, |
| 87 chrome::GetActiveTabContents(browser()), | 87 chrome::GetActiveTabContents(browser()), |
| 88 base::Bind(&DownloadDangerPromptTest::PromptCallback, this, | 88 base::Bind(&DownloadDangerPromptTest::PromptCallback, this, |
| 89 DownloadDangerPrompt::ACCEPT), | 89 DownloadDangerPrompt::ACCEPT), |
| 90 base::Bind(&DownloadDangerPromptTest::PromptCallback, this, | 90 base::Bind(&DownloadDangerPromptTest::PromptCallback, this, |
| 91 DownloadDangerPrompt::CANCEL)); | 91 DownloadDangerPrompt::CANCEL)); |
| 92 ui_test_utils::RunAllPendingInMessageLoop(); | 92 content::RunAllPendingInMessageLoop(); |
| 93 } | 93 } |
| 94 | 94 |
| 95 void PromptCallback(DownloadDangerPrompt::Action action) { | 95 void PromptCallback(DownloadDangerPrompt::Action action) { |
| 96 EXPECT_FALSE(did_receive_callback_); | 96 EXPECT_FALSE(did_receive_callback_); |
| 97 EXPECT_EQ(expected_action_, action); | 97 EXPECT_EQ(expected_action_, action); |
| 98 did_receive_callback_ = true; | 98 did_receive_callback_ = true; |
| 99 prompt_ = NULL; | 99 prompt_ = NULL; |
| 100 } | 100 } |
| 101 | 101 |
| 102 content::MockDownloadItem download_; | 102 content::MockDownloadItem download_; |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 134 EXPECT_CALL(download(), IsDangerous()).WillOnce(Return(false)); | 134 EXPECT_CALL(download(), IsDangerous()).WillOnce(Return(false)); |
| 135 download_observer()->OnDownloadUpdated(&download()); | 135 download_observer()->OnDownloadUpdated(&download()); |
| 136 VerifyExpectations(); | 136 VerifyExpectations(); |
| 137 | 137 |
| 138 // If the containing tab is closed, the dialog should be canceled. | 138 // If the containing tab is closed, the dialog should be canceled. |
| 139 OpenNewTab(); | 139 OpenNewTab(); |
| 140 SetUpExpectations(DownloadDangerPrompt::CANCEL); | 140 SetUpExpectations(DownloadDangerPrompt::CANCEL); |
| 141 chrome::CloseTab(browser()); | 141 chrome::CloseTab(browser()); |
| 142 VerifyExpectations(); | 142 VerifyExpectations(); |
| 143 } | 143 } |
| OLD | NEW |