OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "net/tools/tld_cleanup/tld_cleanup_util.h" | 5 #include "net/tools/tld_cleanup/tld_cleanup_util.h" |
6 | 6 |
7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
8 #include "base/path_service.h" | 8 #include "base/path_service.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 | 10 |
11 namespace net { | 11 namespace net { |
12 namespace tld_cleanup { | 12 namespace tld_cleanup { |
13 | 13 |
14 std::string SetupData(std::string icann_domains, std::string private_domains) { | 14 std::string SetupData(std::string icann_domains, std::string private_domains) { |
15 return "// ===BEGIN ICANN DOMAINS===\n" + | 15 return "// ===BEGIN ICANN DOMAINS===\n" + icann_domains + |
16 icann_domains + | 16 "// ===END ICANN DOMAINS===\n" + "// ===BEGIN PRIVATE DOMAINS===\n" + |
17 "// ===END ICANN DOMAINS===\n" + | 17 private_domains + "// ===END PRIVATE DOMAINS===\n"; |
18 "// ===BEGIN PRIVATE DOMAINS===\n" + | |
19 private_domains + | |
20 "// ===END PRIVATE DOMAINS===\n"; | |
21 } | 18 } |
22 | 19 |
23 TEST(TldCleanupUtilTest, TwoRealTldsSuccessfullyRead) { | 20 TEST(TldCleanupUtilTest, TwoRealTldsSuccessfullyRead) { |
24 std::string icann_domains = "foo\n" | 21 std::string icann_domains = |
25 "bar\n"; | 22 "foo\n" |
| 23 "bar\n"; |
26 std::string private_domains = ""; | 24 std::string private_domains = ""; |
27 std::string data = SetupData(icann_domains, private_domains); | 25 std::string data = SetupData(icann_domains, private_domains); |
28 RuleMap rules; | 26 RuleMap rules; |
29 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); | 27 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); |
30 ASSERT_EQ(kSuccess, result); | 28 ASSERT_EQ(kSuccess, result); |
31 ASSERT_EQ(2U, rules.size()); | 29 ASSERT_EQ(2U, rules.size()); |
32 RuleMap::const_iterator foo_iter = rules.find("foo"); | 30 RuleMap::const_iterator foo_iter = rules.find("foo"); |
33 ASSERT_FALSE(rules.end() == foo_iter); | 31 ASSERT_FALSE(rules.end() == foo_iter); |
34 EXPECT_FALSE(foo_iter->second.wildcard); | 32 EXPECT_FALSE(foo_iter->second.wildcard); |
35 EXPECT_FALSE(foo_iter->second.exception); | 33 EXPECT_FALSE(foo_iter->second.exception); |
(...skipping 19 matching lines...) Expand all Loading... |
55 EXPECT_FALSE(foo_bar_iter->second.exception); | 53 EXPECT_FALSE(foo_bar_iter->second.exception); |
56 EXPECT_FALSE(foo_bar_iter->second.is_private); | 54 EXPECT_FALSE(foo_bar_iter->second.is_private); |
57 RuleMap::const_iterator bar_iter = rules.find("bar"); | 55 RuleMap::const_iterator bar_iter = rules.find("bar"); |
58 ASSERT_FALSE(rules.end() == bar_iter); | 56 ASSERT_FALSE(rules.end() == bar_iter); |
59 EXPECT_FALSE(bar_iter->second.wildcard); | 57 EXPECT_FALSE(bar_iter->second.wildcard); |
60 EXPECT_FALSE(bar_iter->second.exception); | 58 EXPECT_FALSE(bar_iter->second.exception); |
61 EXPECT_FALSE(bar_iter->second.is_private); | 59 EXPECT_FALSE(bar_iter->second.is_private); |
62 } | 60 } |
63 | 61 |
64 TEST(TldCleanupUtilTest, PrivateTldMarkedAsPrivate) { | 62 TEST(TldCleanupUtilTest, PrivateTldMarkedAsPrivate) { |
65 std::string icann_domains = "foo\n" | 63 std::string icann_domains = |
66 "bar\n"; | 64 "foo\n" |
| 65 "bar\n"; |
67 std::string private_domains = "baz\n"; | 66 std::string private_domains = "baz\n"; |
68 std::string data = SetupData(icann_domains, private_domains); | 67 std::string data = SetupData(icann_domains, private_domains); |
69 RuleMap rules; | 68 RuleMap rules; |
70 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); | 69 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); |
71 ASSERT_EQ(kSuccess, result); | 70 ASSERT_EQ(kSuccess, result); |
72 ASSERT_EQ(3U, rules.size()); | 71 ASSERT_EQ(3U, rules.size()); |
73 RuleMap::const_iterator foo_iter = rules.find("foo"); | 72 RuleMap::const_iterator foo_iter = rules.find("foo"); |
74 ASSERT_FALSE(rules.end() == foo_iter); | 73 ASSERT_FALSE(rules.end() == foo_iter); |
75 EXPECT_FALSE(foo_iter->second.wildcard); | 74 EXPECT_FALSE(foo_iter->second.wildcard); |
76 EXPECT_FALSE(foo_iter->second.exception); | 75 EXPECT_FALSE(foo_iter->second.exception); |
(...skipping 24 matching lines...) Expand all Loading... |
101 EXPECT_FALSE(bar_iter->second.exception); | 100 EXPECT_FALSE(bar_iter->second.exception); |
102 EXPECT_FALSE(bar_iter->second.is_private); | 101 EXPECT_FALSE(bar_iter->second.is_private); |
103 RuleMap::const_iterator foo_bar_iter = rules.find("foo.bar"); | 102 RuleMap::const_iterator foo_bar_iter = rules.find("foo.bar"); |
104 ASSERT_FALSE(rules.end() == foo_bar_iter); | 103 ASSERT_FALSE(rules.end() == foo_bar_iter); |
105 EXPECT_FALSE(foo_bar_iter->second.wildcard); | 104 EXPECT_FALSE(foo_bar_iter->second.wildcard); |
106 EXPECT_FALSE(foo_bar_iter->second.exception); | 105 EXPECT_FALSE(foo_bar_iter->second.exception); |
107 EXPECT_TRUE(foo_bar_iter->second.is_private); | 106 EXPECT_TRUE(foo_bar_iter->second.is_private); |
108 } | 107 } |
109 | 108 |
110 TEST(TldCleanupUtilTest, ExtraTldRuleIsNotMarkedPrivate) { | 109 TEST(TldCleanupUtilTest, ExtraTldRuleIsNotMarkedPrivate) { |
111 std::string icann_domains = "foo.bar\n" | 110 std::string icann_domains = |
112 "baz.bar\n"; | 111 "foo.bar\n" |
| 112 "baz.bar\n"; |
113 std::string private_domains = "qux.bar\n"; | 113 std::string private_domains = "qux.bar\n"; |
114 std::string data = SetupData(icann_domains, private_domains); | 114 std::string data = SetupData(icann_domains, private_domains); |
115 RuleMap rules; | 115 RuleMap rules; |
116 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); | 116 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); |
117 ASSERT_EQ(kSuccess, result); | 117 ASSERT_EQ(kSuccess, result); |
118 ASSERT_EQ(4U, rules.size()); | 118 ASSERT_EQ(4U, rules.size()); |
119 RuleMap::const_iterator foo_bar_iter = rules.find("foo.bar"); | 119 RuleMap::const_iterator foo_bar_iter = rules.find("foo.bar"); |
120 ASSERT_FALSE(rules.end() == foo_bar_iter); | 120 ASSERT_FALSE(rules.end() == foo_bar_iter); |
121 EXPECT_FALSE(foo_bar_iter->second.wildcard); | 121 EXPECT_FALSE(foo_bar_iter->second.wildcard); |
122 EXPECT_FALSE(foo_bar_iter->second.exception); | 122 EXPECT_FALSE(foo_bar_iter->second.exception); |
123 EXPECT_FALSE(foo_bar_iter->second.is_private); | 123 EXPECT_FALSE(foo_bar_iter->second.is_private); |
124 RuleMap::const_iterator baz_bar_iter = rules.find("baz.bar"); | 124 RuleMap::const_iterator baz_bar_iter = rules.find("baz.bar"); |
125 ASSERT_FALSE(rules.end() == baz_bar_iter); | 125 ASSERT_FALSE(rules.end() == baz_bar_iter); |
126 EXPECT_FALSE(baz_bar_iter->second.wildcard); | 126 EXPECT_FALSE(baz_bar_iter->second.wildcard); |
127 EXPECT_FALSE(baz_bar_iter->second.exception); | 127 EXPECT_FALSE(baz_bar_iter->second.exception); |
128 EXPECT_FALSE(baz_bar_iter->second.is_private); | 128 EXPECT_FALSE(baz_bar_iter->second.is_private); |
129 RuleMap::const_iterator bar_iter = rules.find("bar"); | 129 RuleMap::const_iterator bar_iter = rules.find("bar"); |
130 ASSERT_FALSE(rules.end() == bar_iter); | 130 ASSERT_FALSE(rules.end() == bar_iter); |
131 EXPECT_FALSE(bar_iter->second.wildcard); | 131 EXPECT_FALSE(bar_iter->second.wildcard); |
132 EXPECT_FALSE(bar_iter->second.exception); | 132 EXPECT_FALSE(bar_iter->second.exception); |
133 EXPECT_FALSE(bar_iter->second.is_private); | 133 EXPECT_FALSE(bar_iter->second.is_private); |
134 RuleMap::const_iterator qux_bar_iter = rules.find("qux.bar"); | 134 RuleMap::const_iterator qux_bar_iter = rules.find("qux.bar"); |
135 ASSERT_FALSE(rules.end() == qux_bar_iter); | 135 ASSERT_FALSE(rules.end() == qux_bar_iter); |
136 EXPECT_FALSE(qux_bar_iter->second.wildcard); | 136 EXPECT_FALSE(qux_bar_iter->second.wildcard); |
137 EXPECT_FALSE(qux_bar_iter->second.exception); | 137 EXPECT_FALSE(qux_bar_iter->second.exception); |
138 EXPECT_TRUE(qux_bar_iter->second.is_private); | 138 EXPECT_TRUE(qux_bar_iter->second.is_private); |
139 } | 139 } |
140 | 140 |
141 TEST(TldCleanupUtilTest, WildcardAndExceptionParsedCorrectly) { | 141 TEST(TldCleanupUtilTest, WildcardAndExceptionParsedCorrectly) { |
142 std::string icann_domains = "*.bar\n" | 142 std::string icann_domains = |
143 "!foo.bar\n"; | 143 "*.bar\n" |
| 144 "!foo.bar\n"; |
144 std::string private_domains = "!baz.bar\n"; | 145 std::string private_domains = "!baz.bar\n"; |
145 std::string data = SetupData(icann_domains, private_domains); | 146 std::string data = SetupData(icann_domains, private_domains); |
146 RuleMap rules; | 147 RuleMap rules; |
147 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); | 148 NormalizeResult result = NormalizeDataToRuleMap(data, &rules); |
148 ASSERT_EQ(kSuccess, result); | 149 ASSERT_EQ(kSuccess, result); |
149 ASSERT_EQ(3U, rules.size()); | 150 ASSERT_EQ(3U, rules.size()); |
150 RuleMap::const_iterator foo_bar_iter = rules.find("bar"); | 151 RuleMap::const_iterator foo_bar_iter = rules.find("bar"); |
151 ASSERT_FALSE(rules.end() == foo_bar_iter); | 152 ASSERT_FALSE(rules.end() == foo_bar_iter); |
152 EXPECT_TRUE(foo_bar_iter->second.wildcard); | 153 EXPECT_TRUE(foo_bar_iter->second.wildcard); |
153 EXPECT_FALSE(foo_bar_iter->second.exception); | 154 EXPECT_FALSE(foo_bar_iter->second.exception); |
154 EXPECT_FALSE(foo_bar_iter->second.is_private); | 155 EXPECT_FALSE(foo_bar_iter->second.is_private); |
155 RuleMap::const_iterator bar_iter = rules.find("foo.bar"); | 156 RuleMap::const_iterator bar_iter = rules.find("foo.bar"); |
156 ASSERT_FALSE(rules.end() == bar_iter); | 157 ASSERT_FALSE(rules.end() == bar_iter); |
157 EXPECT_FALSE(bar_iter->second.wildcard); | 158 EXPECT_FALSE(bar_iter->second.wildcard); |
158 EXPECT_TRUE(bar_iter->second.exception); | 159 EXPECT_TRUE(bar_iter->second.exception); |
159 EXPECT_FALSE(bar_iter->second.is_private); | 160 EXPECT_FALSE(bar_iter->second.is_private); |
160 RuleMap::const_iterator baz_bar_iter = rules.find("baz.bar"); | 161 RuleMap::const_iterator baz_bar_iter = rules.find("baz.bar"); |
161 ASSERT_FALSE(rules.end() == baz_bar_iter); | 162 ASSERT_FALSE(rules.end() == baz_bar_iter); |
162 EXPECT_FALSE(baz_bar_iter->second.wildcard); | 163 EXPECT_FALSE(baz_bar_iter->second.wildcard); |
163 EXPECT_TRUE(baz_bar_iter->second.exception); | 164 EXPECT_TRUE(baz_bar_iter->second.exception); |
164 EXPECT_TRUE(baz_bar_iter->second.is_private); | 165 EXPECT_TRUE(baz_bar_iter->second.is_private); |
165 } | 166 } |
166 | 167 |
167 } // namespace tld_cleanup | 168 } // namespace tld_cleanup |
168 } // namespace net | 169 } // namespace net |
OLD | NEW |