| 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 "chrome/browser/extensions/api/web_request/web_request_permissions.h" | 5 #include "chrome/browser/extensions/api/web_request/web_request_permissions.h" |
| 6 | 6 |
| 7 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
| 8 #include "chrome/browser/extensions/extension_info_map.h" | |
| 9 #include "chrome/common/extensions/extension_constants.h" | 8 #include "chrome/common/extensions/extension_constants.h" |
| 10 #include "chrome/common/extensions/extension_test_util.h" | 9 #include "chrome/common/extensions/extension_test_util.h" |
| 11 #include "chrome/test/base/testing_profile.h" | 10 #include "chrome/test/base/testing_profile.h" |
| 12 #include "content/public/browser/resource_request_info.h" | 11 #include "content/public/browser/resource_request_info.h" |
| 13 #include "content/public/test/test_browser_thread_bundle.h" | 12 #include "content/public/test/test_browser_thread_bundle.h" |
| 13 #include "extensions/browser/info_map.h" |
| 14 #include "net/base/request_priority.h" | 14 #include "net/base/request_priority.h" |
| 15 #include "net/url_request/url_request_test_util.h" | 15 #include "net/url_request/url_request_test_util.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 using content::ResourceRequestInfo; | 18 using content::ResourceRequestInfo; |
| 19 using extensions::Extension; | 19 using extensions::Extension; |
| 20 using extensions::Manifest; | 20 using extensions::Manifest; |
| 21 using extension_test_util::LoadManifestUnchecked; | 21 using extension_test_util::LoadManifestUnchecked; |
| 22 | 22 |
| 23 class ExtensionWebRequestHelpersTestWithThreadsTest : public testing::Test { | 23 class ExtensionWebRequestHelpersTestWithThreadsTest : public testing::Test { |
| 24 public: | 24 public: |
| 25 ExtensionWebRequestHelpersTestWithThreadsTest() | 25 ExtensionWebRequestHelpersTestWithThreadsTest() |
| 26 : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {} | 26 : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {} |
| 27 | 27 |
| 28 protected: | 28 protected: |
| 29 virtual void SetUp() OVERRIDE; | 29 virtual void SetUp() OVERRIDE; |
| 30 | 30 |
| 31 protected: | 31 protected: |
| 32 net::TestURLRequestContext context; | 32 net::TestURLRequestContext context; |
| 33 | 33 |
| 34 // This extension has Web Request permissions, but no host permission. | 34 // This extension has Web Request permissions, but no host permission. |
| 35 scoped_refptr<Extension> permissionless_extension_; | 35 scoped_refptr<Extension> permissionless_extension_; |
| 36 // This extension has Web Request permissions, and *.com a host permission. | 36 // This extension has Web Request permissions, and *.com a host permission. |
| 37 scoped_refptr<Extension> com_extension_; | 37 scoped_refptr<Extension> com_extension_; |
| 38 scoped_refptr<ExtensionInfoMap> extension_info_map_; | 38 scoped_refptr<extensions::InfoMap> extension_info_map_; |
| 39 | 39 |
| 40 private: | 40 private: |
| 41 content::TestBrowserThreadBundle thread_bundle_; | 41 content::TestBrowserThreadBundle thread_bundle_; |
| 42 }; | 42 }; |
| 43 | 43 |
| 44 void ExtensionWebRequestHelpersTestWithThreadsTest::SetUp() { | 44 void ExtensionWebRequestHelpersTestWithThreadsTest::SetUp() { |
| 45 testing::Test::SetUp(); | 45 testing::Test::SetUp(); |
| 46 | 46 |
| 47 std::string error; | 47 std::string error; |
| 48 permissionless_extension_ = LoadManifestUnchecked("permissions", | 48 permissionless_extension_ = LoadManifestUnchecked("permissions", |
| 49 "web_request_no_host.json", | 49 "web_request_no_host.json", |
| 50 Manifest::INVALID_LOCATION, | 50 Manifest::INVALID_LOCATION, |
| 51 Extension::NO_FLAGS, | 51 Extension::NO_FLAGS, |
| 52 "ext_id_1", | 52 "ext_id_1", |
| 53 &error); | 53 &error); |
| 54 ASSERT_TRUE(permissionless_extension_.get()) << error; | 54 ASSERT_TRUE(permissionless_extension_.get()) << error; |
| 55 com_extension_ = | 55 com_extension_ = |
| 56 LoadManifestUnchecked("permissions", | 56 LoadManifestUnchecked("permissions", |
| 57 "web_request_com_host_permissions.json", | 57 "web_request_com_host_permissions.json", |
| 58 Manifest::INVALID_LOCATION, | 58 Manifest::INVALID_LOCATION, |
| 59 Extension::NO_FLAGS, | 59 Extension::NO_FLAGS, |
| 60 "ext_id_2", | 60 "ext_id_2", |
| 61 &error); | 61 &error); |
| 62 ASSERT_TRUE(com_extension_.get()) << error; | 62 ASSERT_TRUE(com_extension_.get()) << error; |
| 63 extension_info_map_ = new ExtensionInfoMap; | 63 extension_info_map_ = new extensions::InfoMap; |
| 64 extension_info_map_->AddExtension(permissionless_extension_.get(), | 64 extension_info_map_->AddExtension(permissionless_extension_.get(), |
| 65 base::Time::Now(), | 65 base::Time::Now(), |
| 66 false /*incognito_enabled*/); | 66 false /*incognito_enabled*/); |
| 67 extension_info_map_->AddExtension( | 67 extension_info_map_->AddExtension( |
| 68 com_extension_.get(), base::Time::Now(), false /*incognito_enabled*/); | 68 com_extension_.get(), base::Time::Now(), false /*incognito_enabled*/); |
| 69 } | 69 } |
| 70 | 70 |
| 71 TEST_F(ExtensionWebRequestHelpersTestWithThreadsTest, TestHideRequestForURL) { | 71 TEST_F(ExtensionWebRequestHelpersTestWithThreadsTest, TestHideRequestForURL) { |
| 72 net::TestURLRequestContext context; | 72 net::TestURLRequestContext context; |
| 73 const char* sensitive_urls[] = { | 73 const char* sensitive_urls[] = { |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 167 request.url(), | 167 request.url(), |
| 168 false /*crosses_incognito*/, | 168 false /*crosses_incognito*/, |
| 169 WebRequestPermissions::REQUIRE_HOST_PERMISSION)); | 169 WebRequestPermissions::REQUIRE_HOST_PERMISSION)); |
| 170 EXPECT_FALSE(WebRequestPermissions::CanExtensionAccessURL( | 170 EXPECT_FALSE(WebRequestPermissions::CanExtensionAccessURL( |
| 171 extension_info_map_.get(), | 171 extension_info_map_.get(), |
| 172 com_extension_->id(), | 172 com_extension_->id(), |
| 173 request.url(), | 173 request.url(), |
| 174 false /*crosses_incognito*/, | 174 false /*crosses_incognito*/, |
| 175 WebRequestPermissions::REQUIRE_ALL_URLS)); | 175 WebRequestPermissions::REQUIRE_ALL_URLS)); |
| 176 } | 176 } |
| OLD | NEW |