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
11namespace {
12typedef ContentSettingsPattern::BuilderInterface BuilderInterface;
13}  // namespace
14
15class 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
29TEST(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
154TEST(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
211TEST(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}
231