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
16namespace net {
17
18class 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_
100