SessionRestore: Store session cookies and restore them if chrome crashes or auto-restarts.

chrome/browser/net/sqlite_persistent_cookie_store_unittest.cc

 // Copyright (c) 2011 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 "base/bind.h"
 #include "base/file_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop.h"
 #include "base/scoped_temp_dir.h"
 #include "base/stl_util.h"
@@ skipping 43 matching lines @@
   void WaitOnDBEvent() {
     db_thread_event_.Wait();
   }
 
   virtual void SetUp() {
     ui_thread_.Start();
     db_thread_.Start();
     io_thread_.Start();
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     store_ = new SQLitePersistentCookieStore(
-        temp_dir_.path().Append(chrome::kCookieFilename));
+        temp_dir_.path().Append(chrome::kCookieFilename), false);
     std::vector<net::CookieMonster::CanonicalCookie*> cookies;
     Load(&cookies);
     ASSERT_EQ(0u, cookies.size());
     // Make sure the store gets written at least once.
     store_->AddCookie(
         net::CookieMonster::CanonicalCookie(GURL(), "A", "B", "http://foo.bar",
                                             "/", std::string(), std::string(),
                                             base::Time::Now(),
                                             base::Time::Now(),
                                             base::Time::Now(),
@@ skipping 49 matching lines @@
   // Replace the store effectively destroying the current one and forcing it
   // to write it's data to disk. Then we can see if after loading it again it
   // is still there.
   store_ = NULL;
   scoped_refptr<base::ThreadTestHelper> helper(
       new base::ThreadTestHelper(
           BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
   // Make sure we wait until the destructor has run.
   ASSERT_TRUE(helper->Run());
   store_ = new SQLitePersistentCookieStore(
-      temp_dir_.path().Append(chrome::kCookieFilename));
+      temp_dir_.path().Append(chrome::kCookieFilename), false);
 
   // Reload and test for persistence
   Load(&cookies);
   ASSERT_EQ(1U, cookies.size());
   ASSERT_STREQ("http://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]);
   store_ = NULL;
   // Make sure we wait until the destructor has run.
   ASSERT_TRUE(helper->Run());
   STLDeleteContainerPointers(cookies.begin(), cookies.end());
   cookies.clear();
   store_ = new SQLitePersistentCookieStore(
-      temp_dir_.path().Append(chrome::kCookieFilename));
+      temp_dir_.path().Append(chrome::kCookieFilename), false);
 
   // Reload and check if the cookie has been removed.
   Load(&cookies);
   ASSERT_EQ(0U, cookies.size());
 }
 
 // Test that priority load of cookies for a specfic domain key could be
 // completed before the entire store is loaded
 TEST_F(SQLitePersistentCookieStoreTest, TestLoadCookiesForKey) {
   base::Time t = base::Time::Now() + base::TimeDelta::FromInternalValue(10);
@@ skipping 17 matching lines @@
                                         t, t, t, false, false, true, true));
   store_ = NULL;
 
   scoped_refptr<base::ThreadTestHelper> helper(
       new base::ThreadTestHelper(
           BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
   // Make sure we wait until the destructor has run.
   ASSERT_TRUE(helper->Run());
 
   store_ = new SQLitePersistentCookieStore(
-    temp_dir_.path().Append(chrome::kCookieFilename));
+      temp_dir_.path().Append(chrome::kCookieFilename), false);
   // 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.
   BrowserThread::PostTask(
       BrowserThread::DB, 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,
@@ skipping 95 matching lines @@
 
   store_->Flush(NewRunnableMethod(counter.get(), &CallbackCounter::Callback));
 
   scoped_refptr<base::ThreadTestHelper> helper(
       new base::ThreadTestHelper(
           BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
   ASSERT_TRUE(helper->Run());
 
   ASSERT_EQ(1, counter->callback_count());
 }
+
+// Test loading old session cookies from the disk.
+TEST_F(SQLitePersistentCookieStoreTest, TestLoadOldSessionCookies) {
+  std::vector<net::CookieMonster::CanonicalCookie*> cookies;
+
+  // Use a separate cookie store for this test.
+  ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), true);
+  // Make sure the database is initialized.
+  Load(&cookies);
+  ASSERT_EQ(0u, cookies.size());
+
+  // Add a session cookie.
+  store_->AddCookie(
+      net::CookieMonster::CanonicalCookie(
+          GURL(), "C", "D", "http://sessioncookie.com", "/", std::string(),
+          std::string(), base::Time::Now(), base::Time::Now(),
+          base::Time::Now(), false, false, true, false /*is_persistent*/));
+
+  // Force the store to write its data to the disk.
+  store_ = NULL;
+  scoped_refptr<base::ThreadTestHelper> helper(
+      new base::ThreadTestHelper(
+          BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
+  // Make sure we wait until the destructor has run.
+  ASSERT_TRUE(helper->Run());
+
+  // Create a store that loads session cookies.
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), true);
+
+  // Reload and test that the session cookie was loaded.
+  Load(&cookies);
+  ASSERT_EQ(1U, cookies.size());
+  ASSERT_STREQ("http://sessioncookie.com", cookies[0]->Domain().c_str());
+  ASSERT_STREQ("C", cookies[0]->Name().c_str());
+  ASSERT_STREQ("D", cookies[0]->Value().c_str());
+}
+
+// Test loading old session cookies from the disk.
+TEST_F(SQLitePersistentCookieStoreTest, TestDontLoadOldSessionCookies) {
+  std::vector<net::CookieMonster::CanonicalCookie*> cookies;
+  // Use a separate cookie store for this test.
+  ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), true);
+  // Make sure the database is initialized.
+  Load(&cookies);
+  ASSERT_EQ(0u, cookies.size());
+
+  // Add a session cookie.
+  store_->AddCookie(
+      net::CookieMonster::CanonicalCookie(
+          GURL(), "C", "D", "http://sessioncookie.com", "/", std::string(),
+          std::string(), base::Time::Now(), base::Time::Now(),
+          base::Time::Now(), false, false, true, false /*is_persistent*/));
+
+  // Force the store to write its data to the disk.
+  store_ = NULL;
+  scoped_refptr<base::ThreadTestHelper> helper(
+      new base::ThreadTestHelper(
+          BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
+  // Make sure we wait until the destructor has run.
+  ASSERT_TRUE(helper->Run());
+
+  // Create a store that doesn't load old session cookies.
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), false);
+
+  // Reload and test that the session cookie was not loaded.
+  Load(&cookies);
+  ASSERT_EQ(0U, cookies.size());
+
+  // The store should also delete the session cookie. Wait until that has been
+  // done.
+  store_ = NULL;
+  helper = new base::ThreadTestHelper(
+      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB));
+  ASSERT_TRUE(helper->Run());
+
+  // Create a store that loads old session cookies.
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), true);
+
+  // Reload and test that the session cookie is gone.
+  Load(&cookies);
+  ASSERT_EQ(0U, cookies.size());
+}
+
+TEST_F(SQLitePersistentCookieStoreTest, PersistHasExpiresAndIsPersistent) {
+  std::vector<net::CookieMonster::CanonicalCookie*> cookies;
+
+  // Use a separate cookie store for this test.
+  ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), true);
+  // Make sure the database is initialized.
+  Load(&cookies);
+  ASSERT_EQ(0u, cookies.size());
+
+  // Add a session cookie with has_expires = false, and another session cookie
+  // with has_expires = true.
+  store_->AddCookie(
+      net::CookieMonster::CanonicalCookie(
+          GURL(), "session-hasexpires", "val", "http://sessioncookie.com", "/",
+          std::string(), std::string(),
+          base::Time::Now() - base::TimeDelta::FromDays(3), base::Time::Now(),
+          base::Time::Now(), false, false, true /* has_expires */,
+          false /* is_persistent */));
+  store_->AddCookie(
+      net::CookieMonster::CanonicalCookie(
+          GURL(), "session-noexpires", "val", "http://sessioncookie.com", "/",
+          std::string(), std::string(),
+          base::Time::Now() - base::TimeDelta::FromDays(2), base::Time::Now(),
+          base::Time::Now(), false, false, false /* has_expires */,
+          false /* is_persistent */));
+  // Add a persistent cookie.
+  store_->AddCookie(
+      net::CookieMonster::CanonicalCookie(
+          GURL(), "persistent", "val", "http://sessioncookie.com", "/",
+          std::string(), std::string(),
+          base::Time::Now() - base::TimeDelta::FromDays(1), base::Time::Now(),
+          base::Time::Now(), false, false, true /* has_expires */,
+          true /* is_persistent */));
+
+  // Force the store to write its data to the disk.
+  store_ = NULL;
+  scoped_refptr<base::ThreadTestHelper> helper(
+      new base::ThreadTestHelper(
+          BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)));
+  // Make sure we wait until the destructor has run.
+  ASSERT_TRUE(helper->Run());
+
+  // Create a store that loads session cookies.
+  store_ = new SQLitePersistentCookieStore(
+      temp_dir.path().Append(chrome::kCookieFilename), true);
+
+  // Reload and test that the the DoesExpire and IsPersistent attributes are
+  // restored.
+  Load(&cookies);
+  ASSERT_EQ(3U, cookies.size());
+
+  std::map<std::string, net::CookieMonster::CanonicalCookie*> cookie_map;
+  std::vector<net::CookieMonster::CanonicalCookie*>::const_iterator it;
+  for (it = cookies.begin(); it != cookies.end(); ++it)
+    cookie_map[(*it)->Name()] = *it;
+
+  EXPECT_TRUE(cookie_map["session-hasexpires"]->DoesExpire());
+  EXPECT_FALSE(cookie_map["session-hasexpires"]->IsPersistent());
+
+  EXPECT_FALSE(cookie_map["session-noexpires"]->DoesExpire());
+  EXPECT_FALSE(cookie_map["session-noexpires"]->IsPersistent());
+
+  EXPECT_TRUE(cookie_map["persistent"]->DoesExpire());
+  EXPECT_TRUE(cookie_map["persistent"]->IsPersistent());
+}