| 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/browsing_data_helper.h" | 5 #include "chrome/browser/browsing_data_helper.h" |
| 6 | 6 |
| 7 #include "base/stringprintf.h" | 7 #include "base/stringprintf.h" |
| 8 #include "chrome/common/url_constants.h" | 8 #include "chrome/common/url_constants.h" |
| 9 #include "content/public/common/url_constants.h" | 9 #include "content/public/common/url_constants.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 #include "googleurl/src/gurl.h" | 11 #include "googleurl/src/gurl.h" |
| 12 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h" | 12 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h" |
| 13 | 13 |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 class BrowsingDataHelperTest : public testing::Test { | 16 class BrowsingDataHelperTest : public testing::Test { |
| 17 public: | 17 public: |
| 18 BrowsingDataHelperTest() {} | 18 BrowsingDataHelperTest() {} |
| 19 virtual ~BrowsingDataHelperTest() {} | 19 virtual ~BrowsingDataHelperTest() {} |
| 20 | 20 |
| 21 bool IsValidScheme(const std::string& scheme) { | 21 bool IsWebScheme(const std::string& scheme) { |
| 22 GURL test(scheme + "://example.com"); | 22 GURL test(scheme + "://example.com"); |
| 23 return (BrowsingDataHelper::HasValidScheme(test) && | 23 return (BrowsingDataHelper::HasWebScheme(test) && |
| 24 BrowsingDataHelper::IsValidScheme(scheme) && | 24 BrowsingDataHelper::IsWebScheme(scheme) && |
| 25 BrowsingDataHelper::IsValidScheme( | 25 BrowsingDataHelper::IsWebScheme( |
| 26 WebKit::WebString::fromUTF8(scheme))); | 26 WebKit::WebString::fromUTF8(scheme))); |
| 27 } | 27 } |
| 28 | 28 |
| 29 bool IsExtensionScheme(const std::string& scheme) { |
| 30 GURL test(scheme + "://example.com"); |
| 31 return (BrowsingDataHelper::HasExtensionScheme(test) && |
| 32 BrowsingDataHelper::IsExtensionScheme(scheme) && |
| 33 BrowsingDataHelper::IsExtensionScheme( |
| 34 WebKit::WebString::fromUTF8(scheme))); |
| 35 } |
| 36 |
| 29 private: | 37 private: |
| 30 DISALLOW_COPY_AND_ASSIGN(BrowsingDataHelperTest); | 38 DISALLOW_COPY_AND_ASSIGN(BrowsingDataHelperTest); |
| 31 }; | 39 }; |
| 32 | 40 |
| 33 TEST_F(BrowsingDataHelperTest, WebSafeSchemesAreValid) { | 41 TEST_F(BrowsingDataHelperTest, WebSafeSchemesAreWebSafe) { |
| 34 EXPECT_TRUE(IsValidScheme(chrome::kHttpScheme)); | 42 EXPECT_TRUE(IsWebScheme(chrome::kHttpScheme)); |
| 35 EXPECT_TRUE(IsValidScheme(chrome::kHttpsScheme)); | 43 EXPECT_TRUE(IsWebScheme(chrome::kHttpsScheme)); |
| 36 EXPECT_TRUE(IsValidScheme(chrome::kFtpScheme)); | 44 EXPECT_TRUE(IsWebScheme(chrome::kFtpScheme)); |
| 37 EXPECT_TRUE(IsValidScheme(chrome::kDataScheme)); | 45 EXPECT_TRUE(IsWebScheme(chrome::kDataScheme)); |
| 38 EXPECT_TRUE(IsValidScheme("feed")); | 46 EXPECT_TRUE(IsWebScheme("feed")); |
| 39 EXPECT_TRUE(IsValidScheme(chrome::kBlobScheme)); | 47 EXPECT_TRUE(IsWebScheme(chrome::kBlobScheme)); |
| 40 EXPECT_TRUE(IsValidScheme(chrome::kFileSystemScheme)); | 48 EXPECT_TRUE(IsWebScheme(chrome::kFileSystemScheme)); |
| 41 | 49 EXPECT_FALSE(IsWebScheme("invalid-scheme-i-just-made-up")); |
| 42 EXPECT_FALSE(IsValidScheme("invalid-scheme-i-just-made-up")); | |
| 43 } | 50 } |
| 44 | 51 |
| 45 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreInvalid) { | 52 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreNotWebSafe) { |
| 46 EXPECT_FALSE(IsValidScheme(chrome::kExtensionScheme)); | 53 EXPECT_FALSE(IsWebScheme(chrome::kExtensionScheme)); |
| 47 EXPECT_FALSE(IsValidScheme(chrome::kAboutScheme)); | 54 EXPECT_FALSE(IsWebScheme(chrome::kAboutScheme)); |
| 48 EXPECT_FALSE(IsValidScheme(chrome::kChromeDevToolsScheme)); | 55 EXPECT_FALSE(IsWebScheme(chrome::kChromeDevToolsScheme)); |
| 49 EXPECT_FALSE(IsValidScheme(chrome::kChromeInternalScheme)); | 56 EXPECT_FALSE(IsWebScheme(chrome::kChromeInternalScheme)); |
| 50 EXPECT_FALSE(IsValidScheme(chrome::kChromeUIScheme)); | 57 EXPECT_FALSE(IsWebScheme(chrome::kChromeUIScheme)); |
| 51 EXPECT_FALSE(IsValidScheme(chrome::kJavaScriptScheme)); | 58 EXPECT_FALSE(IsWebScheme(chrome::kJavaScriptScheme)); |
| 52 EXPECT_FALSE(IsValidScheme(chrome::kMailToScheme)); | 59 EXPECT_FALSE(IsWebScheme(chrome::kMailToScheme)); |
| 53 EXPECT_FALSE(IsValidScheme(chrome::kMetadataScheme)); | 60 EXPECT_FALSE(IsWebScheme(chrome::kMetadataScheme)); |
| 54 EXPECT_FALSE(IsValidScheme(chrome::kSwappedOutScheme)); | 61 EXPECT_FALSE(IsWebScheme(chrome::kSwappedOutScheme)); |
| 55 EXPECT_FALSE(IsValidScheme(chrome::kViewSourceScheme)); | 62 EXPECT_FALSE(IsWebScheme(chrome::kViewSourceScheme)); |
| 63 } |
| 64 |
| 65 TEST_F(BrowsingDataHelperTest, WebSafeSchemesAreNotExtensions) { |
| 66 EXPECT_FALSE(IsExtensionScheme(chrome::kHttpScheme)); |
| 67 EXPECT_FALSE(IsExtensionScheme(chrome::kHttpsScheme)); |
| 68 EXPECT_FALSE(IsExtensionScheme(chrome::kFtpScheme)); |
| 69 EXPECT_FALSE(IsExtensionScheme(chrome::kDataScheme)); |
| 70 EXPECT_FALSE(IsExtensionScheme("feed")); |
| 71 EXPECT_FALSE(IsExtensionScheme(chrome::kBlobScheme)); |
| 72 EXPECT_FALSE(IsExtensionScheme(chrome::kFileSystemScheme)); |
| 73 EXPECT_FALSE(IsExtensionScheme("invalid-scheme-i-just-made-up")); |
| 74 } |
| 75 |
| 76 TEST_F(BrowsingDataHelperTest, ChromeSchemesAreNotAllExtension) { |
| 77 EXPECT_TRUE(IsExtensionScheme(chrome::kExtensionScheme)); |
| 78 |
| 79 EXPECT_FALSE(IsExtensionScheme(chrome::kAboutScheme)); |
| 80 EXPECT_FALSE(IsExtensionScheme(chrome::kChromeDevToolsScheme)); |
| 81 EXPECT_FALSE(IsExtensionScheme(chrome::kChromeInternalScheme)); |
| 82 EXPECT_FALSE(IsExtensionScheme(chrome::kChromeUIScheme)); |
| 83 EXPECT_FALSE(IsExtensionScheme(chrome::kJavaScriptScheme)); |
| 84 EXPECT_FALSE(IsExtensionScheme(chrome::kMailToScheme)); |
| 85 EXPECT_FALSE(IsExtensionScheme(chrome::kMetadataScheme)); |
| 86 EXPECT_FALSE(IsExtensionScheme(chrome::kSwappedOutScheme)); |
| 87 EXPECT_FALSE(IsExtensionScheme(chrome::kViewSourceScheme)); |
| 56 } | 88 } |
| 57 | 89 |
| 58 } // namespace | 90 } // namespace |
| OLD | NEW |