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.
5 #ifndef NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_
6 #define NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_
8 #include "net/cookies/cookie_monster.h"
13 #include "base/callback_forward.h"
14 #include "testing/gtest/include/gtest/gtest.h"
18 class DelayedCookieMonster : public CookieStore {
20 DelayedCookieMonster();
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.
26 void SetCookieWithOptionsAsync(
28 const std::string& cookie_line,
29 const CookieOptions& options,
30 const CookieMonster::SetCookiesCallback& callback) override;
32 void GetCookiesWithOptionsAsync(
34 const CookieOptions& options,
35 const CookieMonster::GetCookiesCallback& callback) override;
37 void GetAllCookiesForURLAsync(const GURL& url,
38 const GetCookieListCallback& callback) override;
40 virtual bool SetCookieWithOptions(const GURL& url,
41 const std::string& cookie_line,
42 const CookieOptions& options);
44 virtual std::string GetCookiesWithOptions(const GURL& url,
45 const CookieOptions& options);
47 virtual void DeleteCookie(const GURL& url,
48 const std::string& cookie_name);
50 void DeleteCookieAsync(const GURL& url,
51 const std::string& cookie_name,
52 const base::Closure& callback) override;
54 void DeleteAllCreatedBetweenAsync(const base::Time& delete_begin,
55 const base::Time& delete_end,
56 const DeleteCallback& callback) override;
58 void DeleteAllCreatedBetweenForHostAsync(
59 const base::Time delete_begin,
60 const base::Time delete_end,
62 const DeleteCallback& callback) override;
64 void DeleteSessionCookiesAsync(const DeleteCallback&) override;
66 CookieMonster* GetCookieMonster() override;
68 scoped_ptr<CookieStore::CookieChangedSubscription>
69 AddCallbackForCookie(const GURL& url, const std::string& name,
70 const CookieChangedCallback& callback) override;
74 // Be called immediately from CookieMonster.
76 void SetCookiesInternalCallback(bool result);
78 void GetCookiesWithOptionsInternalCallback(const std::string& cookie);
80 // Invoke the original callbacks.
82 void InvokeSetCookiesCallback(
83 const CookieMonster::SetCookiesCallback& callback);
85 void InvokeGetCookieStringCallback(
86 const CookieMonster::GetCookiesCallback& callback);
88 friend class base::RefCountedThreadSafe<DelayedCookieMonster>;
89 ~DelayedCookieMonster() override;
91 scoped_refptr<CookieMonster> cookie_monster_;
96 std::string cookie_line_;
101 #endif // NET_COOKIES_COOKIE_STORE_TEST_HELPERS_H_