OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "base/cancelable_callback.h" | 5 #include "base/cancelable_callback.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
| 9 #include "base/location.h" |
9 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
10 #include "base/message_loop/message_loop.h" | |
11 #include "base/run_loop.h" | 11 #include "base/run_loop.h" |
| 12 #include "base/single_thread_task_runner.h" |
| 13 #include "base/thread_task_runner_handle.h" |
12 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
13 | 15 |
14 namespace base { | 16 namespace base { |
15 namespace { | 17 namespace { |
16 | 18 |
17 class TestRefCounted : public RefCountedThreadSafe<TestRefCounted> { | 19 class TestRefCounted : public RefCountedThreadSafe<TestRefCounted> { |
18 private: | 20 private: |
19 friend class RefCountedThreadSafe<TestRefCounted>; | 21 friend class RefCountedThreadSafe<TestRefCounted>; |
20 ~TestRefCounted() {}; | 22 ~TestRefCounted() {}; |
21 }; | 23 }; |
(...skipping 136 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
158 | 160 |
159 // CancelableCallback posted to a MessageLoop with PostTask. | 161 // CancelableCallback posted to a MessageLoop with PostTask. |
160 // - Callbacks posted to a MessageLoop can be cancelled. | 162 // - Callbacks posted to a MessageLoop can be cancelled. |
161 TEST(CancelableCallbackTest, PostTask) { | 163 TEST(CancelableCallbackTest, PostTask) { |
162 MessageLoop loop; | 164 MessageLoop loop; |
163 | 165 |
164 int count = 0; | 166 int count = 0; |
165 CancelableClosure cancelable(base::Bind(&Increment, | 167 CancelableClosure cancelable(base::Bind(&Increment, |
166 base::Unretained(&count))); | 168 base::Unretained(&count))); |
167 | 169 |
168 MessageLoop::current()->PostTask(FROM_HERE, cancelable.callback()); | 170 ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, cancelable.callback()); |
169 RunLoop().RunUntilIdle(); | 171 RunLoop().RunUntilIdle(); |
170 | 172 |
171 EXPECT_EQ(1, count); | 173 EXPECT_EQ(1, count); |
172 | 174 |
173 MessageLoop::current()->PostTask(FROM_HERE, cancelable.callback()); | 175 ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, cancelable.callback()); |
174 | 176 |
175 // Cancel before running the message loop. | 177 // Cancel before running the message loop. |
176 cancelable.Cancel(); | 178 cancelable.Cancel(); |
177 RunLoop().RunUntilIdle(); | 179 RunLoop().RunUntilIdle(); |
178 | 180 |
179 // Callback never ran due to cancellation; count is the same. | 181 // Callback never ran due to cancellation; count is the same. |
180 EXPECT_EQ(1, count); | 182 EXPECT_EQ(1, count); |
181 } | 183 } |
182 | 184 |
183 } // namespace | 185 } // namespace |
184 } // namespace base | 186 } // namespace base |
OLD | NEW |