| OLD | NEW |
| (Empty) |
| 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_ | |
| OLD | NEW |