net/cookies/cookie_store_test_callbacks.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_CALLBACKS_H_
   6 #define NET_COOKIES_COOKIE_STORE_TEST_CALLBACKS_H_
   7
   8 #include <string>
   9 #include <vector>
  10
  11 #include "net/cookies/cookie_store.h"
  12
  13 namespace base {
  14 class MessageLoop;
  15 class Thread;
  16 }
  17
  18 namespace net {
  19
  20 // Defines common behaviour for the callbacks from GetCookies, SetCookies, etc.
  21 // Asserts that the current thread is the expected invocation thread, sends a
  22 // quit to the thread in which it was constructed.
  23 class CookieCallback {
  24  public:
  25   // Indicates whether the callback has been called.
  26   bool did_run() { return did_run_; }
  27
  28  protected:
  29   // Constructs a callback that expects to be called in the given thread and
  30   // will, upon execution, send a QUIT to the constructing thread.
  31   explicit CookieCallback(base::Thread* run_in_thread);
  32
  33   // Constructs a callback that expects to be called in current thread and will
  34   // send a QUIT to the constructing thread.
  35   CookieCallback();
  36
  37   // Tests whether the current thread was the caller's thread.
  38   // Sends a QUIT to the constructing thread.
  39   void CallbackEpilogue();
  40
  41  private:
  42   bool did_run_;
  43   base::Thread* run_in_thread_;
  44   base::MessageLoop* run_in_loop_;
  45   base::MessageLoop* parent_loop_;
  46   base::MessageLoop* loop_to_quit_;
  47 };
  48
  49 // Callback implementations for the asynchronous CookieStore methods.
  50
  51 template <typename T>
  52 class ResultSavingCookieCallback : public CookieCallback {
  53  public:
  54   ResultSavingCookieCallback() {
  55   }
  56   explicit ResultSavingCookieCallback(base::Thread* run_in_thread)
  57       : CookieCallback(run_in_thread) {
  58   }
  59
  60   void Run(T result) {
  61     result_ = result;
  62     CallbackEpilogue();
  63   }
  64
  65   const T& result() { return result_; }
  66
  67  private:
  68   T result_;
  69 };
  70
  71 class StringResultCookieCallback : public CookieCallback {
  72  public:
  73   StringResultCookieCallback();
  74   explicit StringResultCookieCallback(base::Thread* run_in_thread);
  75
  76   void Run(const std::string& result) {
  77     result_ = result;
  78     CallbackEpilogue();
  79   }
  80
  81   const std::string& result() { return result_; }
  82
  83  private:
  84   std::string result_;
  85 };
  86
  87 class NoResultCookieCallback : public CookieCallback {
  88  public:
  89   NoResultCookieCallback();
  90   explicit NoResultCookieCallback(base::Thread* run_in_thread);
  91
  92   void Run() {
  93     CallbackEpilogue();
  94   }
  95 };
  96
  97 }  // namespace net
  98
  99 #endif  // NET_COOKIES_COOKIE_STORE_TEST_CALLBACKS_H_