src/chrome/common/content_settings_pattern_parser_unittest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/common/content_settings_pattern.h"
   6 #include "chrome/common/content_settings_pattern_parser.h"
   7
   8 #include "testing/gmock/include/gmock/gmock.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 namespace {
  12 typedef ContentSettingsPattern::BuilderInterface BuilderInterface;
  13 }  // namespace
  14
  15 class MockBuilder : public ContentSettingsPattern::BuilderInterface {
  16  public:
  17   MOCK_METHOD0(WithSchemeWildcard, BuilderInterface*());
  18   MOCK_METHOD0(WithDomainWildcard, BuilderInterface*());
  19   MOCK_METHOD0(WithPortWildcard, BuilderInterface*());
  20   MOCK_METHOD1(WithScheme, BuilderInterface*(const std::string& scheme));
  21   MOCK_METHOD1(WithHost, BuilderInterface*(const std::string& host));
  22   MOCK_METHOD1(WithPort, BuilderInterface*(const std::string& port));
  23   MOCK_METHOD1(WithPath, BuilderInterface*(const std::string& path));
  24   MOCK_METHOD0(WithPathWildcard, BuilderInterface*());
  25   MOCK_METHOD0(Invalid, BuilderInterface*());
  26   MOCK_METHOD0(Build, ContentSettingsPattern());
  27 };
  28
  29 TEST(ContentSettingsPatternParserTest, ParsePatterns) {
  30   // Test valid patterns
  31   ::testing::StrictMock<MockBuilder> builder;
  32
  33   // WithPathWildcard() is not called for "*". (Need a strict Mock for this
  34   // case.)
  35   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
  36       ::testing::Return(&builder));
  37   EXPECT_CALL(builder, WithDomainWildcard()).Times(1).WillOnce(
  38       ::testing::Return(&builder));
  39   EXPECT_CALL(builder, WithPortWildcard()).Times(1).WillOnce(
  40       ::testing::Return(&builder));
  41   content_settings::PatternParser::Parse("*", &builder);
  42   ::testing::Mock::VerifyAndClear(&builder);
  43
  44   EXPECT_CALL(builder, WithScheme("http")).Times(1).WillOnce(
  45       ::testing::Return(&builder));
  46   EXPECT_CALL(builder, WithHost("www.youtube.com")).Times(1).WillOnce(
  47       ::testing::Return(&builder));
  48   EXPECT_CALL(builder, WithPort("8080")).Times(1).WillOnce(
  49       ::testing::Return(&builder));
  50   content_settings::PatternParser::Parse(
  51       "http://www.youtube.com:8080", &builder);
  52   ::testing::Mock::VerifyAndClear(&builder);
  53
  54   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
  55       ::testing::Return(&builder));
  56   EXPECT_CALL(builder, WithHost("www.gmail.com")).Times(1).WillOnce(
  57       ::testing::Return(&builder));
  58   EXPECT_CALL(builder, WithPort("80")).Times(1).WillOnce(
  59       ::testing::Return(&builder));
  60   content_settings::PatternParser::Parse("*://www.gmail.com:80", &builder);
  61   ::testing::Mock::VerifyAndClear(&builder);
  62
  63   EXPECT_CALL(builder, WithScheme("http")).Times(1).WillOnce(
  64       ::testing::Return(&builder));
  65   EXPECT_CALL(builder, WithHost("www.gmail.com")).Times(1).WillOnce(
  66       ::testing::Return(&builder));
  67   EXPECT_CALL(builder, WithPortWildcard()).Times(1).WillOnce(
  68       ::testing::Return(&builder));
  69   content_settings::PatternParser::Parse("http://www.gmail.com:*", &builder);
  70   ::testing::Mock::VerifyAndClear(&builder);
  71
  72   EXPECT_CALL(builder, WithScheme("http")).Times(1).WillOnce(
  73       ::testing::Return(&builder));
  74   EXPECT_CALL(builder, WithDomainWildcard()).WillOnce(
  75       ::testing::Return(&builder));
  76   EXPECT_CALL(builder, WithHost("google.com")).Times(1).WillOnce(
  77       ::testing::Return(&builder));
  78   EXPECT_CALL(builder, WithPort("80")).Times(1).WillOnce(
  79       ::testing::Return(&builder));
  80   content_settings::PatternParser::Parse("http://[*.]google.com:80", &builder);
  81   ::testing::Mock::VerifyAndClear(&builder);
  82
  83   EXPECT_CALL(builder, WithScheme("https")).Times(1).WillOnce(
  84       ::testing::Return(&builder));
  85   EXPECT_CALL(builder, WithHost("[::1]")).Times(1).WillOnce(
  86       ::testing::Return(&builder));
  87   EXPECT_CALL(builder, WithPort("8080")).Times(1).WillOnce(
  88       ::testing::Return(&builder));
  89   content_settings::PatternParser::Parse("https://[::1]:8080", &builder);
  90   ::testing::Mock::VerifyAndClear(&builder);
  91
  92   EXPECT_CALL(builder, WithScheme("http")).Times(1).WillOnce(
  93       ::testing::Return(&builder));
  94   EXPECT_CALL(builder, WithHost("127.0.0.1")).Times(1).WillOnce(
  95       ::testing::Return(&builder));
  96   EXPECT_CALL(builder, WithPort("8080")).Times(1).WillOnce(
  97       ::testing::Return(&builder));
  98   content_settings::PatternParser::Parse("http://127.0.0.1:8080", &builder);
  99   ::testing::Mock::VerifyAndClear(&builder);
 100
 101   // Test valid pattern short forms
 102   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
 103       ::testing::Return(&builder));
 104   EXPECT_CALL(builder, WithHost("www.youtube.com")).Times(1).WillOnce(
 105       ::testing::Return(&builder));
 106   EXPECT_CALL(builder, WithPort("8080")).Times(1).WillOnce(
 107       ::testing::Return(&builder));
 108   content_settings::PatternParser::Parse("www.youtube.com:8080", &builder);
 109   ::testing::Mock::VerifyAndClear(&builder);
 110
 111   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
 112       ::testing::Return(&builder));
 113   EXPECT_CALL(builder, WithHost("www.youtube.com")).Times(1).WillOnce(
 114       ::testing::Return(&builder));
 115   EXPECT_CALL(builder, WithPortWildcard()).Times(1).WillOnce(
 116       ::testing::Return(&builder));
 117   content_settings::PatternParser::Parse("www.youtube.com", &builder);
 118   ::testing::Mock::VerifyAndClear(&builder);
 119
 120   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
 121       ::testing::Return(&builder));
 122   EXPECT_CALL(builder, WithDomainWildcard()).Times(1).WillOnce(
 123       ::testing::Return(&builder));
 124   EXPECT_CALL(builder, WithHost("youtube.com")).Times(1).WillOnce(
 125       ::testing::Return(&builder));
 126   EXPECT_CALL(builder, WithPortWildcard()).Times(1).WillOnce(
 127       ::testing::Return(&builder));
 128   content_settings::PatternParser::Parse("[*.]youtube.com", &builder);
 129   ::testing::Mock::VerifyAndClear(&builder);
 130
 131   // Test invalid patterns
 132   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
 133       ::testing::Return(&builder));
 134   EXPECT_CALL(builder, Invalid()).Times(1).WillOnce(
 135       ::testing::Return(&builder));
 136   content_settings::PatternParser::Parse("*youtube.com", &builder);
 137   ::testing::Mock::VerifyAndClear(&builder);
 138
 139   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
 140       ::testing::Return(&builder));
 141   EXPECT_CALL(builder, Invalid()).Times(1).WillOnce(
 142       ::testing::Return(&builder));
 143   content_settings::PatternParser::Parse("*.youtube.com", &builder);
 144   ::testing::Mock::VerifyAndClear(&builder);
 145
 146   EXPECT_CALL(builder, WithSchemeWildcard()).Times(1).WillOnce(
 147       ::testing::Return(&builder));
 148   EXPECT_CALL(builder, Invalid()).Times(1).WillOnce(
 149       ::testing::Return(&builder));
 150   content_settings::PatternParser::Parse("www.youtube.com*", &builder);
 151   ::testing::Mock::VerifyAndClear(&builder);
 152 }
 153
 154 TEST(ContentSettingsPatternParserTest, ParseFilePatterns) {
 155   ::testing::StrictMock<MockBuilder> builder;
 156
 157   EXPECT_CALL(builder, WithScheme("file")).Times(1).WillOnce(
 158       ::testing::Return(&builder));
 159   EXPECT_CALL(builder, WithPath("/foo/bar/test.html")).Times(1).WillOnce(
 160       ::testing::Return(&builder));
 161   content_settings::PatternParser::Parse(
 162       "file:///foo/bar/test.html", &builder);
 163   ::testing::Mock::VerifyAndClear(&builder);
 164
 165   EXPECT_CALL(builder, WithScheme("file")).Times(1).WillOnce(
 166       ::testing::Return(&builder));
 167   EXPECT_CALL(builder, WithDomainWildcard()).Times(1).WillOnce(
 168       ::testing::Return(&builder));
 169   content_settings::PatternParser::Parse(
 170       "file://*", &builder);
 171   ::testing::Mock::VerifyAndClear(&builder);
 172
 173   EXPECT_CALL(builder, WithScheme("file")).Times(1).WillOnce(
 174       ::testing::Return(&builder));
 175   EXPECT_CALL(builder, WithDomainWildcard()).Times(1).WillOnce(
 176       ::testing::Return(&builder));
 177   EXPECT_CALL(builder, WithPath("/")).Times(1).WillOnce(
 178       ::testing::Return(&builder));
 179   content_settings::PatternParser::Parse(
 180       "file://*/", &builder);
 181   ::testing::Mock::VerifyAndClear(&builder);
 182
 183   EXPECT_CALL(builder, WithScheme("file")).Times(1).WillOnce(
 184       ::testing::Return(&builder));
 185   EXPECT_CALL(builder, WithDomainWildcard()).Times(1).WillOnce(
 186       ::testing::Return(&builder));
 187   EXPECT_CALL(builder, WithPathWildcard()).Times(1).WillOnce(
 188       ::testing::Return(&builder));
 189   content_settings::PatternParser::Parse(
 190       "file://*/*", &builder);
 191   ::testing::Mock::VerifyAndClear(&builder);
 192
 193   EXPECT_CALL(builder, WithScheme("file")).Times(1).WillOnce(
 194       ::testing::Return(&builder));
 195   EXPECT_CALL(builder, WithPathWildcard()).Times(1).WillOnce(
 196       ::testing::Return(&builder));
 197   content_settings::PatternParser::Parse(
 198       "file:///*", &builder);
 199   ::testing::Mock::VerifyAndClear(&builder);
 200
 201   // Invalid file patterns.
 202   EXPECT_CALL(builder, WithScheme("file")).Times(1).WillOnce(
 203       ::testing::Return(&builder));
 204   EXPECT_CALL(builder, Invalid()).Times(1).WillOnce(
 205       ::testing::Return(&builder));
 206   content_settings::PatternParser::Parse(
 207       "file://**", &builder);
 208   ::testing::Mock::VerifyAndClear(&builder);
 209 }
 210
 211 TEST(ContentSettingsPatternParserTest, SerializePatterns) {
 212   ContentSettingsPattern::PatternParts parts;
 213   parts.scheme = "http";
 214   parts.host = "www.youtube.com";
 215   parts.port = "8080";
 216   EXPECT_STREQ("http://www.youtube.com:8080",
 217                content_settings::PatternParser::ToString(parts).c_str());
 218
 219   parts = ContentSettingsPattern::PatternParts();
 220   parts.scheme = "file";
 221   parts.path = "/foo/bar/test.html";
 222   EXPECT_STREQ("file:///foo/bar/test.html",
 223                content_settings::PatternParser::ToString(parts).c_str());
 224
 225   parts = ContentSettingsPattern::PatternParts();
 226   parts.scheme = "file";
 227   parts.path = "";
 228   parts.is_path_wildcard = true;
 229   EXPECT_EQ("file:///*", content_settings::PatternParser::ToString(parts));
 230 }