net/cookies/cookie_store_test_helpers.h

   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 #ifndef NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_
   6 #define NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_
   7
   8 #include "net/cookies/cookie_monster.h"
   9
  10 #include <string>
  11 #include <vector>
  12
  13 #include "base/callback_forward.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15
  16 namespace net {
  17
  18 class DelayedCookieMonster : public CookieStore {
  19  public:
  20   DelayedCookieMonster();
  21
  22   // Call the asynchronous CookieMonster function, expect it to immediately
  23   // invoke the internal callback.
  24   // Post a delayed task to invoke the original callback with the results.
  25
  26   virtual void SetCookieWithOptionsAsync(
  27       const GURL& url,
  28       const std::string& cookie_line,
  29       const CookieOptions& options,
  30       const CookieMonster::SetCookiesCallback& callback) OVERRIDE;
  31
  32   virtual void GetCookiesWithOptionsAsync(
  33       const GURL& url,
  34       const CookieOptions& options,
  35       const CookieMonster::GetCookiesCallback& callback) OVERRIDE;
  36
  37   virtual void GetAllCookiesForURLAsync(
  38       const GURL& url,
  39       const GetCookieListCallback& callback) OVERRIDE;
  40
  41   virtual bool SetCookieWithOptions(const GURL& url,
  42                                     const std::string& cookie_line,
  43                                     const CookieOptions& options);
  44
  45   virtual std::string GetCookiesWithOptions(const GURL& url,
  46                                             const CookieOptions& options);
  47
  48   virtual void DeleteCookie(const GURL& url,
  49                             const std::string& cookie_name);
  50
  51   virtual void DeleteCookieAsync(const GURL& url,
  52                                  const std::string& cookie_name,
  53                                  const base::Closure& callback) OVERRIDE;
  54
  55   virtual void DeleteAllCreatedBetweenAsync(
  56       const base::Time& delete_begin,
  57       const base::Time& delete_end,
  58       const DeleteCallback& callback) OVERRIDE;
  59
  60   virtual void DeleteAllCreatedBetweenForHostAsync(
  61       const base::Time delete_begin,
  62       const base::Time delete_end,
  63       const GURL& url,
  64       const DeleteCallback& callback) OVERRIDE;
  65
  66   virtual void DeleteSessionCookiesAsync(const DeleteCallback&) OVERRIDE;
  67
  68   virtual CookieMonster* GetCookieMonster() OVERRIDE;
  69
  70  private:
  71
  72   // Be called immediately from CookieMonster.
  73
  74   void SetCookiesInternalCallback(bool result);
  75
  76   void GetCookiesWithOptionsInternalCallback(const std::string& cookie);
  77
  78   // Invoke the original callbacks.
  79
  80   void InvokeSetCookiesCallback(
  81       const CookieMonster::SetCookiesCallback& callback);
  82
  83   void InvokeGetCookieStringCallback(
  84       const CookieMonster::GetCookiesCallback& callback);
  85
  86   friend class base::RefCountedThreadSafe<DelayedCookieMonster>;
  87   virtual ~DelayedCookieMonster();
  88
  89   scoped_refptr<CookieMonster> cookie_monster_;
  90
  91   bool did_run_;
  92   bool result_;
  93   std::string cookie_;
  94   std::string cookie_line_;
  95 };
  96
  97 }  // namespace net
  98
  99 #endif  // NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_