13345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick// Copyright (c) 2010 The Chromium Authors. All rights reserved. 2c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// Use of this source code is governed by a BSD-style license that can be 3c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott// found in the LICENSE file. 4c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include <algorithm> 672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 7731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick#include "net/base/cookie_monster.h" 8731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 9c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "base/perftimer.h" 10c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "base/string_util.h" 113345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "base/stringprintf.h" 123345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "googleurl/src/gurl.h" 13c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "net/base/cookie_monster.h" 143345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick#include "net/base/cookie_monster_store_test.h" 15c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott#include "testing/gtest/include/gtest/gtest.h" 16c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 17c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottnamespace { 18c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott class ParsedCookieTest : public testing::Test { }; 19c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott class CookieMonsterTest : public testing::Test { }; 20c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 21c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 22c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic const int kNumCookies = 20000; 23c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic const char kCookieLine[] = "A = \"b=;\\\"\" ;secure;;;"; 24c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 253345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merricknamespace net { 263345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 27c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick ScottTEST(ParsedCookieTest, TestParseCookies) { 28c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott std::string cookie(kCookieLine); 29c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer("Parsed_cookie_parse_cookies"); 30c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (int i = 0; i < kNumCookies; ++i) { 313345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick CookieMonster::ParsedCookie pc(cookie); 32c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott EXPECT_TRUE(pc.IsValid()); 33c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 34c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer.Done(); 35c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 36c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 37c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick ScottTEST(ParsedCookieTest, TestParseBigCookies) { 38c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott std::string cookie(3800, 'z'); 39c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cookie += kCookieLine; 40c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer("Parsed_cookie_parse_big_cookies"); 41c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (int i = 0; i < kNumCookies; ++i) { 423345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick CookieMonster::ParsedCookie pc(cookie); 43c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott EXPECT_TRUE(pc.IsValid()); 44c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 45c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer.Done(); 46c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 47c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 48c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scottstatic const GURL kUrlGoogle("http://www.google.izzle"); 49c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 50c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick ScottTEST(CookieMonsterTest, TestAddCookiesOnSingleHost) { 513345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); 52c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott std::vector<std::string> cookies; 53c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (int i = 0; i < kNumCookies; i++) { 543345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick cookies.push_back(base::StringPrintf("a%03d=b", i)); 55c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 56c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 57c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Add a bunch of cookies on a single host 58c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer("Cookie_monster_add_single_host"); 59c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (std::vector<std::string>::const_iterator it = cookies.begin(); 60c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott it != cookies.end(); ++it) { 61c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott EXPECT_TRUE(cm->SetCookie(kUrlGoogle, *it)); 62c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 63c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer.Done(); 64c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 65c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer2("Cookie_monster_query_single_host"); 66c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (std::vector<std::string>::const_iterator it = cookies.begin(); 67c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott it != cookies.end(); ++it) { 68c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cm->GetCookies(kUrlGoogle); 69c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 70c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer2.Done(); 71c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 72c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer3("Cookie_monster_deleteall_single_host"); 73c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cm->DeleteAll(false); 74c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer3.Done(); 75c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 76c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 77c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick ScottTEST(CookieMonsterTest, TestAddCookieOnManyHosts) { 783345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); 79c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott std::string cookie(kCookieLine); 80c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott std::vector<GURL> gurls; // just wanna have ffffuunnn 81c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (int i = 0; i < kNumCookies; ++i) { 823345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick gurls.push_back(GURL(base::StringPrintf("http://a%04d.izzle", i))); 83c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 84c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 85c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott // Add a cookie on a bunch of host 86c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer("Cookie_monster_add_many_hosts"); 87c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (std::vector<GURL>::const_iterator it = gurls.begin(); 88c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott it != gurls.end(); ++it) { 89c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott EXPECT_TRUE(cm->SetCookie(*it, cookie)); 90c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 91c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer.Done(); 92c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 93c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer2("Cookie_monster_query_many_hosts"); 94c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott for (std::vector<GURL>::const_iterator it = gurls.begin(); 95c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott it != gurls.end(); ++it) { 96c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cm->GetCookies(*it); 97c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott } 98c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer2.Done(); 99c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott 100c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott PerfTimeLogger timer3("Cookie_monster_deleteall_many_hosts"); 101c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott cm->DeleteAll(false); 102c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott timer3.Done(); 103c7f5f8508d98d5952d42ed7648c2a8f30a4da156Patrick Scott} 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1053345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrickstatic int CountInString(const std::string& str, char c) { 1063345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return std::count(str.begin(), str.end(), c); 1073345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} 1083345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochTEST(CookieMonsterTest, TestDomainTree) { 1103345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); 1113345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const char* domain_cookie_format_tree = "a=b; domain=%s"; 1123345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string domain_base("top.com"); 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1143345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::vector<std::string> domain_list; 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 116c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Create a balanced binary tree of domains on which the cookie is set. 1173345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back(domain_base); 1183345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int i1 = 0; i1 < 2; i1++) { 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch std::string domain_base_1((i1 ? "a." : "b.") + domain_base); 1203345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ("top.com", cm->GetKey(domain_base_1)); 1213345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back(domain_base_1); 1223345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int i2 = 0; i2 < 2; i2++) { 1233345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string domain_base_2((i2 ? "a." : "b.") + domain_base_1); 1243345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ("top.com", cm->GetKey(domain_base_2)); 1253345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back(domain_base_2); 1263345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int i3 = 0; i3 < 2; i3++) { 1273345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string domain_base_3((i3 ? "a." : "b.") + domain_base_2); 1283345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ("top.com", cm->GetKey(domain_base_3)); 1293345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back(domain_base_3); 1303345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int i4 = 0; i4 < 2; i4++) { 1313345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string domain_base_4((i4 ? "a." : "b.") + domain_base_3); 1323345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ("top.com", cm->GetKey(domain_base_4)); 1333345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back(domain_base_4); 134c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 135c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 136c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 137c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 138c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1393345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 1403345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ(31u, domain_list.size()); 1413345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (std::vector<std::string>::const_iterator it = domain_list.begin(); 1423345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick it != domain_list.end(); it++) { 1433345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL gurl("https://" + *it + "/"); 1443345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string cookie = base::StringPrintf(domain_cookie_format_tree, 1453345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick it->c_str()); 1463345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_TRUE(cm->SetCookie(gurl, cookie)); 147c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 1483345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ(31u, cm->GetAllCookies().size()); 149c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1503345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL probe_gurl("https://b.a.b.a.top.com/"); 1513345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string cookie_line; 1523345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick cookie_line = cm->GetCookies(probe_gurl); 1533345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ(5, CountInString(cookie_line, '=')) << "Cookie line: " 1543345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick << cookie_line; 155c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch PerfTimeLogger timer("Cookie_monster_query_domain_tree"); 156c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (int i = 0; i < kNumCookies; i++) { 157c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch cm->GetCookies(probe_gurl); 158c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 159c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch timer.Done(); 160c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1613345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} 1623345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 1633345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickTEST(CookieMonsterTest, TestDomainLine) { 1643345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); 1653345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::vector<std::string> domain_list; 1663345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL probe_gurl("https://b.a.b.a.top.com/"); 1673345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string cookie_line; 168c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 169c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // Create a line of 32 domain cookies such that all cookies stored 170c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // by effective TLD+1 will apply to probe GURL. 171c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // (TLD + 1 is the level above .com/org/net/etc, e.g. "top.com" 172c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // or "google.com". "Effective" is added to include sites like 173c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // bbc.co.uk, where the effetive TLD+1 is more than one level 174c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // below the top level.) 1753345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back("a.top.com"); 1763345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back("b.a.top.com"); 1773345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back("a.b.a.top.com"); 1783345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick domain_list.push_back("b.a.b.a.top.com"); 1793345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ(4u, domain_list.size()); 180c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 1813345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const char* domain_cookie_format_line = "a%03d=b; domain=%s"; 182c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (int i = 0; i < 8; i++) { 1833345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (std::vector<std::string>::const_iterator it = domain_list.begin(); 1843345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick it != domain_list.end(); it++) { 1853345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL gurl("https://" + *it + "/"); 1863345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick const std::string cookie = base::StringPrintf(domain_cookie_format_line, 1873345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick i, it->c_str()); 1883345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_TRUE(cm->SetCookie(gurl, cookie)); 189c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 190c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 1913345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ(32u, cm->GetAllCookies().size()); 1923345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 1933345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick cookie_line = cm->GetCookies(probe_gurl); 1943345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ(32, CountInString(cookie_line, '=')); 195c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch PerfTimeLogger timer2("Cookie_monster_query_domain_line"); 196c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch for (int i = 0; i < kNumCookies; i++) { 197c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch cm->GetCookies(probe_gurl); 198c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 199c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch timer2.Done(); 200c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 2013345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2023345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickTEST(CookieMonsterTest, TestImport) { 2033345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<MockPersistentCookieStore> store(new MockPersistentCookieStore); 2043345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::vector<CookieMonster::CanonicalCookie*> initial_cookies; 2053345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2063345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // We want to setup a fairly large backing store, with 300 domains of 50 2073345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // cookies each. Creation times must be unique. 2083345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick int64 time_tick(base::Time::Now().ToInternalValue()); 2093345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2103345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int domain_num = 0; domain_num < 300; domain_num++) { 2113345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string domain_name(base::StringPrintf(".Domain_%d.com", domain_num)); 2123345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string gurl("www" + domain_name); 2133345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int cookie_num = 0; cookie_num < 50; cookie_num++) { 2143345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick std::string cookie_line(base::StringPrintf("Cookie_%d=1; Path=/", 2153345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick cookie_num)); 2163345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick AddCookieToList(gurl, cookie_line, 2173345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick base::Time::FromInternalValue(time_tick++), 2183345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick &initial_cookies); 2193345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 2203345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick } 2213345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2223345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick store->SetLoadExpectation(true, initial_cookies); 2233345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2243345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<CookieMonster> cm(new CookieMonster(store, NULL)); 2253345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2263345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // Import will happen on first access. 2273345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick GURL gurl("www.google.com"); 2283345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick CookieOptions options; 2293345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick PerfTimeLogger timer("Cookie_monster_import_from_store"); 2303345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick cm->GetCookiesWithOptions(gurl, options); 2313345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick timer.Done(); 2323345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2333345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick // Just confirm keys were set as expected. 2343345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick EXPECT_EQ("domain_1.com", cm->GetKey("www.Domain_1.com")); 2353345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} 2363345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 2373345a6884c488ff3a535c2c9acdd33d74b37e311Iain MerrickTEST(CookieMonsterTest, TestGetKey) { 2383345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick scoped_refptr<CookieMonster> cm(new CookieMonster(NULL, NULL)); 2393345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick PerfTimeLogger timer("Cookie_monster_get_key"); 2403345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick for (int i = 0; i < kNumCookies; i++) 2413345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick cm->GetKey("www.google.com"); 2423345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick timer.Done(); 2433345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} 2443345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick 245731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// This test is probing for whether garbage collection happens when it 246731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// shouldn't. This will not in general be visible functionally, since 247731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// if GC runs twice in a row without any change to the store, the second 248731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// GC run will not do anything the first one didn't. That's why this is 249731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// a performance test. The test should be considered to pass if all the 250731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// times reported are approximately the same--this indicates that no GC 251731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick// happened repeatedly for any case. 252731df977c0511bca2206b5f333555b1205ff1f43Iain MerrickTEST(CookieMonsterTest, TestGCTimes) { 253731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick const struct TestCase { 254731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick const char* name; 255731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick int num_cookies; 256731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick int num_old_cookies; 257731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick } test_cases[] = { 258731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick { 259731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // A whole lot of recent cookies; gc shouldn't happen. 260731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick "all_recent", 261731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies * 2, 262731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 0, 263731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick }, { 264731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // Some old cookies, but still overflowing max. 265731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick "mostly_recent", 266731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies * 2, 267731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies / 2, 268731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick }, { 269731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // Old cookies enough to bring us right down to our purge line. 270731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick "balanced", 271731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies * 2, 272731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies + CookieMonster::kPurgeCookies + 1, 273731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick }, { 274731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick "mostly_old", 275731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // Old cookies enough to bring below our purge line (which we 276731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // shouldn't do). 277731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies * 2, 278731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies * 3 / 4, 279731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick }, { 280731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick "less_than_gc_thresh", 281731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // Few enough cookies that gc shouldn't happen at all. 282731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CookieMonster::kMaxCookies - 5, 283731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 0, 284731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick }, 285731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick }; 286731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick for (int ci = 0; ci < static_cast<int>(ARRAYSIZE_UNSAFE(test_cases)); ++ci) { 287731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick const TestCase& test_case(test_cases[ci]); 288513209b27ff55e2841eac0e4120199c23acce758Ben Murdoch scoped_refptr<CookieMonster> cm( 289731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick CreateMonsterFromStoreForGC( 290731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick test_case.num_cookies, test_case.num_old_cookies, 291513209b27ff55e2841eac0e4120199c23acce758Ben Murdoch CookieMonster::kSafeFromGlobalPurgeDays * 2)); 292731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 293731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick GURL gurl("http://google.com"); 294731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick std::string cookie_line("z=3"); 295731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick // Trigger the Garbage collection we're allowed. 296731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick EXPECT_TRUE(cm->SetCookie(gurl, cookie_line)); 297731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 298731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick PerfTimeLogger timer((std::string("GC_") + test_case.name).c_str()); 299731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick for (int i = 0; i < kNumCookies; i++) 300731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick EXPECT_TRUE(cm->SetCookie(gurl, cookie_line)); 301731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick timer.Done(); 302731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick } 303731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick} 304731df977c0511bca2206b5f333555b1205ff1f43Iain Merrick 3053345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick} // namespace 306