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