Index: content/browser/net/sqlite_persistent_cookie_store_unittest.cc |
diff --git a/content/browser/net/sqlite_persistent_cookie_store_unittest.cc b/content/browser/net/sqlite_persistent_cookie_store_unittest.cc |
deleted file mode 100644 |
index af7859a26a9f1695dc4e8aa7a809245b2ebfad18..0000000000000000000000000000000000000000 |
--- a/content/browser/net/sqlite_persistent_cookie_store_unittest.cc |
+++ /dev/null |
@@ -1,681 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "content/browser/net/sqlite_persistent_cookie_store.h" |
- |
-#include <map> |
-#include <set> |
- |
-#include "base/bind.h" |
-#include "base/callback.h" |
-#include "base/files/file_util.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/memory/ref_counted.h" |
-#include "base/message_loop/message_loop.h" |
-#include "base/sequenced_task_runner.h" |
-#include "base/stl_util.h" |
-#include "base/synchronization/waitable_event.h" |
-#include "base/test/sequenced_worker_pool_owner.h" |
-#include "base/threading/sequenced_worker_pool.h" |
-#include "base/time/time.h" |
-#include "content/public/browser/cookie_store_factory.h" |
-#include "crypto/encryptor.h" |
-#include "crypto/symmetric_key.h" |
-#include "net/cookies/canonical_cookie.h" |
-#include "net/cookies/cookie_constants.h" |
-#include "net/extras/sqlite/cookie_crypto_delegate.h" |
-#include "sql/connection.h" |
-#include "sql/meta_table.h" |
-#include "sql/statement.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
-#include "url/gurl.h" |
- |
-namespace content { |
- |
-namespace { |
- |
-const base::FilePath::CharType kCookieFilename[] = FILE_PATH_LITERAL("Cookies"); |
- |
-class CookieCryptor : public net::CookieCryptoDelegate { |
- public: |
- CookieCryptor(); |
- bool EncryptString(const std::string& plaintext, |
- std::string* ciphertext) override; |
- bool DecryptString(const std::string& ciphertext, |
- std::string* plaintext) override; |
- |
- private: |
- scoped_ptr<crypto::SymmetricKey> key_; |
- crypto::Encryptor encryptor_; |
-}; |
- |
-CookieCryptor::CookieCryptor() : key_( |
- crypto::SymmetricKey::DeriveKeyFromPassword( |
- crypto::SymmetricKey::AES, "password", "saltiest", 1000, 256)) { |
- std::string iv("the iv: 16 bytes"); |
- encryptor_.Init(key_.get(), crypto::Encryptor::CBC, iv); |
-} |
- |
-bool CookieCryptor::EncryptString(const std::string& plaintext, |
- std::string* ciphertext) { |
- return encryptor_.Encrypt(plaintext, ciphertext); |
-} |
- |
-bool CookieCryptor::DecryptString(const std::string& ciphertext, |
- std::string* plaintext) { |
- return encryptor_.Decrypt(ciphertext, plaintext); |
-} |
- |
-} // namespace |
- |
-typedef std::vector<net::CanonicalCookie*> CanonicalCookieVector; |
- |
-class SQLitePersistentCookieStoreTest : public testing::Test { |
- public: |
- SQLitePersistentCookieStoreTest() |
- : pool_owner_(new base::SequencedWorkerPoolOwner(3, "Background Pool")), |
- loaded_event_(false, false), |
- key_loaded_event_(false, false), |
- db_thread_event_(false, false) { |
- } |
- |
- void OnLoaded(const CanonicalCookieVector& cookies) { |
- cookies_ = cookies; |
- loaded_event_.Signal(); |
- } |
- |
- void OnKeyLoaded(const CanonicalCookieVector& cookies) { |
- cookies_ = cookies; |
- key_loaded_event_.Signal(); |
- } |
- |
- void Load(CanonicalCookieVector* cookies) { |
- EXPECT_FALSE(loaded_event_.IsSignaled()); |
- store_->Load(base::Bind(&SQLitePersistentCookieStoreTest::OnLoaded, |
- base::Unretained(this))); |
- loaded_event_.Wait(); |
- *cookies = cookies_; |
- } |
- |
- void Flush() { |
- base::WaitableEvent event(false, false); |
- store_->Flush(base::Bind(&base::WaitableEvent::Signal, |
- base::Unretained(&event))); |
- event.Wait(); |
- } |
- |
- scoped_refptr<base::SequencedTaskRunner> background_task_runner() { |
- return pool_owner_->pool()->GetSequencedTaskRunner( |
- pool_owner_->pool()->GetNamedSequenceToken("background")); |
- } |
- |
- scoped_refptr<base::SequencedTaskRunner> client_task_runner() { |
- return pool_owner_->pool()->GetSequencedTaskRunner( |
- pool_owner_->pool()->GetNamedSequenceToken("client")); |
- } |
- |
- void DestroyStore() { |
- store_ = NULL; |
- // Make sure we wait until the destructor has run by shutting down the pool |
- // resetting the owner (whose destructor blocks on the pool completion). |
- pool_owner_->pool()->Shutdown(); |
- // Create a new pool for the few tests that create multiple stores. In other |
- // cases this is wasted but harmless. |
- pool_owner_.reset(new base::SequencedWorkerPoolOwner(3, "Background Pool")); |
- } |
- |
- void CreateAndLoad(bool crypt_cookies, |
- bool restore_old_session_cookies, |
- CanonicalCookieVector* cookies) { |
- if (crypt_cookies) |
- cookie_crypto_delegate_.reset(new CookieCryptor()); |
- |
- store_ = new SQLitePersistentCookieStore( |
- temp_dir_.path().Append(kCookieFilename), |
- client_task_runner(), |
- background_task_runner(), |
- restore_old_session_cookies, |
- NULL, |
- cookie_crypto_delegate_.get()); |
- Load(cookies); |
- } |
- |
- void InitializeStore(bool crypt, bool restore_old_session_cookies) { |
- CanonicalCookieVector cookies; |
- CreateAndLoad(crypt, restore_old_session_cookies, &cookies); |
- EXPECT_EQ(0U, cookies.size()); |
- } |
- |
- // We have to create this method to wrap WaitableEvent::Wait, since we cannot |
- // bind a non-void returning method as a Closure. |
- void WaitOnDBEvent() { |
- db_thread_event_.Wait(); |
- } |
- |
- // Adds a persistent cookie to store_. |
- void AddCookie(const std::string& name, |
- const std::string& value, |
- const std::string& domain, |
- const std::string& path, |
- const base::Time& creation) { |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), name, value, domain, path, creation, creation, creation, false, |
- false, false, net::COOKIE_PRIORITY_DEFAULT)); |
- } |
- |
- void AddCookieWithExpiration(const std::string& name, |
- const std::string& value, |
- const std::string& domain, |
- const std::string& path, |
- const base::Time& creation, |
- const base::Time& expiration) { |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), name, value, domain, path, creation, expiration, creation, |
- false, false, false, net::COOKIE_PRIORITY_DEFAULT)); |
- } |
- |
- std::string ReadRawDBContents() { |
- std::string contents; |
- if (!base::ReadFileToString(temp_dir_.path().Append(kCookieFilename), |
- &contents)) |
- return std::string(); |
- return contents; |
- } |
- |
- void SetUp() override { ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); } |
- |
- void TearDown() override { |
- DestroyStore(); |
- pool_owner_->pool()->Shutdown(); |
- } |
- |
- protected: |
- base::MessageLoop main_loop_; |
- scoped_ptr<base::SequencedWorkerPoolOwner> pool_owner_; |
- base::WaitableEvent loaded_event_; |
- base::WaitableEvent key_loaded_event_; |
- base::WaitableEvent db_thread_event_; |
- CanonicalCookieVector cookies_; |
- base::ScopedTempDir temp_dir_; |
- scoped_refptr<SQLitePersistentCookieStore> store_; |
- scoped_ptr<net::CookieCryptoDelegate> cookie_crypto_delegate_; |
-}; |
- |
-TEST_F(SQLitePersistentCookieStoreTest, TestInvalidMetaTableRecovery) { |
- InitializeStore(false, false); |
- AddCookie("A", "B", "foo.bar", "/", base::Time::Now()); |
- DestroyStore(); |
- |
- // Load up the store and verify that it has good data in it. |
- CanonicalCookieVector cookies; |
- CreateAndLoad(false, false, &cookies); |
- ASSERT_EQ(1U, cookies.size()); |
- ASSERT_STREQ("foo.bar", cookies[0]->Domain().c_str()); |
- ASSERT_STREQ("A", cookies[0]->Name().c_str()); |
- ASSERT_STREQ("B", cookies[0]->Value().c_str()); |
- DestroyStore(); |
- STLDeleteElements(&cookies); |
- |
- // Now corrupt the meta table. |
- { |
- sql::Connection db; |
- ASSERT_TRUE(db.Open(temp_dir_.path().Append(kCookieFilename))); |
- sql::MetaTable meta_table_; |
- meta_table_.Init(&db, 1, 1); |
- ASSERT_TRUE(db.Execute("DELETE FROM meta")); |
- db.Close(); |
- } |
- |
- // Upon loading, the database should be reset to a good, blank state. |
- CreateAndLoad(false, false, &cookies); |
- ASSERT_EQ(0U, cookies.size()); |
- |
- // Verify that, after, recovery, the database persists properly. |
- AddCookie("X", "Y", "foo.bar", "/", base::Time::Now()); |
- DestroyStore(); |
- CreateAndLoad(false, false, &cookies); |
- ASSERT_EQ(1U, cookies.size()); |
- ASSERT_STREQ("foo.bar", cookies[0]->Domain().c_str()); |
- ASSERT_STREQ("X", cookies[0]->Name().c_str()); |
- ASSERT_STREQ("Y", cookies[0]->Value().c_str()); |
- STLDeleteElements(&cookies); |
-} |
- |
-// Test if data is stored as expected in the SQLite database. |
-TEST_F(SQLitePersistentCookieStoreTest, TestPersistance) { |
- InitializeStore(false, false); |
- AddCookie("A", "B", "foo.bar", "/", base::Time::Now()); |
- // Replace the store effectively destroying the current one and forcing it |
- // to write its data to disk. Then we can see if after loading it again it |
- // is still there. |
- DestroyStore(); |
- // Reload and test for persistence |
- CanonicalCookieVector cookies; |
- CreateAndLoad(false, false, &cookies); |
- ASSERT_EQ(1U, cookies.size()); |
- ASSERT_STREQ("foo.bar", cookies[0]->Domain().c_str()); |
- ASSERT_STREQ("A", cookies[0]->Name().c_str()); |
- ASSERT_STREQ("B", cookies[0]->Value().c_str()); |
- |
- // Now delete the cookie and check persistence again. |
- store_->DeleteCookie(*cookies[0]); |
- DestroyStore(); |
- STLDeleteElements(&cookies); |
- |
- // Reload and check if the cookie has been removed. |
- CreateAndLoad(false, false, &cookies); |
- ASSERT_EQ(0U, cookies.size()); |
-} |
- |
-TEST_F(SQLitePersistentCookieStoreTest, TestSessionCookiesDeletedOnStartup) { |
- // Initialize the cookie store with 3 persistent cookies, 5 transient |
- // cookies. |
- InitializeStore(false, false); |
- |
- // Add persistent cookies. |
- base::Time t = base::Time::Now(); |
- AddCookie("A", "B", "a1.com", "/", t); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookie("A", "B", "a2.com", "/", t); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookie("A", "B", "a3.com", "/", t); |
- |
- // Add transient cookies. |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookieWithExpiration("A", "B", "b1.com", "/", t, base::Time()); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookieWithExpiration("A", "B", "b2.com", "/", t, base::Time()); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookieWithExpiration("A", "B", "b3.com", "/", t, base::Time()); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookieWithExpiration("A", "B", "b4.com", "/", t, base::Time()); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookieWithExpiration("A", "B", "b5.com", "/", t, base::Time()); |
- DestroyStore(); |
- |
- // Load the store a second time. Before the store finishes loading, add a |
- // transient cookie and flush it to disk. |
- store_ = new SQLitePersistentCookieStore( |
- temp_dir_.path().Append(kCookieFilename), |
- client_task_runner(), |
- background_task_runner(), |
- false, NULL, NULL); |
- |
- // Posting a blocking task to db_thread_ makes sure that the DB thread waits |
- // until both Load and Flush have been posted to its task queue. |
- background_task_runner()->PostTask( |
- FROM_HERE, |
- base::Bind(&SQLitePersistentCookieStoreTest::WaitOnDBEvent, |
- base::Unretained(this))); |
- store_->Load(base::Bind(&SQLitePersistentCookieStoreTest::OnLoaded, |
- base::Unretained(this))); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookieWithExpiration("A", "B", "c.com", "/", t, base::Time()); |
- base::WaitableEvent event(false, false); |
- store_->Flush(base::Bind(&base::WaitableEvent::Signal, |
- base::Unretained(&event))); |
- |
- // Now the DB-thread queue contains: |
- // (active:) |
- // 1. Wait (on db_event) |
- // (pending:) |
- // 2. "Init And Chain-Load First Domain" |
- // 3. Add Cookie (c.com) |
- // 4. Flush Cookie (c.com) |
- db_thread_event_.Signal(); |
- event.Wait(); |
- loaded_event_.Wait(); |
- STLDeleteElements(&cookies_); |
- DestroyStore(); |
- |
- // Load the store a third time, this time restoring session cookies. The |
- // store should contain exactly 4 cookies: the 3 persistent, and "c.com", |
- // which was added during the second cookie store load. |
- store_ = new SQLitePersistentCookieStore( |
- temp_dir_.path().Append(kCookieFilename), |
- client_task_runner(), |
- background_task_runner(), |
- true, NULL, NULL); |
- store_->Load(base::Bind(&SQLitePersistentCookieStoreTest::OnLoaded, |
- base::Unretained(this))); |
- loaded_event_.Wait(); |
- ASSERT_EQ(4u, cookies_.size()); |
- STLDeleteElements(&cookies_); |
-} |
- |
-// Test that priority load of cookies for a specfic domain key could be |
-// completed before the entire store is loaded |
-TEST_F(SQLitePersistentCookieStoreTest, TestLoadCookiesForKey) { |
- InitializeStore(false, false); |
- base::Time t = base::Time::Now(); |
- AddCookie("A", "B", "foo.bar", "/", t); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookie("A", "B", "www.aaa.com", "/", t); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookie("A", "B", "travel.aaa.com", "/", t); |
- t += base::TimeDelta::FromInternalValue(10); |
- AddCookie("A", "B", "www.bbb.com", "/", t); |
- DestroyStore(); |
- |
- store_ = new SQLitePersistentCookieStore( |
- temp_dir_.path().Append(kCookieFilename), |
- client_task_runner(), |
- background_task_runner(), |
- false, NULL, NULL); |
- |
- // Posting a blocking task to db_thread_ makes sure that the DB thread waits |
- // until both Load and LoadCookiesForKey have been posted to its task queue. |
- background_task_runner()->PostTask( |
- FROM_HERE, |
- base::Bind(&SQLitePersistentCookieStoreTest::WaitOnDBEvent, |
- base::Unretained(this))); |
- store_->Load(base::Bind(&SQLitePersistentCookieStoreTest::OnLoaded, |
- base::Unretained(this))); |
- store_->LoadCookiesForKey("aaa.com", |
- base::Bind(&SQLitePersistentCookieStoreTest::OnKeyLoaded, |
- base::Unretained(this))); |
- background_task_runner()->PostTask( |
- FROM_HERE, |
- base::Bind(&SQLitePersistentCookieStoreTest::WaitOnDBEvent, |
- base::Unretained(this))); |
- |
- // Now the DB-thread queue contains: |
- // (active:) |
- // 1. Wait (on db_event) |
- // (pending:) |
- // 2. "Init And Chain-Load First Domain" |
- // 3. Priority Load (aaa.com) |
- // 4. Wait (on db_event) |
- db_thread_event_.Signal(); |
- key_loaded_event_.Wait(); |
- ASSERT_EQ(loaded_event_.IsSignaled(), false); |
- std::set<std::string> cookies_loaded; |
- for (CanonicalCookieVector::const_iterator it = cookies_.begin(); |
- it != cookies_.end(); |
- ++it) { |
- cookies_loaded.insert((*it)->Domain().c_str()); |
- } |
- STLDeleteElements(&cookies_); |
- ASSERT_GT(4U, cookies_loaded.size()); |
- ASSERT_EQ(true, cookies_loaded.find("www.aaa.com") != cookies_loaded.end()); |
- ASSERT_EQ(true, |
- cookies_loaded.find("travel.aaa.com") != cookies_loaded.end()); |
- |
- db_thread_event_.Signal(); |
- loaded_event_.Wait(); |
- for (CanonicalCookieVector::const_iterator it = cookies_.begin(); |
- it != cookies_.end(); |
- ++it) { |
- cookies_loaded.insert((*it)->Domain().c_str()); |
- } |
- ASSERT_EQ(4U, cookies_loaded.size()); |
- ASSERT_EQ(cookies_loaded.find("foo.bar") != cookies_loaded.end(), |
- true); |
- ASSERT_EQ(cookies_loaded.find("www.bbb.com") != cookies_loaded.end(), true); |
- STLDeleteElements(&cookies_); |
-} |
- |
-// Test that we can force the database to be written by calling Flush(). |
-TEST_F(SQLitePersistentCookieStoreTest, TestFlush) { |
- InitializeStore(false, false); |
- // File timestamps don't work well on all platforms, so we'll determine |
- // whether the DB file has been modified by checking its size. |
- base::FilePath path = temp_dir_.path().Append(kCookieFilename); |
- base::File::Info info; |
- ASSERT_TRUE(base::GetFileInfo(path, &info)); |
- int64 base_size = info.size; |
- |
- // Write some large cookies, so the DB will have to expand by several KB. |
- for (char c = 'a'; c < 'z'; ++c) { |
- // Each cookie needs a unique timestamp for creation_utc (see DB schema). |
- base::Time t = base::Time::Now() + base::TimeDelta::FromMicroseconds(c); |
- std::string name(1, c); |
- std::string value(1000, c); |
- AddCookie(name, value, "foo.bar", "/", t); |
- } |
- |
- Flush(); |
- |
- // We forced a write, so now the file will be bigger. |
- ASSERT_TRUE(base::GetFileInfo(path, &info)); |
- ASSERT_GT(info.size, base_size); |
-} |
- |
-// Test loading old session cookies from the disk. |
-TEST_F(SQLitePersistentCookieStoreTest, TestLoadOldSessionCookies) { |
- InitializeStore(false, true); |
- |
- // Add a session cookie. |
- store_->AddCookie(net::CanonicalCookie(GURL(), "C", "D", "sessioncookie.com", |
- "/", base::Time::Now(), base::Time(), |
- base::Time::Now(), false, false, false, |
- net::COOKIE_PRIORITY_DEFAULT)); |
- |
- // Force the store to write its data to the disk. |
- DestroyStore(); |
- |
- // Create a store that loads session cookies and test that the session cookie |
- // was loaded. |
- CanonicalCookieVector cookies; |
- CreateAndLoad(false, true, &cookies); |
- |
- ASSERT_EQ(1U, cookies.size()); |
- ASSERT_STREQ("sessioncookie.com", cookies[0]->Domain().c_str()); |
- ASSERT_STREQ("C", cookies[0]->Name().c_str()); |
- ASSERT_STREQ("D", cookies[0]->Value().c_str()); |
- ASSERT_EQ(net::COOKIE_PRIORITY_DEFAULT, cookies[0]->Priority()); |
- |
- STLDeleteElements(&cookies); |
-} |
- |
-// Test loading old session cookies from the disk. |
-TEST_F(SQLitePersistentCookieStoreTest, TestDontLoadOldSessionCookies) { |
- InitializeStore(false, true); |
- |
- // Add a session cookie. |
- store_->AddCookie(net::CanonicalCookie(GURL(), "C", "D", "sessioncookie.com", |
- "/", base::Time::Now(), base::Time(), |
- base::Time::Now(), false, false, false, |
- net::COOKIE_PRIORITY_DEFAULT)); |
- |
- // Force the store to write its data to the disk. |
- DestroyStore(); |
- |
- // Create a store that doesn't load old session cookies and test that the |
- // session cookie was not loaded. |
- CanonicalCookieVector cookies; |
- CreateAndLoad(false, false, &cookies); |
- ASSERT_EQ(0U, cookies.size()); |
- |
- // The store should also delete the session cookie. Wait until that has been |
- // done. |
- DestroyStore(); |
- |
- // Create a store that loads old session cookies and test that the session |
- // cookie is gone. |
- CreateAndLoad(false, true, &cookies); |
- ASSERT_EQ(0U, cookies.size()); |
-} |
- |
-TEST_F(SQLitePersistentCookieStoreTest, PersistIsPersistent) { |
- InitializeStore(false, true); |
- static const char kSessionName[] = "session"; |
- static const char kPersistentName[] = "persistent"; |
- |
- // Add a session cookie. |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), kSessionName, "val", "sessioncookie.com", "/", base::Time::Now(), |
- base::Time(), base::Time::Now(), false, false, false, |
- net::COOKIE_PRIORITY_DEFAULT)); |
- // Add a persistent cookie. |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), kPersistentName, "val", "sessioncookie.com", "/", |
- base::Time::Now() - base::TimeDelta::FromDays(1), |
- base::Time::Now() + base::TimeDelta::FromDays(1), base::Time::Now(), |
- false, false, false, net::COOKIE_PRIORITY_DEFAULT)); |
- |
- // Force the store to write its data to the disk. |
- DestroyStore(); |
- |
- // Create a store that loads session cookie and test that the IsPersistent |
- // attribute is restored. |
- CanonicalCookieVector cookies; |
- CreateAndLoad(false, true, &cookies); |
- ASSERT_EQ(2U, cookies.size()); |
- |
- std::map<std::string, net::CanonicalCookie*> cookie_map; |
- for (CanonicalCookieVector::const_iterator it = cookies.begin(); |
- it != cookies.end(); |
- ++it) { |
- cookie_map[(*it)->Name()] = *it; |
- } |
- |
- std::map<std::string, net::CanonicalCookie*>::const_iterator it = |
- cookie_map.find(kSessionName); |
- ASSERT_TRUE(it != cookie_map.end()); |
- EXPECT_FALSE(cookie_map[kSessionName]->IsPersistent()); |
- |
- it = cookie_map.find(kPersistentName); |
- ASSERT_TRUE(it != cookie_map.end()); |
- EXPECT_TRUE(cookie_map[kPersistentName]->IsPersistent()); |
- |
- STLDeleteElements(&cookies); |
-} |
- |
-TEST_F(SQLitePersistentCookieStoreTest, PriorityIsPersistent) { |
- static const char kLowName[] = "low"; |
- static const char kMediumName[] = "medium"; |
- static const char kHighName[] = "high"; |
- static const char kCookieDomain[] = "sessioncookie.com"; |
- static const char kCookieValue[] = "value"; |
- static const char kCookiePath[] = "/"; |
- |
- InitializeStore(false, true); |
- |
- // Add a low-priority persistent cookie. |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), kLowName, kCookieValue, kCookieDomain, kCookiePath, |
- base::Time::Now() - base::TimeDelta::FromMinutes(1), |
- base::Time::Now() + base::TimeDelta::FromDays(1), base::Time::Now(), |
- false, false, false, net::COOKIE_PRIORITY_LOW)); |
- |
- // Add a medium-priority persistent cookie. |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), kMediumName, kCookieValue, kCookieDomain, kCookiePath, |
- base::Time::Now() - base::TimeDelta::FromMinutes(2), |
- base::Time::Now() + base::TimeDelta::FromDays(1), base::Time::Now(), |
- false, false, false, net::COOKIE_PRIORITY_MEDIUM)); |
- |
- // Add a high-priority peristent cookie. |
- store_->AddCookie(net::CanonicalCookie( |
- GURL(), kHighName, kCookieValue, kCookieDomain, kCookiePath, |
- base::Time::Now() - base::TimeDelta::FromMinutes(3), |
- base::Time::Now() + base::TimeDelta::FromDays(1), base::Time::Now(), |
- false, false, false, net::COOKIE_PRIORITY_HIGH)); |
- |
- // Force the store to write its data to the disk. |
- DestroyStore(); |
- |
- // Create a store that loads session cookie and test that the priority |
- // attribute values are restored. |
- CanonicalCookieVector cookies; |
- CreateAndLoad(false, true, &cookies); |
- ASSERT_EQ(3U, cookies.size()); |
- |
- // Put the cookies into a map, by name, so we can easily find them. |
- std::map<std::string, net::CanonicalCookie*> cookie_map; |
- for (CanonicalCookieVector::const_iterator it = cookies.begin(); |
- it != cookies.end(); |
- ++it) { |
- cookie_map[(*it)->Name()] = *it; |
- } |
- |
- // Validate that each cookie has the correct priority. |
- std::map<std::string, net::CanonicalCookie*>::const_iterator it = |
- cookie_map.find(kLowName); |
- ASSERT_TRUE(it != cookie_map.end()); |
- EXPECT_EQ(net::COOKIE_PRIORITY_LOW, cookie_map[kLowName]->Priority()); |
- |
- it = cookie_map.find(kMediumName); |
- ASSERT_TRUE(it != cookie_map.end()); |
- EXPECT_EQ(net::COOKIE_PRIORITY_MEDIUM, cookie_map[kMediumName]->Priority()); |
- |
- it = cookie_map.find(kHighName); |
- ASSERT_TRUE(it != cookie_map.end()); |
- EXPECT_EQ(net::COOKIE_PRIORITY_HIGH, cookie_map[kHighName]->Priority()); |
- |
- STLDeleteElements(&cookies); |
-} |
- |
-TEST_F(SQLitePersistentCookieStoreTest, UpdateToEncryption) { |
- CanonicalCookieVector cookies; |
- |
- // Create unencrypted cookie store and write something to it. |
- InitializeStore(false, false); |
- AddCookie("name", "value123XYZ", "foo.bar", "/", base::Time::Now()); |
- DestroyStore(); |
- |
- // Verify that "value" is visible in the file. This is necessary in order to |
- // have confidence in a later test that "encrypted_value" is not visible. |
- std::string contents = ReadRawDBContents(); |
- EXPECT_NE(0U, contents.length()); |
- EXPECT_NE(contents.find("value123XYZ"), std::string::npos); |
- |
- // Create encrypted cookie store and ensure old cookie still reads. |
- STLDeleteElements(&cookies_); |
- EXPECT_EQ(0U, cookies_.size()); |
- CreateAndLoad(true, false, &cookies); |
- EXPECT_EQ(1U, cookies_.size()); |
- EXPECT_EQ("name", cookies_[0]->Name()); |
- EXPECT_EQ("value123XYZ", cookies_[0]->Value()); |
- |
- // Make sure we can update existing cookie and add new cookie as encrypted. |
- store_->DeleteCookie(*(cookies_[0])); |
- AddCookie("name", "encrypted_value123XYZ", "foo.bar", "/", base::Time::Now()); |
- AddCookie("other", "something456ABC", "foo.bar", "/", |
- base::Time::Now() + base::TimeDelta::FromInternalValue(10)); |
- DestroyStore(); |
- STLDeleteElements(&cookies_); |
- CreateAndLoad(true, false, &cookies); |
- EXPECT_EQ(2U, cookies_.size()); |
- net::CanonicalCookie* cookie_name = NULL; |
- net::CanonicalCookie* cookie_other = NULL; |
- if (cookies_[0]->Name() == "name") { |
- cookie_name = cookies_[0]; |
- cookie_other = cookies_[1]; |
- } else { |
- cookie_name = cookies_[1]; |
- cookie_other = cookies_[0]; |
- } |
- EXPECT_EQ("encrypted_value123XYZ", cookie_name->Value()); |
- EXPECT_EQ("something456ABC", cookie_other->Value()); |
- DestroyStore(); |
- STLDeleteElements(&cookies_); |
- |
- // Examine the real record to make sure plaintext version doesn't exist. |
- sql::Connection db; |
- sql::Statement smt; |
- int resultcount = 0; |
- ASSERT_TRUE(db.Open(temp_dir_.path().Append(kCookieFilename))); |
- smt.Assign(db.GetCachedStatement(SQL_FROM_HERE, |
- "SELECT * " |
- "FROM cookies " |
- "WHERE host_key = 'foo.bar'")); |
- while (smt.Step()) { |
- resultcount++; |
- for (int i=0; i < smt.ColumnCount(); i++) { |
- EXPECT_EQ(smt.ColumnString(i).find("value"), std::string::npos); |
- EXPECT_EQ(smt.ColumnString(i).find("something"), std::string::npos); |
- } |
- } |
- EXPECT_EQ(2, resultcount); |
- |
- // Verify that "encrypted_value" is NOT visible in the file. |
- contents = ReadRawDBContents(); |
- EXPECT_NE(0U, contents.length()); |
- EXPECT_EQ(contents.find("encrypted_value123XYZ"), std::string::npos); |
- EXPECT_EQ(contents.find("something456ABC"), std::string::npos); |
-} |
- |
-} // namespace content |