Unittest for CookieStore::DeleteAllCreatedBetweenAsync().

net/base/cookie_store_unittest.h

 // 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.
 
 #ifndef NET_BASE_COOKIE_STORE_UNITTEST_H_
 #define NET_BASE_COOKIE_STORE_UNITTEST_H_
 #pragma once
 
 #include "base/bind.h"
 #include "base/message_loop.h"
@@ skipping 144 matching lines @@
                     const std::string& cookie_name) {
     DCHECK(cs);
     DeleteCookieCallback callback;
     cs->DeleteCookieAsync(
         url, cookie_name,
         base::Bind(&DeleteCookieCallback::Run, base::Unretained(&callback)));
     RunFor(kTimeout);
     EXPECT_TRUE(callback.did_run());
   }
 
+  int DeleteCreatedBetween(CookieStore* cs,
+                            const base::Time& delete_begin,
+                            const base::Time& delete_end) {
+    DCHECK(cs);
+    DeleteCallback callback;
+    cs->DeleteAllCreatedBetweenAsync(
+        delete_begin, delete_end,
+        base::Bind(&DeleteCallback::Run, base::Unretained(&callback)));
+    RunFor(kTimeout);
+    EXPECT_TRUE(callback.did_run());
+    return callback.num_deleted();
+  }
+
   void RunFor(int ms) {
     // Runs the test thread message loop for up to |ms| milliseconds.
     MessageLoop::current()->PostDelayedTask(
         FROM_HERE, base::Bind(&MessageLoop::Quit, weak_factory_.GetWeakPtr()),
         base::TimeDelta::FromMilliseconds(ms));
     MessageLoop::current()->Run();
     weak_factory_.InvalidateWeakPtrs();
   }
 
   scoped_refptr<CookieStore> GetCookieStore() {
@@ skipping 496 matching lines @@
                               std::string(kValidCookieLine) +
                               "; expires=Mon, 18-Apr-22 22:50:13 GMT"));
   this->MatchCookieLines("A=B", this->GetCookies(cs, this->url_google_));
   // Delete it via Expires, with a unix epoch of 0.
   EXPECT_TRUE(this->SetCookie(cs, this->url_google_,
                               std::string(kValidCookieLine) +
                               "; expires=Thu, 1-Jan-1970 00:00:00 GMT"));
   this->MatchCookieLines("", this->GetCookies(cs, this->url_google_));
 }
 
+TYPED_TEST_P(CookieStoreTest, TestDeleteAllCreatedBetween) {
+  scoped_refptr<CookieStore> cs(this->GetCookieStore());
+  const base::Time last_month = base::Time::Now() -
+                                base::TimeDelta::FromDays(30);
+  const base::Time last_minute = base::Time::Now() -
+                                 base::TimeDelta::FromMinutes(1);
+  const base::Time next_minute = base::Time::Now() +
+                                 base::TimeDelta::FromMinutes(1);
+  const base::Time next_month = base::Time::Now() +
+                                base::TimeDelta::FromDays(30);
+
+  // Add a cookie.
+  EXPECT_TRUE(this->SetCookie(cs, this->url_google_, "A=B"));
+   // Check that the cookie is in the store.
+  this->MatchCookieLines("A=B", this->GetCookies(cs, this->url_google_));
+
+  // Remove cookies in empty intervals.
+  EXPECT_EQ(0, this->DeleteCreatedBetween(cs, last_month, last_minute));
+  EXPECT_EQ(0, this->DeleteCreatedBetween(cs, next_minute, next_month));
+  // Check that the cookie is still there.
+  this->MatchCookieLines("A=B", this->GetCookies(cs, this->url_google_));
+
+  // Remove the cookie with an interval defined by two dates.
+  EXPECT_EQ(1, this->DeleteCreatedBetween(cs, last_minute, next_minute));
+  // Check that the cookie disappeared.
+  this->MatchCookieLines("", this->GetCookies(cs, this->url_google_));
+
+  // Add another cookie.
+  EXPECT_TRUE(this->SetCookie(cs, this->url_google_, "C=D"));
+  // Check that the cookie is in the store.
+  this->MatchCookieLines("C=D", this->GetCookies(cs, this->url_google_));
+
+  // Remove the cookie with a null ending time.
+  EXPECT_EQ(1, this->DeleteCreatedBetween(cs, last_minute, base::Time()));
+  // Check that the cookie disappeared.
+  this->MatchCookieLines("", this->GetCookies(cs, this->url_google_));
+}
+
 TYPED_TEST_P(CookieStoreTest, TestSecure) {
     scoped_refptr<CookieStore> cs(this->GetCookieStore());
 
   EXPECT_TRUE(this->SetCookie(cs, this->url_google_, "A=B"));
   this->MatchCookieLines("A=B", this->GetCookies(cs, this->url_google_));
   this->MatchCookieLines("A=B", this->GetCookies(cs, this->url_google_secure_));
 
   EXPECT_TRUE(this->SetCookie(cs, this->url_google_secure_, "A=B; secure"));
   // The secure should overwrite the non-secure.
   this->MatchCookieLines("", this->GetCookies(cs, this->url_google_));
@@ skipping 112 matching lines @@
                               "g=10"));
   EXPECT_EQ("d=1; a=4; e=1; b=1; c=1",
       this->GetCookies(cs, GURL("http://d.c.b.a.google.com/aa/bb/cc/dd")));
 }
 
 REGISTER_TYPED_TEST_CASE_P(CookieStoreTest,
     TypeTest, DomainTest, DomainWithTrailingDotTest, ValidSubdomainTest,
     InvalidDomainTest, DomainWithoutLeadingDotTest, CaseInsensitiveDomainTest,
     TestIpAddress, TestNonDottedAndTLD, TestHostEndsWithDot, InvalidScheme,
     InvalidScheme_Read, PathTest, HttpOnlyTest, TestGetCookiesWithInfo,
-    TestCookieDeletion, TestSecure, NetUtilCookieTest,
-    OverwritePersistentCookie, CookieOrdering);
+    TestCookieDeletion, TestDeleteAllCreatedBetween, TestSecure,
+    NetUtilCookieTest, OverwritePersistentCookie, CookieOrdering);
 
 template<class CookieStoreTestTraits>
 class MultiThreadedCookieStoreTest :
     public CookieStoreTest<CookieStoreTestTraits> {
  public:
   MultiThreadedCookieStoreTest() : other_thread_("CMTthread") {}
 
   // Helper methods for calling the asynchronous CookieStore methods
   // from a different thread.
 
@@ skipping 159 matching lines @@
 }
 
 REGISTER_TYPED_TEST_CASE_P(MultiThreadedCookieStoreTest,
     ThreadCheckGetCookies, ThreadCheckGetCookiesWithOptions,
     ThreadCheckGetCookiesWithInfo, ThreadCheckSetCookieWithOptions,
     ThreadCheckDeleteCookie);
 
 }  // namespace net
 
 #endif  // NET_BASE_COOKIE_STORE_UNITTEST_H_