Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2)

Side by Side Diff: trunk/src/chrome/browser/download/download_request_limiter_unittest.cc

Issue 23551005: Revert 219709 "Remove the Extensions URLRequestContext." (Closed) Base URL: svn://svn.chromium.org/chrome/
Patch Set: Created 7 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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 "chrome/browser/download/download_request_limiter.h" 5 #include "chrome/browser/download/download_request_limiter.h"
6 6
7 #include "base/bind.h" 7 #include "base/bind.h"
8 #include "base/run_loop.h" 8 #include "base/run_loop.h"
9 #include "chrome/browser/content_settings/host_content_settings_map.h" 9 #include "chrome/browser/content_settings/host_content_settings_map.h"
10 #include "chrome/browser/download/download_request_infobar_delegate.h" 10 #include "chrome/browser/download/download_request_infobar_delegate.h"
11 #include "chrome/browser/infobars/infobar_service.h" 11 #include "chrome/browser/infobars/infobar_service.h"
12 #include "chrome/browser/ui/blocked_content/blocked_content_tab_helper.h" 12 #include "chrome/browser/ui/blocked_content/blocked_content_tab_helper.h"
13 #include "chrome/test/base/chrome_render_view_host_test_harness.h" 13 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
14 #include "chrome/test/base/testing_profile.h" 14 #include "chrome/test/base/testing_profile.h"
15 #include "content/public/browser/navigation_controller.h" 15 #include "content/public/browser/navigation_controller.h"
16 #include "content/public/browser/web_contents.h" 16 #include "content/public/browser/web_contents.h"
17 #include "content/public/test/test_browser_thread_bundle.h"
18 #include "testing/gtest/include/gtest/gtest.h" 17 #include "testing/gtest/include/gtest/gtest.h"
19 18
20 using content::WebContents; 19 using content::WebContents;
21 20
22 class DownloadRequestLimiterTest : public ChromeRenderViewHostTestHarness { 21 class DownloadRequestLimiterTest : public ChromeRenderViewHostTestHarness {
23 public: 22 public:
24 enum TestingAction { 23 enum TestingAction {
25 ACCEPT, 24 ACCEPT,
26 CANCEL, 25 CANCEL,
27 WAIT 26 WAIT
28 }; 27 };
29 28
30 virtual void SetUp() { 29 virtual void SetUp() {
31 ChromeRenderViewHostTestHarness::SetUp(); 30 ChromeRenderViewHostTestHarness::SetUp();
32 BlockedContentTabHelper::CreateForWebContents(web_contents()); 31 BlockedContentTabHelper::CreateForWebContents(web_contents());
33 InfoBarService::CreateForWebContents(web_contents()); 32 InfoBarService::CreateForWebContents(web_contents());
34 testing_action_ = ACCEPT; 33 testing_action_ = ACCEPT;
35 ask_allow_count_ = cancel_count_ = continue_count_ = 0; 34 ask_allow_count_ = cancel_count_ = continue_count_ = 0;
36 download_request_limiter_ = new DownloadRequestLimiter(); 35 download_request_limiter_ = new DownloadRequestLimiter();
37 fake_create_callback_ = base::Bind( 36 fake_create_callback_ = base::Bind(
38 &DownloadRequestLimiterTest::FakeCreate, base::Unretained(this)); 37 &DownloadRequestLimiterTest::FakeCreate, base::Unretained(this));
39 DownloadRequestInfoBarDelegate::SetCallbackForTesting( 38 DownloadRequestInfoBarDelegate::SetCallbackForTesting(
40 &fake_create_callback_); 39 &fake_create_callback_);
41 content_settings_ = new HostContentSettingsMap( 40 content_settings_ = new HostContentSettingsMap(profile_.GetPrefs(), false);
42 Profile::FromBrowserContext(browser_context())->GetPrefs(), false);
43 DownloadRequestLimiter::SetContentSettingsForTesting( 41 DownloadRequestLimiter::SetContentSettingsForTesting(
44 content_settings_.get()); 42 content_settings_.get());
45 } 43 }
46 44
47 void FakeCreate( 45 void FakeCreate(
48 InfoBarService* infobar_service, 46 InfoBarService* infobar_service,
49 base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host) { 47 base::WeakPtr<DownloadRequestLimiter::TabDownloadState> host) {
50 ask_allow_count_++; 48 ask_allow_count_++;
51 switch (testing_action_) { 49 switch (testing_action_) {
52 case ACCEPT: 50 case ACCEPT:
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after
134 // Number of times CancelDownload was invoked. 132 // Number of times CancelDownload was invoked.
135 int cancel_count_; 133 int cancel_count_;
136 134
137 // Number of times ShouldAllowDownload was invoked. 135 // Number of times ShouldAllowDownload was invoked.
138 int ask_allow_count_; 136 int ask_allow_count_;
139 137
140 scoped_refptr<HostContentSettingsMap> content_settings_; 138 scoped_refptr<HostContentSettingsMap> content_settings_;
141 139
142 private: 140 private:
143 DownloadRequestInfoBarDelegate::FakeCreateCallback fake_create_callback_; 141 DownloadRequestInfoBarDelegate::FakeCreateCallback fake_create_callback_;
144 virtual content::BrowserContext* CreateBrowserContext() OVERRIDE { 142 TestingProfile profile_;
145 return new TestingProfile();
146 }
147 }; 143 };
148 144
149 TEST_F(DownloadRequestLimiterTest, 145 TEST_F(DownloadRequestLimiterTest,
150 DownloadRequestLimiter_Allow) { 146 DownloadRequestLimiter_Allow) {
151 // All tabs should initially start at ALLOW_ONE_DOWNLOAD. 147 // All tabs should initially start at ALLOW_ONE_DOWNLOAD.
152 ASSERT_EQ(DownloadRequestLimiter::ALLOW_ONE_DOWNLOAD, 148 ASSERT_EQ(DownloadRequestLimiter::ALLOW_ONE_DOWNLOAD,
153 download_request_limiter_->GetDownloadStatus(web_contents())); 149 download_request_limiter_->GetDownloadStatus(web_contents()));
154 150
155 // Ask if the tab can do a download. This moves to PROMPT_BEFORE_DOWNLOAD. 151 // Ask if the tab can do a download. This moves to PROMPT_BEFORE_DOWNLOAD.
156 CanDownload(); 152 CanDownload();
(...skipping 191 matching lines...) Expand 10 before | Expand all | Expand 10 after
348 EXPECT_EQ(DownloadRequestLimiter::DOWNLOADS_NOT_ALLOWED, 344 EXPECT_EQ(DownloadRequestLimiter::DOWNLOADS_NOT_ALLOWED,
349 download_request_limiter_->GetDownloadStatus(web_contents.get())); 345 download_request_limiter_->GetDownloadStatus(web_contents.get()));
350 OnUserGestureFor(web_contents.get()); 346 OnUserGestureFor(web_contents.get());
351 EXPECT_EQ(DownloadRequestLimiter::ALLOW_ONE_DOWNLOAD, 347 EXPECT_EQ(DownloadRequestLimiter::ALLOW_ONE_DOWNLOAD,
352 download_request_limiter_->GetDownloadStatus(web_contents.get())); 348 download_request_limiter_->GetDownloadStatus(web_contents.get()));
353 CanDownloadFor(web_contents.get()); 349 CanDownloadFor(web_contents.get());
354 ExpectAndResetCounts(1, 0, 0, __LINE__); 350 ExpectAndResetCounts(1, 0, 0, __LINE__);
355 EXPECT_EQ(DownloadRequestLimiter::PROMPT_BEFORE_DOWNLOAD, 351 EXPECT_EQ(DownloadRequestLimiter::PROMPT_BEFORE_DOWNLOAD,
356 download_request_limiter_->GetDownloadStatus(web_contents.get())); 352 download_request_limiter_->GetDownloadStatus(web_contents.get()));
357 } 353 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698