| 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 |