jhawkins@chromium.org | 4a31021 | 2012-01-04 04:36:57 +0900 | [diff] [blame] | 1 | // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
jam@chromium.org | 2750bf8 | 2009-07-18 03:14:47 +0900 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
fischman@chromium.org | dc221a7 | 2012-03-25 05:37:27 +0900 | [diff] [blame] | 5 | #include "base/bind.h" |
akalin@chromium.org | 6ed4f88 | 2010-02-19 12:15:59 +0900 | [diff] [blame] | 6 | #include "base/callback.h" |
fischman@chromium.org | dc221a7 | 2012-03-25 05:37:27 +0900 | [diff] [blame] | 7 | #include "base/callback_helpers.h" |
ajwong@chromium.org | fa0ff43 | 2011-02-19 08:29:31 +0900 | [diff] [blame] | 8 | #include "base/callback_internal.h" |
mattm@chromium.org | bf27cb7 | 2012-06-14 08:15:16 +0900 | [diff] [blame] | 9 | #include "base/memory/ref_counted.h" |
levin@chromium.org | 5c52868 | 2011-03-28 10:54:15 +0900 | [diff] [blame] | 10 | #include "base/memory/scoped_ptr.h" |
jam@chromium.org | 2750bf8 | 2009-07-18 03:14:47 +0900 | [diff] [blame] | 11 | #include "testing/gtest/include/gtest/gtest.h" |
| 12 | |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 13 | namespace base { |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 14 | |
jam@chromium.org | 2750bf8 | 2009-07-18 03:14:47 +0900 | [diff] [blame] | 15 | namespace { |
| 16 | |
ajwong@chromium.org | e064823 | 2011-02-19 09:52:15 +0900 | [diff] [blame] | 17 | struct FakeInvoker { |
ajwong@chromium.org | c9c79af | 2011-11-22 04:23:44 +0900 | [diff] [blame] | 18 | typedef void(RunType)(internal::BindStateBase*); |
| 19 | static void Run(internal::BindStateBase*) { |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 20 | } |
| 21 | }; |
fischman@chromium.org | dc221a7 | 2012-03-25 05:37:27 +0900 | [diff] [blame] | 22 | |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 23 | } // namespace |
| 24 | |
| 25 | namespace internal { |
| 26 | template <typename Runnable, typename RunType, typename BoundArgsType> |
| 27 | struct BindState; |
jam@chromium.org | 2750bf8 | 2009-07-18 03:14:47 +0900 | [diff] [blame] | 28 | |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 29 | // White-box testpoints to inject into a Callback<> object for checking |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 30 | // comparators and emptiness APIs. Use a BindState that is specialized |
| 31 | // based on a type we declared in the anonymous namespace above to remove any |
| 32 | // chance of colliding with another instantiation and breaking the |
| 33 | // one-definition-rule. |
| 34 | template <> |
| 35 | struct BindState<void(void), void(void), void(FakeInvoker)> |
| 36 | : public BindStateBase { |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 37 | public: |
ajwong@chromium.org | c9c79af | 2011-11-22 04:23:44 +0900 | [diff] [blame] | 38 | typedef FakeInvoker InvokerType; |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 39 | }; |
| 40 | |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 41 | template <> |
| 42 | struct BindState<void(void), void(void), |
| 43 | void(FakeInvoker, FakeInvoker)> |
| 44 | : public BindStateBase { |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 45 | public: |
ajwong@chromium.org | c9c79af | 2011-11-22 04:23:44 +0900 | [diff] [blame] | 46 | typedef FakeInvoker InvokerType; |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 47 | }; |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 48 | } // namespace internal |
| 49 | |
| 50 | namespace { |
| 51 | |
| 52 | typedef internal::BindState<void(void), void(void), void(FakeInvoker)> |
| 53 | FakeBindState1; |
| 54 | typedef internal::BindState<void(void), void(void), |
| 55 | void(FakeInvoker, FakeInvoker)> |
| 56 | FakeBindState2; |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 57 | |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 58 | class CallbackTest : public ::testing::Test { |
| 59 | public: |
| 60 | CallbackTest() |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 61 | : callback_a_(new FakeBindState1()), |
| 62 | callback_b_(new FakeBindState2()) { |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 63 | } |
| 64 | |
| 65 | virtual ~CallbackTest() { |
| 66 | } |
| 67 | |
| 68 | protected: |
| 69 | Callback<void(void)> callback_a_; |
| 70 | const Callback<void(void)> callback_b_; // Ensure APIs work with const. |
| 71 | Callback<void(void)> null_callback_; |
| 72 | }; |
| 73 | |
| 74 | // Ensure we can create unbound callbacks. We need this to be able to store |
| 75 | // them in class members that can be initialized later. |
| 76 | TEST_F(CallbackTest, DefaultConstruction) { |
| 77 | Callback<void(void)> c0; |
| 78 | Callback<void(int)> c1; |
| 79 | Callback<void(int,int)> c2; |
| 80 | Callback<void(int,int,int)> c3; |
| 81 | Callback<void(int,int,int,int)> c4; |
| 82 | Callback<void(int,int,int,int,int)> c5; |
| 83 | Callback<void(int,int,int,int,int,int)> c6; |
| 84 | |
| 85 | EXPECT_TRUE(c0.is_null()); |
| 86 | EXPECT_TRUE(c1.is_null()); |
| 87 | EXPECT_TRUE(c2.is_null()); |
| 88 | EXPECT_TRUE(c3.is_null()); |
| 89 | EXPECT_TRUE(c4.is_null()); |
| 90 | EXPECT_TRUE(c5.is_null()); |
| 91 | EXPECT_TRUE(c6.is_null()); |
| 92 | } |
| 93 | |
| 94 | TEST_F(CallbackTest, IsNull) { |
| 95 | EXPECT_TRUE(null_callback_.is_null()); |
| 96 | EXPECT_FALSE(callback_a_.is_null()); |
| 97 | EXPECT_FALSE(callback_b_.is_null()); |
| 98 | } |
| 99 | |
| 100 | TEST_F(CallbackTest, Equals) { |
| 101 | EXPECT_TRUE(callback_a_.Equals(callback_a_)); |
| 102 | EXPECT_FALSE(callback_a_.Equals(callback_b_)); |
| 103 | EXPECT_FALSE(callback_b_.Equals(callback_a_)); |
| 104 | |
| 105 | // We should compare based on instance, not type. |
ajwong@chromium.org | abd7000 | 2011-12-20 09:10:04 +0900 | [diff] [blame] | 106 | Callback<void(void)> callback_c(new FakeBindState1()); |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 107 | Callback<void(void)> callback_a2 = callback_a_; |
| 108 | EXPECT_TRUE(callback_a_.Equals(callback_a2)); |
| 109 | EXPECT_FALSE(callback_a_.Equals(callback_c)); |
| 110 | |
| 111 | // Empty, however, is always equal to empty. |
| 112 | Callback<void(void)> empty2; |
| 113 | EXPECT_TRUE(null_callback_.Equals(empty2)); |
| 114 | } |
| 115 | |
| 116 | TEST_F(CallbackTest, Reset) { |
| 117 | // Resetting should bring us back to empty. |
| 118 | ASSERT_FALSE(callback_a_.is_null()); |
| 119 | ASSERT_FALSE(callback_a_.Equals(null_callback_)); |
| 120 | |
| 121 | callback_a_.Reset(); |
| 122 | |
| 123 | EXPECT_TRUE(callback_a_.is_null()); |
| 124 | EXPECT_TRUE(callback_a_.Equals(null_callback_)); |
| 125 | } |
| 126 | |
fischman@chromium.org | dc221a7 | 2012-03-25 05:37:27 +0900 | [diff] [blame] | 127 | struct TestForReentrancy { |
| 128 | TestForReentrancy() |
| 129 | : cb_already_run(false), |
| 130 | cb(Bind(&TestForReentrancy::AssertCBIsNull, Unretained(this))) { |
| 131 | } |
| 132 | void AssertCBIsNull() { |
| 133 | ASSERT_TRUE(cb.is_null()); |
| 134 | cb_already_run = true; |
| 135 | } |
| 136 | bool cb_already_run; |
| 137 | Closure cb; |
| 138 | }; |
| 139 | |
| 140 | TEST_F(CallbackTest, ResetAndReturn) { |
| 141 | TestForReentrancy tfr; |
| 142 | ASSERT_FALSE(tfr.cb.is_null()); |
| 143 | ASSERT_FALSE(tfr.cb_already_run); |
| 144 | ResetAndReturn(&tfr.cb).Run(); |
| 145 | ASSERT_TRUE(tfr.cb.is_null()); |
| 146 | ASSERT_TRUE(tfr.cb_already_run); |
| 147 | } |
| 148 | |
mattm@chromium.org | bf27cb7 | 2012-06-14 08:15:16 +0900 | [diff] [blame] | 149 | class CallbackOwner : public base::RefCounted<CallbackOwner> { |
| 150 | public: |
hans@chromium.org | 07615b4 | 2013-01-15 19:52:11 +0900 | [diff] [blame^] | 151 | explicit CallbackOwner(bool* deleted) { |
mattm@chromium.org | bf27cb7 | 2012-06-14 08:15:16 +0900 | [diff] [blame] | 152 | callback_ = Bind(&CallbackOwner::Unused, this); |
| 153 | deleted_ = deleted; |
| 154 | } |
| 155 | void Reset() { |
| 156 | callback_.Reset(); |
| 157 | // We are deleted here if no-one else had a ref to us. |
| 158 | } |
| 159 | |
| 160 | private: |
| 161 | friend class base::RefCounted<CallbackOwner>; |
| 162 | virtual ~CallbackOwner() { |
| 163 | *deleted_ = true; |
| 164 | } |
| 165 | void Unused() { |
| 166 | FAIL() << "Should never be called"; |
| 167 | } |
| 168 | |
| 169 | Closure callback_; |
| 170 | bool* deleted_; |
| 171 | }; |
| 172 | |
| 173 | TEST_F(CallbackTest, CallbackHasLastRefOnContainingObject) { |
| 174 | bool deleted = false; |
| 175 | CallbackOwner* owner = new CallbackOwner(&deleted); |
| 176 | owner->Reset(); |
| 177 | ASSERT_TRUE(deleted); |
| 178 | } |
| 179 | |
ajwong@chromium.org | 62fb0a0 | 2011-02-18 13:05:14 +0900 | [diff] [blame] | 180 | } // namespace |
| 181 | } // namespace base |