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 <queue> | 5 #include <queue> |
6 #include <map> | 6 #include <map> |
7 | 7 |
8 #include "base/bind.h" | 8 #include "base/bind.h" |
9 #include "base/callback.h" | 9 #include "base/callback.h" |
10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
(...skipping 10 matching lines...) Expand all Loading... |
21 #include "chrome/browser/extensions/extension_event_router_forwarder.h" | 21 #include "chrome/browser/extensions/extension_event_router_forwarder.h" |
22 #include "chrome/browser/net/chrome_network_delegate.h" | 22 #include "chrome/browser/net/chrome_network_delegate.h" |
23 #include "chrome/browser/prefs/pref_member.h" | 23 #include "chrome/browser/prefs/pref_member.h" |
24 #include "chrome/common/extensions/extension_messages.h" | 24 #include "chrome/common/extensions/extension_messages.h" |
25 #include "chrome/common/pref_names.h" | 25 #include "chrome/common/pref_names.h" |
26 #include "chrome/test/base/testing_browser_process.h" | 26 #include "chrome/test/base/testing_browser_process.h" |
27 #include "chrome/test/base/testing_pref_service.h" | 27 #include "chrome/test/base/testing_pref_service.h" |
28 #include "chrome/test/base/testing_profile.h" | 28 #include "chrome/test/base/testing_profile.h" |
29 #include "content/test/test_browser_thread.h" | 29 #include "content/test/test_browser_thread.h" |
30 #include "net/base/auth.h" | 30 #include "net/base/auth.h" |
| 31 #include "net/base/mock_host_resolver.h" |
31 #include "net/base/net_util.h" | 32 #include "net/base/net_util.h" |
32 #include "net/url_request/url_request_test_util.h" | 33 #include "net/url_request/url_request_test_util.h" |
33 #include "testing/gtest/include/gtest/gtest.h" | 34 #include "testing/gtest/include/gtest/gtest.h" |
34 | 35 |
35 namespace helpers = extension_web_request_api_helpers; | 36 namespace helpers = extension_web_request_api_helpers; |
36 namespace keys = extension_web_request_api_constants; | 37 namespace keys = extension_web_request_api_constants; |
37 | 38 |
38 using helpers::CalculateOnAuthRequiredDelta; | 39 using helpers::CalculateOnAuthRequiredDelta; |
39 using helpers::CalculateOnBeforeRequestDelta; | 40 using helpers::CalculateOnBeforeRequestDelta; |
40 using helpers::CalculateOnBeforeSendHeadersDelta; | 41 using helpers::CalculateOnBeforeSendHeadersDelta; |
(...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
120 io_thread_(content::BrowserThread::IO, &message_loop_) {} | 121 io_thread_(content::BrowserThread::IO, &message_loop_) {} |
121 | 122 |
122 protected: | 123 protected: |
123 virtual void SetUp() OVERRIDE { | 124 virtual void SetUp() OVERRIDE { |
124 event_router_ = new ExtensionEventRouterForwarder(); | 125 event_router_ = new ExtensionEventRouterForwarder(); |
125 enable_referrers_.Init( | 126 enable_referrers_.Init( |
126 prefs::kEnableReferrers, profile_.GetTestingPrefService(), NULL); | 127 prefs::kEnableReferrers, profile_.GetTestingPrefService(), NULL); |
127 network_delegate_.reset(new ChromeNetworkDelegate( | 128 network_delegate_.reset(new ChromeNetworkDelegate( |
128 event_router_.get(), NULL, NULL, &profile_, | 129 event_router_.get(), NULL, NULL, &profile_, |
129 CookieSettings::Factory::GetForProfile(&profile_), &enable_referrers_)); | 130 CookieSettings::Factory::GetForProfile(&profile_), &enable_referrers_)); |
130 context_.reset(new TestURLRequestContext()); | 131 context_.reset(new TestURLRequestContext(true)); |
131 context_->set_network_delegate(network_delegate_.get()); | 132 context_->set_network_delegate(network_delegate_.get()); |
| 133 context_->Init(); |
132 } | 134 } |
133 | 135 |
134 MessageLoopForIO message_loop_; | 136 MessageLoopForIO message_loop_; |
135 content::TestBrowserThread ui_thread_; | 137 content::TestBrowserThread ui_thread_; |
136 content::TestBrowserThread io_thread_; | 138 content::TestBrowserThread io_thread_; |
137 TestingProfile profile_; | 139 TestingProfile profile_; |
138 TestDelegate delegate_; | 140 TestDelegate delegate_; |
139 BooleanPrefMember enable_referrers_; | 141 BooleanPrefMember enable_referrers_; |
140 TestIPCSender ipc_sender_; | 142 TestIPCSender ipc_sender_; |
141 scoped_refptr<ExtensionEventRouterForwarder> event_router_; | 143 scoped_refptr<ExtensionEventRouterForwarder> event_router_; |
(...skipping 295 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
437 io_thread_(content::BrowserThread::IO, &message_loop_) {} | 439 io_thread_(content::BrowserThread::IO, &message_loop_) {} |
438 | 440 |
439 protected: | 441 protected: |
440 virtual void SetUp() { | 442 virtual void SetUp() { |
441 event_router_ = new ExtensionEventRouterForwarder(); | 443 event_router_ = new ExtensionEventRouterForwarder(); |
442 enable_referrers_.Init( | 444 enable_referrers_.Init( |
443 prefs::kEnableReferrers, profile_.GetTestingPrefService(), NULL); | 445 prefs::kEnableReferrers, profile_.GetTestingPrefService(), NULL); |
444 network_delegate_.reset(new ChromeNetworkDelegate( | 446 network_delegate_.reset(new ChromeNetworkDelegate( |
445 event_router_.get(), NULL, NULL, &profile_, | 447 event_router_.get(), NULL, NULL, &profile_, |
446 CookieSettings::Factory::GetForProfile(&profile_), &enable_referrers_)); | 448 CookieSettings::Factory::GetForProfile(&profile_), &enable_referrers_)); |
447 context_.reset(new TestURLRequestContext()); | 449 context_.reset(new TestURLRequestContext(true)); |
| 450 host_resolver_.reset(new net::MockHostResolver()); |
| 451 host_resolver_->rules()->AddSimulatedFailure("doesnotexist"); |
| 452 context_->set_host_resolver(host_resolver_.get()); |
448 context_->set_network_delegate(network_delegate_.get()); | 453 context_->set_network_delegate(network_delegate_.get()); |
| 454 context_->Init(); |
449 } | 455 } |
450 | 456 |
451 MessageLoopForIO message_loop_; | 457 MessageLoopForIO message_loop_; |
452 content::TestBrowserThread ui_thread_; | 458 content::TestBrowserThread ui_thread_; |
453 content::TestBrowserThread io_thread_; | 459 content::TestBrowserThread io_thread_; |
454 TestingProfile profile_; | 460 TestingProfile profile_; |
455 TestDelegate delegate_; | 461 TestDelegate delegate_; |
456 BooleanPrefMember enable_referrers_; | 462 BooleanPrefMember enable_referrers_; |
457 TestIPCSender ipc_sender_; | 463 TestIPCSender ipc_sender_; |
458 scoped_refptr<ExtensionEventRouterForwarder> event_router_; | 464 scoped_refptr<ExtensionEventRouterForwarder> event_router_; |
459 scoped_refptr<ExtensionInfoMap> extension_info_map_; | 465 scoped_refptr<ExtensionInfoMap> extension_info_map_; |
460 scoped_ptr<ChromeNetworkDelegate> network_delegate_; | 466 scoped_ptr<ChromeNetworkDelegate> network_delegate_; |
| 467 scoped_ptr<net::MockHostResolver> host_resolver_; |
461 scoped_ptr<TestURLRequestContext> context_; | 468 scoped_ptr<TestURLRequestContext> context_; |
462 }; | 469 }; |
463 | 470 |
464 TEST_P(ExtensionWebRequestHeaderModificationTest, TestModifications) { | 471 TEST_P(ExtensionWebRequestHeaderModificationTest, TestModifications) { |
465 std::string extension1_id("1"); | 472 std::string extension1_id("1"); |
466 std::string extension2_id("2"); | 473 std::string extension2_id("2"); |
467 std::string extension3_id("3"); | 474 std::string extension3_id("3"); |
468 ExtensionWebRequestEventRouter::RequestFilter filter; | 475 ExtensionWebRequestEventRouter::RequestFilter filter; |
469 const std::string kEventName(keys::kOnBeforeSendHeaders); | 476 const std::string kEventName(keys::kOnBeforeSendHeaders); |
470 base::WeakPtrFactory<TestIPCSender> ipc_sender_factory(&ipc_sender_); | 477 base::WeakPtrFactory<TestIPCSender> ipc_sender_factory(&ipc_sender_); |
(...skipping 1029 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
1500 }; | 1507 }; |
1501 for (size_t i = 0; i < arraysize(sensitive_urls); ++i) { | 1508 for (size_t i = 0; i < arraysize(sensitive_urls); ++i) { |
1502 EXPECT_TRUE(helpers::HideRequestForURL(GURL(sensitive_urls[i]))) | 1509 EXPECT_TRUE(helpers::HideRequestForURL(GURL(sensitive_urls[i]))) |
1503 << sensitive_urls[i]; | 1510 << sensitive_urls[i]; |
1504 } | 1511 } |
1505 for (size_t i = 0; i < arraysize(non_sensitive_urls); ++i) { | 1512 for (size_t i = 0; i < arraysize(non_sensitive_urls); ++i) { |
1506 EXPECT_FALSE(helpers::HideRequestForURL(GURL(non_sensitive_urls[i]))) | 1513 EXPECT_FALSE(helpers::HideRequestForURL(GURL(non_sensitive_urls[i]))) |
1507 << non_sensitive_urls[i]; | 1514 << non_sensitive_urls[i]; |
1508 } | 1515 } |
1509 } | 1516 } |
OLD | NEW |