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/renderer/content_settings_observer.h" | 5 #include "chrome/renderer/content_settings_observer.h" |
6 | 6 |
7 #include "chrome/common/url_constants.h" | 7 #include "chrome/common/url_constants.h" |
8 #include "content/public/common/url_constants.h" | 8 #include "content/public/common/url_constants.h" |
9 #include "extensions/common/constants.h" | |
10 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
11 #include "third_party/WebKit/public/platform/WebURL.h" | 10 #include "third_party/WebKit/public/platform/WebURL.h" |
12 #include "third_party/WebKit/public/web/WebSecurityOrigin.h" | 11 #include "third_party/WebKit/public/web/WebSecurityOrigin.h" |
13 #include "url/gurl.h" | 12 #include "url/gurl.h" |
14 | 13 |
| 14 #if defined(ENABLE_EXTENSIONS) |
| 15 #include "extensions/common/constants.h" |
| 16 #endif |
| 17 |
15 using blink::WebSecurityOrigin; | 18 using blink::WebSecurityOrigin; |
16 | 19 |
17 typedef testing::Test ContentSettingsObserverTest; | 20 typedef testing::Test ContentSettingsObserverTest; |
18 | 21 |
19 TEST_F(ContentSettingsObserverTest, WhitelistedSchemes) { | 22 TEST_F(ContentSettingsObserverTest, WhitelistedSchemes) { |
20 std::string end_url = ":something"; | 23 std::string end_url = ":something"; |
21 | 24 |
22 GURL chrome_ui_url = | 25 GURL chrome_ui_url = |
23 GURL(std::string(content::kChromeUIScheme).append(end_url)); | 26 GURL(std::string(content::kChromeUIScheme).append(end_url)); |
24 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( | 27 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( |
25 WebSecurityOrigin::create(chrome_ui_url), | 28 WebSecurityOrigin::create(chrome_ui_url), |
26 GURL())); | 29 GURL())); |
27 | 30 |
28 GURL chrome_dev_tools_url = | 31 GURL chrome_dev_tools_url = |
29 GURL(std::string(content::kChromeDevToolsScheme).append(end_url)); | 32 GURL(std::string(content::kChromeDevToolsScheme).append(end_url)); |
30 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( | 33 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( |
31 WebSecurityOrigin::create(chrome_dev_tools_url), | 34 WebSecurityOrigin::create(chrome_dev_tools_url), |
32 GURL())); | 35 GURL())); |
33 | 36 |
| 37 #if defined(ENABLE_EXTENSIONS) |
34 GURL extension_url = | 38 GURL extension_url = |
35 GURL(std::string(extensions::kExtensionScheme).append(end_url)); | 39 GURL(std::string(extensions::kExtensionScheme).append(end_url)); |
36 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( | 40 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( |
37 WebSecurityOrigin::create(extension_url), | 41 WebSecurityOrigin::create(extension_url), |
38 GURL())); | 42 GURL())); |
| 43 #endif |
39 | 44 |
40 GURL file_url("file:///dir/"); | 45 GURL file_url("file:///dir/"); |
41 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( | 46 EXPECT_TRUE(ContentSettingsObserver::IsWhitelistedForContentSettings( |
42 WebSecurityOrigin::create(file_url), | 47 WebSecurityOrigin::create(file_url), |
43 GURL("file:///dir/"))); | 48 GURL("file:///dir/"))); |
44 EXPECT_FALSE(ContentSettingsObserver::IsWhitelistedForContentSettings( | 49 EXPECT_FALSE(ContentSettingsObserver::IsWhitelistedForContentSettings( |
45 WebSecurityOrigin::create(file_url), | 50 WebSecurityOrigin::create(file_url), |
46 GURL("file:///dir/file"))); | 51 GURL("file:///dir/file"))); |
47 | 52 |
48 GURL http_url = | 53 GURL http_url = |
49 GURL("http://server.com/path"); | 54 GURL("http://server.com/path"); |
50 EXPECT_FALSE(ContentSettingsObserver::IsWhitelistedForContentSettings( | 55 EXPECT_FALSE(ContentSettingsObserver::IsWhitelistedForContentSettings( |
51 WebSecurityOrigin::create(http_url), | 56 WebSecurityOrigin::create(http_url), |
52 GURL())); | 57 GURL())); |
53 } | 58 } |
OLD | NEW |