| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "extensions/common/one_shot_event.h" | 5 #include "extensions/common/one_shot_event.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
| 9 #include "base/single_thread_task_runner.h" | 9 #include "base/single_thread_task_runner.h" |
| 10 #include "base/test/test_simple_task_runner.h" | 10 #include "base/test/test_simple_task_runner.h" |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 EXPECT_TRUE(event.is_signaled()); | 23 EXPECT_TRUE(event.is_signaled()); |
| 24 } | 24 } |
| 25 | 25 |
| 26 TEST(OneShotEventTest, CallsQueue) { | 26 TEST(OneShotEventTest, CallsQueue) { |
| 27 OneShotEvent event; | 27 OneShotEvent event; |
| 28 scoped_refptr<base::TestSimpleTaskRunner> runner( | 28 scoped_refptr<base::TestSimpleTaskRunner> runner( |
| 29 new base::TestSimpleTaskRunner); | 29 new base::TestSimpleTaskRunner); |
| 30 int i = 0; | 30 int i = 0; |
| 31 event.Post(FROM_HERE, base::Bind(&Increment, &i), runner); | 31 event.Post(FROM_HERE, base::Bind(&Increment, &i), runner); |
| 32 event.Post(FROM_HERE, base::Bind(&Increment, &i), runner); | 32 event.Post(FROM_HERE, base::Bind(&Increment, &i), runner); |
| 33 EXPECT_EQ(0U, runner->GetPendingTasks().size()); | 33 EXPECT_EQ(0U, runner->NumPendingTasks()); |
| 34 event.Signal(); | 34 event.Signal(); |
| 35 ASSERT_EQ(2U, runner->GetPendingTasks().size()); | 35 ASSERT_EQ(2U, runner->NumPendingTasks()); |
| 36 EXPECT_NE(runner->GetPendingTasks()[0].location.line_number(), | 36 EXPECT_NE(runner->GetPendingTasks()[0].location.line_number(), |
| 37 runner->GetPendingTasks()[1].location.line_number()) | 37 runner->GetPendingTasks()[1].location.line_number()) |
| 38 << "Make sure FROM_HERE is propagated."; | 38 << "Make sure FROM_HERE is propagated."; |
| 39 EXPECT_EQ(0, i); | 39 EXPECT_EQ(0, i); |
| 40 runner->RunPendingTasks(); | 40 runner->RunPendingTasks(); |
| 41 EXPECT_EQ(2, i); | 41 EXPECT_EQ(2, i); |
| 42 } | 42 } |
| 43 | 43 |
| 44 TEST(OneShotEventTest, CallsAfterSignalDontRunInline) { | 44 TEST(OneShotEventTest, CallsAfterSignalDontRunInline) { |
| 45 OneShotEvent event; | 45 OneShotEvent event; |
| 46 scoped_refptr<base::TestSimpleTaskRunner> runner( | 46 scoped_refptr<base::TestSimpleTaskRunner> runner( |
| 47 new base::TestSimpleTaskRunner); | 47 new base::TestSimpleTaskRunner); |
| 48 int i = 0; | 48 int i = 0; |
| 49 | 49 |
| 50 event.Signal(); | 50 event.Signal(); |
| 51 event.Post(FROM_HERE, base::Bind(&Increment, &i), runner); | 51 event.Post(FROM_HERE, base::Bind(&Increment, &i), runner); |
| 52 EXPECT_EQ(1U, runner->GetPendingTasks().size()); | 52 EXPECT_EQ(1U, runner->NumPendingTasks()); |
| 53 EXPECT_EQ(0, i); | 53 EXPECT_EQ(0, i); |
| 54 runner->RunPendingTasks(); | 54 runner->RunPendingTasks(); |
| 55 EXPECT_EQ(1, i); | 55 EXPECT_EQ(1, i); |
| 56 } | 56 } |
| 57 | 57 |
| 58 TEST(OneShotEventTest, PostDefaultsToCurrentMessageLoop) { | 58 TEST(OneShotEventTest, PostDefaultsToCurrentMessageLoop) { |
| 59 OneShotEvent event; | 59 OneShotEvent event; |
| 60 scoped_refptr<base::TestSimpleTaskRunner> runner( | 60 scoped_refptr<base::TestSimpleTaskRunner> runner( |
| 61 new base::TestSimpleTaskRunner); | 61 new base::TestSimpleTaskRunner); |
| 62 base::MessageLoop loop; | 62 base::MessageLoop loop; |
| 63 int runner_i = 0; | 63 int runner_i = 0; |
| 64 int loop_i = 0; | 64 int loop_i = 0; |
| 65 | 65 |
| 66 event.Post(FROM_HERE, base::Bind(&Increment, &runner_i), runner); | 66 event.Post(FROM_HERE, base::Bind(&Increment, &runner_i), runner); |
| 67 event.Post(FROM_HERE, base::Bind(&Increment, &loop_i)); | 67 event.Post(FROM_HERE, base::Bind(&Increment, &loop_i)); |
| 68 event.Signal(); | 68 event.Signal(); |
| 69 EXPECT_EQ(1U, runner->GetPendingTasks().size()); | 69 EXPECT_EQ(1U, runner->NumPendingTasks()); |
| 70 EXPECT_EQ(0, runner_i); | 70 EXPECT_EQ(0, runner_i); |
| 71 runner->RunPendingTasks(); | 71 runner->RunPendingTasks(); |
| 72 EXPECT_EQ(1, runner_i); | 72 EXPECT_EQ(1, runner_i); |
| 73 EXPECT_EQ(0, loop_i); | 73 EXPECT_EQ(0, loop_i); |
| 74 base::RunLoop().RunUntilIdle(); | 74 base::RunLoop().RunUntilIdle(); |
| 75 EXPECT_EQ(1, loop_i); | 75 EXPECT_EQ(1, loop_i); |
| 76 } | 76 } |
| 77 | 77 |
| 78 void CheckSignaledAndPostIncrement( | 78 void CheckSignaledAndPostIncrement( |
| 79 OneShotEvent* event, | 79 OneShotEvent* event, |
| 80 const scoped_refptr<base::SingleThreadTaskRunner>& runner, | 80 const scoped_refptr<base::SingleThreadTaskRunner>& runner, |
| 81 int* i) { | 81 int* i) { |
| 82 EXPECT_TRUE(event->is_signaled()); | 82 EXPECT_TRUE(event->is_signaled()); |
| 83 event->Post(FROM_HERE, base::Bind(&Increment, i), runner); | 83 event->Post(FROM_HERE, base::Bind(&Increment, i), runner); |
| 84 } | 84 } |
| 85 | 85 |
| 86 TEST(OneShotEventTest, IsSignaledAndPostsFromCallbackWork) { | 86 TEST(OneShotEventTest, IsSignaledAndPostsFromCallbackWork) { |
| 87 OneShotEvent event; | 87 OneShotEvent event; |
| 88 scoped_refptr<base::TestSimpleTaskRunner> runner( | 88 scoped_refptr<base::TestSimpleTaskRunner> runner( |
| 89 new base::TestSimpleTaskRunner); | 89 new base::TestSimpleTaskRunner); |
| 90 int i = 0; | 90 int i = 0; |
| 91 | 91 |
| 92 event.Post(FROM_HERE, | 92 event.Post(FROM_HERE, |
| 93 base::Bind(&CheckSignaledAndPostIncrement, &event, runner, &i), | 93 base::Bind(&CheckSignaledAndPostIncrement, &event, runner, &i), |
| 94 runner); | 94 runner); |
| 95 EXPECT_EQ(0, i); | 95 EXPECT_EQ(0, i); |
| 96 event.Signal(); | 96 event.Signal(); |
| 97 | 97 |
| 98 // CheckSignaledAndPostIncrement is queued on |runner|. | 98 // CheckSignaledAndPostIncrement is queued on |runner|. |
| 99 EXPECT_EQ(1U, runner->GetPendingTasks().size()); | 99 EXPECT_EQ(1U, runner->NumPendingTasks()); |
| 100 EXPECT_EQ(0, i); | 100 EXPECT_EQ(0, i); |
| 101 runner->RunPendingTasks(); | 101 runner->RunPendingTasks(); |
| 102 // Increment is queued on |runner|. | 102 // Increment is queued on |runner|. |
| 103 EXPECT_EQ(1U, runner->GetPendingTasks().size()); | 103 EXPECT_EQ(1U, runner->NumPendingTasks()); |
| 104 EXPECT_EQ(0, i); | 104 EXPECT_EQ(0, i); |
| 105 runner->RunPendingTasks(); | 105 runner->RunPendingTasks(); |
| 106 // Increment has run. | 106 // Increment has run. |
| 107 EXPECT_EQ(0U, runner->GetPendingTasks().size()); | 107 EXPECT_EQ(0U, runner->NumPendingTasks()); |
| 108 EXPECT_EQ(1, i); | 108 EXPECT_EQ(1, i); |
| 109 } | 109 } |
| 110 | 110 |
| 111 } // namespace | 111 } // namespace |
| 112 } // namespace extensions | 112 } // namespace extensions |
| OLD | NEW |