summaryrefslogtreecommitdiff
path: root/components/policy/core/common/policy_scheduler_unittest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'components/policy/core/common/policy_scheduler_unittest.cc')
-rw-r--r--components/policy/core/common/policy_scheduler_unittest.cc133
1 files changed, 0 insertions, 133 deletions
diff --git a/components/policy/core/common/policy_scheduler_unittest.cc b/components/policy/core/common/policy_scheduler_unittest.cc
deleted file mode 100644
index 2950f3f112..0000000000
--- a/components/policy/core/common/policy_scheduler_unittest.cc
+++ /dev/null
@@ -1,133 +0,0 @@
-// 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/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_ = std::make_unique<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_ = std::make_unique<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_ = std::make_unique<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_ = std::make_unique<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