| 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/bind.h" | 5 #include "base/bind.h" |
| 6 #include "base/memory/ref_counted.h" | 6 #include "base/memory/ref_counted.h" |
| 7 #include "base/task.h" | 7 #include "base/task.h" |
| 8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
| 9 | 9 |
| 10 namespace { | 10 namespace { |
| 11 | 11 |
| 12 class DoneTask : public Task { | |
| 13 public: | |
| 14 DoneTask(int* run_count, bool* was_deleted) | |
| 15 : run_count_(run_count), was_deleted_(was_deleted) { | |
| 16 } | |
| 17 virtual ~DoneTask() { | |
| 18 *was_deleted_ = true; | |
| 19 } | |
| 20 | |
| 21 virtual void Run() { | |
| 22 ++(*run_count_); | |
| 23 } | |
| 24 | |
| 25 int* run_count_; | |
| 26 bool* was_deleted_; | |
| 27 }; | |
| 28 | |
| 29 TEST(TaskTest, TestScopedTaskRunnerExitScope) { | |
| 30 int run_count = 0; | |
| 31 bool was_deleted = false; | |
| 32 { | |
| 33 base::ScopedTaskRunner runner(new DoneTask(&run_count, &was_deleted)); | |
| 34 EXPECT_EQ(0, run_count); | |
| 35 } | |
| 36 EXPECT_EQ(1, run_count); | |
| 37 EXPECT_TRUE(was_deleted); | |
| 38 } | |
| 39 | |
| 40 TEST(TaskTest, TestScopedTaskRunnerRelease) { | |
| 41 int run_count = 0; | |
| 42 bool was_deleted = false; | |
| 43 { | |
| 44 base::ScopedTaskRunner runner(new DoneTask(&run_count, &was_deleted)); | |
| 45 delete runner.Release(); | |
| 46 EXPECT_TRUE(was_deleted); | |
| 47 } | |
| 48 EXPECT_EQ(0, run_count); | |
| 49 } | |
| 50 | |
| 51 TEST(TaskTest, TestScopedTaskRunnerManualRun) { | |
| 52 int run_count = 0; | |
| 53 Task* done_task = NULL; | |
| 54 bool was_deleted = false; | |
| 55 { | |
| 56 base::ScopedTaskRunner runner(new DoneTask(&run_count, &was_deleted)); | |
| 57 done_task = runner.Release(); | |
| 58 EXPECT_TRUE(NULL != done_task); | |
| 59 EXPECT_FALSE(was_deleted); | |
| 60 EXPECT_EQ(0, run_count); | |
| 61 done_task->Run(); | |
| 62 EXPECT_FALSE(was_deleted); | |
| 63 EXPECT_EQ(1, run_count); | |
| 64 } | |
| 65 EXPECT_EQ(1, run_count); | |
| 66 delete done_task; | |
| 67 EXPECT_TRUE(was_deleted); | |
| 68 } | |
| 69 | |
| 70 void Increment(int* value) { | 12 void Increment(int* value) { |
| 71 (*value)++; | 13 (*value)++; |
| 72 } | 14 } |
| 73 | 15 |
| 74 TEST(TaskTest, TestScopedClosureRunnerExitScope) { | 16 TEST(TaskTest, TestScopedClosureRunnerExitScope) { |
| 75 int run_count = 0; | 17 int run_count = 0; |
| 76 { | 18 { |
| 77 base::ScopedClosureRunner runner(base::Bind(&Increment, &run_count)); | 19 base::ScopedClosureRunner runner(base::Bind(&Increment, &run_count)); |
| 78 EXPECT_EQ(0, run_count); | 20 EXPECT_EQ(0, run_count); |
| 79 } | 21 } |
| 80 EXPECT_EQ(1, run_count); | 22 EXPECT_EQ(1, run_count); |
| 81 } | 23 } |
| 82 | 24 |
| 83 TEST(TaskTest, TestScopedClosureRunnerRelease) { | 25 TEST(TaskTest, TestScopedClosureRunnerRelease) { |
| 84 int run_count = 0; | 26 int run_count = 0; |
| 85 base::Closure c; | 27 base::Closure c; |
| 86 { | 28 { |
| 87 base::ScopedClosureRunner runner(base::Bind(&Increment, &run_count)); | 29 base::ScopedClosureRunner runner(base::Bind(&Increment, &run_count)); |
| 88 c = runner.Release(); | 30 c = runner.Release(); |
| 89 EXPECT_EQ(0, run_count); | 31 EXPECT_EQ(0, run_count); |
| 90 } | 32 } |
| 91 EXPECT_EQ(0, run_count); | 33 EXPECT_EQ(0, run_count); |
| 92 c.Run(); | 34 c.Run(); |
| 93 EXPECT_EQ(1, run_count); | 35 EXPECT_EQ(1, run_count); |
| 94 } | 36 } |
| 95 | 37 |
| 96 } // namespace | 38 } // namespace |
| OLD | NEW |