Index: components/policy/core/common/policy_scheduler_unittest.cc |
diff --git a/components/policy/core/common/policy_scheduler_unittest.cc b/components/policy/core/common/policy_scheduler_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..cb5758cab775e1f39fba7dc016a8cd1a4f73cf44 |
--- /dev/null |
+++ b/components/policy/core/common/policy_scheduler_unittest.cc |
@@ -0,0 +1,134 @@ |
+// Copyright 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "components/policy/core/common/policy_scheduler.h" |
+ |
+#include <memory> |
+ |
+#include "base/bind.h" |
+#include "base/memory/ptr_util.h" |
+#include "base/run_loop.h" |
+#include "base/test/scoped_task_environment.h" |
+#include "base/threading/thread_task_runner_handle.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace policy { |
+ |
+class PolicySchedulerTest : public testing::Test { |
+ public: |
+ void DoTask(PolicyScheduler::TaskCallback callback) { |
+ do_counter_++; |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
+ FROM_HERE, base::BindOnce(std::move(callback), true)); |
+ } |
+ |
+ void OnTaskDone(bool success) { |
+ done_counter_++; |
+ |
+ // Terminate PolicyScheduler after 5 iterations. |
+ if (done_counter_ >= 5) { |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
+ FROM_HERE, base::BindOnce(&PolicySchedulerTest::Terminate, |
+ base::Unretained(this))); |
+ } |
+ } |
+ |
+ // To simulate a slow task the callback is captured instead of running it. |
+ void CaptureCallbackForSlowTask(PolicyScheduler::TaskCallback callback) { |
+ do_counter_++; |
+ slow_callback_ = std::move(callback); |
+ } |
+ |
+ // Runs the captured callback to simulate the end of the slow task. |
+ void PostSlowTaskCallback() { |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
+ FROM_HERE, base::BindOnce(std::move(slow_callback_), true)); |
+ } |
+ |
+ void Terminate() { scheduler_.reset(); } |
+ |
+ protected: |
+ int do_counter_ = 0; |
+ int done_counter_ = 0; |
+ std::unique_ptr<PolicyScheduler> scheduler_; |
+ |
+ PolicyScheduler::TaskCallback slow_callback_; |
+ |
+ base::test::ScopedTaskEnvironment scoped_task_environment_; |
+}; |
+ |
+TEST_F(PolicySchedulerTest, Run) { |
+ scheduler_ = base::MakeUnique<PolicyScheduler>( |
+ base::BindRepeating(&PolicySchedulerTest::DoTask, base::Unretained(this)), |
+ base::BindRepeating(&PolicySchedulerTest::OnTaskDone, |
+ base::Unretained(this)), |
+ base::TimeDelta::Max()); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, done_counter_); |
+} |
+ |
+TEST_F(PolicySchedulerTest, Loop) { |
+ scheduler_ = base::MakeUnique<PolicyScheduler>( |
+ base::BindRepeating(&PolicySchedulerTest::DoTask, base::Unretained(this)), |
+ base::BindRepeating(&PolicySchedulerTest::OnTaskDone, |
+ base::Unretained(this)), |
+ base::TimeDelta()); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(5, done_counter_); |
+} |
+ |
+TEST_F(PolicySchedulerTest, Reschedule) { |
+ scheduler_ = base::MakeUnique<PolicyScheduler>( |
+ base::BindRepeating(&PolicySchedulerTest::DoTask, base::Unretained(this)), |
+ base::BindRepeating(&PolicySchedulerTest::OnTaskDone, |
+ base::Unretained(this)), |
+ base::TimeDelta::Max()); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, done_counter_); |
+ |
+ // Delayed action is not run. |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, done_counter_); |
+ |
+ // Rescheduling with 0 delay causes it to run. |
+ scheduler_->ScheduleTaskNow(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(2, done_counter_); |
+} |
+ |
+TEST_F(PolicySchedulerTest, OverlappingTasks) { |
+ scheduler_ = base::MakeUnique<PolicyScheduler>( |
+ base::BindRepeating(&PolicySchedulerTest::CaptureCallbackForSlowTask, |
+ base::Unretained(this)), |
+ base::BindRepeating(&PolicySchedulerTest::OnTaskDone, |
+ base::Unretained(this)), |
+ base::TimeDelta::Max()); |
+ |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, do_counter_); |
+ EXPECT_EQ(0, done_counter_); |
+ |
+ // Second action doesn't start while first is still pending. |
+ scheduler_->ScheduleTaskNow(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(1, do_counter_); |
+ EXPECT_EQ(0, done_counter_); |
+ |
+ // After first action has finished, the second is started. |
+ PostSlowTaskCallback(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(2, do_counter_); |
+ EXPECT_EQ(1, done_counter_); |
+ |
+ // Let the second action finish. |
+ PostSlowTaskCallback(); |
+ base::RunLoop().RunUntilIdle(); |
+ EXPECT_EQ(2, do_counter_); |
+ EXPECT_EQ(2, done_counter_); |
+} |
+ |
+} // namespace policy |