| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/cookies/cookie_store_test_callbacks.h" | 5 #include "net/cookies/cookie_store_test_callbacks.h" |
| 6 | 6 |
| 7 #include "base/message_loop/message_loop.h" | 7 #include "base/message_loop/message_loop.h" |
| 8 #include "base/threading/thread.h" | 8 #include "base/threading/thread.h" |
| 9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
| 10 | 10 |
| 11 namespace net { | 11 namespace net { |
| 12 | 12 |
| 13 CookieCallback::CookieCallback(base::Thread* run_in_thread) | 13 CookieCallback::CookieCallback(base::Thread* run_in_thread) |
| 14 : did_run_(false), | 14 : did_run_(false), |
| 15 run_in_thread_(run_in_thread), | 15 run_in_thread_(run_in_thread), |
| 16 run_in_loop_(NULL), | 16 run_in_loop_(NULL), |
| 17 parent_loop_(base::MessageLoop::current()), | 17 parent_loop_(base::MessageLoop::current()), |
| 18 loop_to_quit_(base::MessageLoop::current()) {} | 18 loop_to_quit_(base::MessageLoop::current()) { |
| 19 } |
| 19 | 20 |
| 20 CookieCallback::CookieCallback() | 21 CookieCallback::CookieCallback() |
| 21 : did_run_(false), | 22 : did_run_(false), |
| 22 run_in_thread_(NULL), | 23 run_in_thread_(NULL), |
| 23 run_in_loop_(base::MessageLoop::current()), | 24 run_in_loop_(base::MessageLoop::current()), |
| 24 parent_loop_(NULL), | 25 parent_loop_(NULL), |
| 25 loop_to_quit_(base::MessageLoop::current()) {} | 26 loop_to_quit_(base::MessageLoop::current()) { |
| 27 } |
| 26 | 28 |
| 27 void CookieCallback::CallbackEpilogue() { | 29 void CookieCallback::CallbackEpilogue() { |
| 28 base::MessageLoop* expected_loop = NULL; | 30 base::MessageLoop* expected_loop = NULL; |
| 29 if (run_in_thread_) { | 31 if (run_in_thread_) { |
| 30 DCHECK(!run_in_loop_); | 32 DCHECK(!run_in_loop_); |
| 31 expected_loop = run_in_thread_->message_loop(); | 33 expected_loop = run_in_thread_->message_loop(); |
| 32 } else if (run_in_loop_) { | 34 } else if (run_in_loop_) { |
| 33 expected_loop = run_in_loop_; | 35 expected_loop = run_in_loop_; |
| 34 } | 36 } |
| 35 ASSERT_TRUE(expected_loop != NULL); | 37 ASSERT_TRUE(expected_loop != NULL); |
| 36 | 38 |
| 37 did_run_ = true; | 39 did_run_ = true; |
| 38 EXPECT_EQ(expected_loop, base::MessageLoop::current()); | 40 EXPECT_EQ(expected_loop, base::MessageLoop::current()); |
| 39 loop_to_quit_->PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); | 41 loop_to_quit_->PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); |
| 40 } | 42 } |
| 41 | 43 |
| 42 StringResultCookieCallback::StringResultCookieCallback() {} | 44 StringResultCookieCallback::StringResultCookieCallback() { |
| 45 } |
| 43 StringResultCookieCallback::StringResultCookieCallback( | 46 StringResultCookieCallback::StringResultCookieCallback( |
| 44 base::Thread* run_in_thread) | 47 base::Thread* run_in_thread) |
| 45 : CookieCallback(run_in_thread) {} | 48 : CookieCallback(run_in_thread) { |
| 49 } |
| 46 | 50 |
| 47 NoResultCookieCallback::NoResultCookieCallback() {} | 51 NoResultCookieCallback::NoResultCookieCallback() { |
| 52 } |
| 48 NoResultCookieCallback::NoResultCookieCallback(base::Thread* run_in_thread) | 53 NoResultCookieCallback::NoResultCookieCallback(base::Thread* run_in_thread) |
| 49 : CookieCallback(run_in_thread) {} | 54 : CookieCallback(run_in_thread) { |
| 55 } |
| 50 | 56 |
| 51 } // namespace net | 57 } // namespace net |
| OLD | NEW |