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 bool SetCookieWithOptions(const GURL& url,
  38                                     const std::string& cookie_line,
  39                                     const CookieOptions& options);
  40
  41   virtual std::string GetCookiesWithOptions(const GURL& url,
  42                                             const CookieOptions& options);
  43
  44   virtual void DeleteCookie(const GURL& url,
  45                             const std::string& cookie_name);
  46
  47   virtual void DeleteCookieAsync(const GURL& url,
  48                                  const std::string& cookie_name,
  49                                  const base::Closure& callback) OVERRIDE;
  50
  51   virtual void DeleteAllCreatedBetweenAsync(
  52       const base::Time& delete_begin,
  53       const base::Time& delete_end,
  54       const DeleteCallback& callback) OVERRIDE;
  55
  56   virtual void DeleteAllCreatedBetweenForHostAsync(
  57       const base::Time delete_begin,
  58       const base::Time delete_end,
  59       const GURL& url,
  60       const DeleteCallback& callback) OVERRIDE;
  61
  62   virtual void DeleteSessionCookiesAsync(const DeleteCallback&) OVERRIDE;
  63
  64   virtual CookieMonster* GetCookieMonster() OVERRIDE;
  65
  66  private:
  67
  68   // Be called immediately from CookieMonster.
  69
  70   void SetCookiesInternalCallback(bool result);
  71
  72   void GetCookiesWithOptionsInternalCallback(const std::string& cookie);
  73
  74   // Invoke the original callbacks.
  75
  76   void InvokeSetCookiesCallback(
  77       const CookieMonster::SetCookiesCallback& callback);
  78
  79   void InvokeGetCookieStringCallback(
  80       const CookieMonster::GetCookiesCallback& callback);
  81
  82   friend class base::RefCountedThreadSafe<DelayedCookieMonster>;
  83   virtual ~DelayedCookieMonster();
  84
  85   scoped_refptr<CookieMonster> cookie_monster_;
  86
  87   bool did_run_;
  88   bool result_;
  89   std::string cookie_;
  90   std::string cookie_line_;
  91 };
  92
  93 }  // namespace net
  94
  95 #endif  // NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_