OLD | NEW |
| (Empty) |
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 <algorithm> | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/message_loop/message_loop.h" | |
9 #include "base/strings/string_util.h" | |
10 #include "base/strings/stringprintf.h" | |
11 #include "base/test/perf_time_logger.h" | |
12 #include "net/cookies/canonical_cookie.h" | |
13 #include "net/cookies/cookie_monster.h" | |
14 #include "net/cookies/cookie_monster_store_test.h" | |
15 #include "net/cookies/parsed_cookie.h" | |
16 #include "testing/gtest/include/gtest/gtest.h" | |
17 #include "url/gurl.h" | |
18 | |
19 namespace net { | |
20 | |
21 namespace { | |
22 | |
23 const int kNumCookies = 20000; | |
24 const char kCookieLine[] = "A = \"b=;\\\"\" ;secure;;;"; | |
25 const char kGoogleURL[] = "http://www.google.izzle"; | |
26 | |
27 int CountInString(const std::string& str, char c) { | |
28 return std::count(str.begin(), str.end(), c); | |
29 } | |
30 | |
31 class CookieMonsterTest : public testing::Test { | |
32 public: | |
33 CookieMonsterTest() : message_loop_(new base::MessageLoopForIO()) {} | |
34 | |
35 private: | |
36 scoped_ptr<base::MessageLoop> message_loop_; | |
37 }; | |
38 | |
39 class BaseCallback { | |
40 public: | |
41 BaseCallback() : has_run_(false) {} | |
42 | |
43 protected: | |
44 void WaitForCallback() { | |
45 // Note that the performance tests currently all operate on a loaded cookie | |
46 // store (or, more precisely, one that has no backing persistent store). | |
47 // Therefore, callbacks will actually always complete synchronously. If the | |
48 // tests get more advanced we need to add other means of signaling | |
49 // completion. | |
50 base::MessageLoop::current()->RunUntilIdle(); | |
51 EXPECT_TRUE(has_run_); | |
52 has_run_ = false; | |
53 } | |
54 | |
55 void Run() { | |
56 has_run_ = true; | |
57 } | |
58 | |
59 bool has_run_; | |
60 }; | |
61 | |
62 class SetCookieCallback : public BaseCallback { | |
63 public: | |
64 void SetCookie( | |
65 CookieMonster* cm, const GURL& gurl, const std::string& cookie) { | |
66 cm->SetCookieWithOptionsAsync(gurl, cookie, options_, base::Bind( | |
67 &SetCookieCallback::Run, base::Unretained(this))); | |
68 WaitForCallback(); | |
69 } | |
70 private: | |
71 void Run(bool success) { | |
72 EXPECT_TRUE(success); | |
73 BaseCallback::Run(); | |
74 } | |
75 net::CookieOptions options_; | |
76 }; | |
77 | |
78 class GetCookiesCallback : public BaseCallback { | |
79 public: | |
80 const std::string& GetCookies(CookieMonster* cm, const GURL& gurl) { | |
81 cm->GetCookiesWithOptionsAsync(gurl, options_, base::Bind( | |
82 &GetCookiesCallback::Run, base::Unretained(this))); | |
83 WaitForCallback(); | |
84 return cookies_; | |
85 } | |
86 | |
87 private: | |
88 void Run(const std::string& cookies) { | |
89 cookies_ = cookies; | |
90 BaseCallback::Run(); | |
91 } | |
92 std::string cookies_; | |
93 net::CookieOptions options_; | |
94 }; | |
95 | |
96 } // namespace | |
97 | |
98 TEST(ParsedCookieTest, TestParseCookies) { | |
99 std::string cookie(kCookieLine); | |
100 base::PerfTimeLogger timer("Parsed_cookie_parse_cookies"); | |
101 for (int i = 0; i < kNumCookies; ++i) { | |
102 ParsedCookie pc(cookie); | |
103 EXPECT_TRUE(pc.IsValid()); | |
104 } | |
105 timer.Done(); | |
106 } | |
107 | |
108 TEST(ParsedCookieTest, TestParseBigCookies) { | |
109 std::string cookie(3800, 'z'); | |
110 cookie += kCookieLine; | |
111 base::PerfTimeLogger timer("Parsed_cookie_parse_big_cookies"); | |
112 for (int i = 0; i < kNumCookies; ++i) { | |
113 ParsedCookie pc(cookie); | |
114 EXPECT_TRUE(pc.IsValid()); | |
115 } | |
116 timer.Done(); | |
117 } | |
118 | |
119 TEST_F(CookieMonsterTest, TestAddCookiesOnSingleHost) { | |
120 scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); | |
121 std::vector<std::string> cookies; | |
122 for (int i = 0; i < kNumCookies; i++) { | |
123 cookies.push_back(base::StringPrintf("a%03d=b", i)); | |
124 } | |
125 | |
126 SetCookieCallback setCookieCallback; | |
127 | |
128 // Add a bunch of cookies on a single host | |
129 base::PerfTimeLogger timer("Cookie_monster_add_single_host"); | |
130 | |
131 for (std::vector<std::string>::const_iterator it = cookies.begin(); | |
132 it != cookies.end(); ++it) { | |
133 setCookieCallback.SetCookie(cm.get(), GURL(kGoogleURL), *it); | |
134 } | |
135 timer.Done(); | |
136 | |
137 GetCookiesCallback getCookiesCallback; | |
138 | |
139 base::PerfTimeLogger timer2("Cookie_monster_query_single_host"); | |
140 for (std::vector<std::string>::const_iterator it = cookies.begin(); | |
141 it != cookies.end(); ++it) { | |
142 getCookiesCallback.GetCookies(cm.get(), GURL(kGoogleURL)); | |
143 } | |
144 timer2.Done(); | |
145 | |
146 base::PerfTimeLogger timer3("Cookie_monster_deleteall_single_host"); | |
147 cm->DeleteAllAsync(CookieMonster::DeleteCallback()); | |
148 base::MessageLoop::current()->RunUntilIdle(); | |
149 timer3.Done(); | |
150 } | |
151 | |
152 TEST_F(CookieMonsterTest, TestAddCookieOnManyHosts) { | |
153 scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); | |
154 std::string cookie(kCookieLine); | |
155 std::vector<GURL> gurls; // just wanna have ffffuunnn | |
156 for (int i = 0; i < kNumCookies; ++i) { | |
157 gurls.push_back(GURL(base::StringPrintf("https://a%04d.izzle", i))); | |
158 } | |
159 | |
160 SetCookieCallback setCookieCallback; | |
161 | |
162 // Add a cookie on a bunch of host | |
163 base::PerfTimeLogger timer("Cookie_monster_add_many_hosts"); | |
164 for (std::vector<GURL>::const_iterator it = gurls.begin(); | |
165 it != gurls.end(); ++it) { | |
166 setCookieCallback.SetCookie(cm.get(), *it, cookie); | |
167 } | |
168 timer.Done(); | |
169 | |
170 GetCookiesCallback getCookiesCallback; | |
171 | |
172 base::PerfTimeLogger timer2("Cookie_monster_query_many_hosts"); | |
173 for (std::vector<GURL>::const_iterator it = gurls.begin(); | |
174 it != gurls.end(); ++it) { | |
175 getCookiesCallback.GetCookies(cm.get(), *it); | |
176 } | |
177 timer2.Done(); | |
178 | |
179 base::PerfTimeLogger timer3("Cookie_monster_deleteall_many_hosts"); | |
180 cm->DeleteAllAsync(CookieMonster::DeleteCallback()); | |
181 base::MessageLoop::current()->RunUntilIdle(); | |
182 timer3.Done(); | |
183 } | |
184 | |
185 TEST_F(CookieMonsterTest, TestDomainTree) { | |
186 scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); | |
187 GetCookiesCallback getCookiesCallback; | |
188 SetCookieCallback setCookieCallback; | |
189 const char domain_cookie_format_tree[] = "a=b; domain=%s"; | |
190 const std::string domain_base("top.com"); | |
191 | |
192 std::vector<std::string> domain_list; | |
193 | |
194 // Create a balanced binary tree of domains on which the cookie is set. | |
195 domain_list.push_back(domain_base); | |
196 for (int i1 = 0; i1 < 2; i1++) { | |
197 std::string domain_base_1((i1 ? "a." : "b.") + domain_base); | |
198 EXPECT_EQ("top.com", cm->GetKey(domain_base_1)); | |
199 domain_list.push_back(domain_base_1); | |
200 for (int i2 = 0; i2 < 2; i2++) { | |
201 std::string domain_base_2((i2 ? "a." : "b.") + domain_base_1); | |
202 EXPECT_EQ("top.com", cm->GetKey(domain_base_2)); | |
203 domain_list.push_back(domain_base_2); | |
204 for (int i3 = 0; i3 < 2; i3++) { | |
205 std::string domain_base_3((i3 ? "a." : "b.") + domain_base_2); | |
206 EXPECT_EQ("top.com", cm->GetKey(domain_base_3)); | |
207 domain_list.push_back(domain_base_3); | |
208 for (int i4 = 0; i4 < 2; i4++) { | |
209 std::string domain_base_4((i4 ? "a." : "b.") + domain_base_3); | |
210 EXPECT_EQ("top.com", cm->GetKey(domain_base_4)); | |
211 domain_list.push_back(domain_base_4); | |
212 } | |
213 } | |
214 } | |
215 } | |
216 | |
217 | |
218 EXPECT_EQ(31u, domain_list.size()); | |
219 for (std::vector<std::string>::const_iterator it = domain_list.begin(); | |
220 it != domain_list.end(); it++) { | |
221 GURL gurl("https://" + *it + "/"); | |
222 const std::string cookie = base::StringPrintf(domain_cookie_format_tree, | |
223 it->c_str()); | |
224 setCookieCallback.SetCookie(cm.get(), gurl, cookie); | |
225 } | |
226 EXPECT_EQ(31u, cm->GetAllCookies().size()); | |
227 | |
228 GURL probe_gurl("https://b.a.b.a.top.com/"); | |
229 std::string cookie_line = getCookiesCallback.GetCookies(cm.get(), probe_gurl); | |
230 EXPECT_EQ(5, CountInString(cookie_line, '=')) | |
231 << "Cookie line: " << cookie_line; | |
232 base::PerfTimeLogger timer("Cookie_monster_query_domain_tree"); | |
233 for (int i = 0; i < kNumCookies; i++) { | |
234 getCookiesCallback.GetCookies(cm.get(), probe_gurl); | |
235 } | |
236 timer.Done(); | |
237 } | |
238 | |
239 TEST_F(CookieMonsterTest, TestDomainLine) { | |
240 scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); | |
241 SetCookieCallback setCookieCallback; | |
242 GetCookiesCallback getCookiesCallback; | |
243 std::vector<std::string> domain_list; | |
244 GURL probe_gurl("https://b.a.b.a.top.com/"); | |
245 std::string cookie_line; | |
246 | |
247 // Create a line of 32 domain cookies such that all cookies stored | |
248 // by effective TLD+1 will apply to probe GURL. | |
249 // (TLD + 1 is the level above .com/org/net/etc, e.g. "top.com" | |
250 // or "google.com". "Effective" is added to include sites like | |
251 // bbc.co.uk, where the effetive TLD+1 is more than one level | |
252 // below the top level.) | |
253 domain_list.push_back("a.top.com"); | |
254 domain_list.push_back("b.a.top.com"); | |
255 domain_list.push_back("a.b.a.top.com"); | |
256 domain_list.push_back("b.a.b.a.top.com"); | |
257 EXPECT_EQ(4u, domain_list.size()); | |
258 | |
259 const char domain_cookie_format_line[] = "a%03d=b; domain=%s"; | |
260 for (int i = 0; i < 8; i++) { | |
261 for (std::vector<std::string>::const_iterator it = domain_list.begin(); | |
262 it != domain_list.end(); it++) { | |
263 GURL gurl("https://" + *it + "/"); | |
264 const std::string cookie = base::StringPrintf(domain_cookie_format_line, | |
265 i, it->c_str()); | |
266 setCookieCallback.SetCookie(cm.get(), gurl, cookie); | |
267 } | |
268 } | |
269 | |
270 cookie_line = getCookiesCallback.GetCookies(cm.get(), probe_gurl); | |
271 EXPECT_EQ(32, CountInString(cookie_line, '=')); | |
272 base::PerfTimeLogger timer2("Cookie_monster_query_domain_line"); | |
273 for (int i = 0; i < kNumCookies; i++) { | |
274 getCookiesCallback.GetCookies(cm.get(), probe_gurl); | |
275 } | |
276 timer2.Done(); | |
277 } | |
278 | |
279 TEST_F(CookieMonsterTest, TestImport) { | |
280 scoped_refptr<MockPersistentCookieStore> store(new MockPersistentCookieStore); | |
281 std::vector<CanonicalCookie*> initial_cookies; | |
282 GetCookiesCallback getCookiesCallback; | |
283 | |
284 // We want to setup a fairly large backing store, with 300 domains of 50 | |
285 // cookies each. Creation times must be unique. | |
286 int64 time_tick(base::Time::Now().ToInternalValue()); | |
287 | |
288 for (int domain_num = 0; domain_num < 300; domain_num++) { | |
289 std::string domain_name(base::StringPrintf(".Domain_%d.com", domain_num)); | |
290 std::string gurl("www" + domain_name); | |
291 for (int cookie_num = 0; cookie_num < 50; cookie_num++) { | |
292 std::string cookie_line(base::StringPrintf("Cookie_%d=1; Path=/", | |
293 cookie_num)); | |
294 AddCookieToList(gurl, cookie_line, | |
295 base::Time::FromInternalValue(time_tick++), | |
296 &initial_cookies); | |
297 } | |
298 } | |
299 | |
300 store->SetLoadExpectation(true, initial_cookies); | |
301 | |
302 scoped_refptr<CookieMonster> cm(new CookieMonster(store.get(), NULL)); | |
303 | |
304 // Import will happen on first access. | |
305 GURL gurl("www.google.com"); | |
306 CookieOptions options; | |
307 base::PerfTimeLogger timer("Cookie_monster_import_from_store"); | |
308 getCookiesCallback.GetCookies(cm.get(), gurl); | |
309 timer.Done(); | |
310 | |
311 // Just confirm keys were set as expected. | |
312 EXPECT_EQ("domain_1.com", cm->GetKey("www.Domain_1.com")); | |
313 } | |
314 | |
315 TEST_F(CookieMonsterTest, TestGetKey) { | |
316 scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); | |
317 base::PerfTimeLogger timer("Cookie_monster_get_key"); | |
318 for (int i = 0; i < kNumCookies; i++) | |
319 cm->GetKey("www.google.com"); | |
320 timer.Done(); | |
321 } | |
322 | |
323 // This test is probing for whether garbage collection happens when it | |
324 // shouldn't. This will not in general be visible functionally, since | |
325 // if GC runs twice in a row without any change to the store, the second | |
326 // GC run will not do anything the first one didn't. That's why this is | |
327 // a performance test. The test should be considered to pass if all the | |
328 // times reported are approximately the same--this indicates that no GC | |
329 // happened repeatedly for any case. | |
330 TEST_F(CookieMonsterTest, TestGCTimes) { | |
331 SetCookieCallback setCookieCallback; | |
332 | |
333 const struct TestCase { | |
334 const char* const name; | |
335 size_t num_cookies; | |
336 size_t num_old_cookies; | |
337 } test_cases[] = { | |
338 { | |
339 // A whole lot of recent cookies; gc shouldn't happen. | |
340 "all_recent", | |
341 CookieMonster::kMaxCookies * 2, | |
342 0, | |
343 }, { | |
344 // Some old cookies, but still overflowing max. | |
345 "mostly_recent", | |
346 CookieMonster::kMaxCookies * 2, | |
347 CookieMonster::kMaxCookies / 2, | |
348 }, { | |
349 // Old cookies enough to bring us right down to our purge line. | |
350 "balanced", | |
351 CookieMonster::kMaxCookies * 2, | |
352 CookieMonster::kMaxCookies + CookieMonster::kPurgeCookies + 1, | |
353 }, { | |
354 "mostly_old", | |
355 // Old cookies enough to bring below our purge line (which we | |
356 // shouldn't do). | |
357 CookieMonster::kMaxCookies * 2, | |
358 CookieMonster::kMaxCookies * 3 / 4, | |
359 }, { | |
360 "less_than_gc_thresh", | |
361 // Few enough cookies that gc shouldn't happen at all. | |
362 CookieMonster::kMaxCookies - 5, | |
363 0, | |
364 }, | |
365 }; | |
366 for (int ci = 0; ci < static_cast<int>(arraysize(test_cases)); ++ci) { | |
367 const TestCase& test_case(test_cases[ci]); | |
368 scoped_refptr<CookieMonster> cm( | |
369 CreateMonsterFromStoreForGC( | |
370 test_case.num_cookies, test_case.num_old_cookies, | |
371 CookieMonster::kSafeFromGlobalPurgeDays * 2)); | |
372 | |
373 GURL gurl("http://google.com"); | |
374 std::string cookie_line("z=3"); | |
375 // Trigger the Garbage collection we're allowed. | |
376 setCookieCallback.SetCookie(cm.get(), gurl, cookie_line); | |
377 | |
378 base::PerfTimeLogger timer((std::string("GC_") + test_case.name).c_str()); | |
379 for (int i = 0; i < kNumCookies; i++) | |
380 setCookieCallback.SetCookie(cm.get(), gurl, cookie_line); | |
381 timer.Done(); | |
382 } | |
383 } | |
384 | |
385 } // namespace net | |
OLD | NEW |