[Chromecast] Add an alarm manager for firing events on wall clock time.

chromecast/base/alarm_manager_unittest.cc

+// 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 "chromecast/base/alarm_manager.h"
+
+#include "base/memory/ptr_util.h"
+#include "base/message_loop/message_loop.h"
+#include "base/test/simple_test_clock.h"
+#include "base/test/test_mock_time_task_runner.h"
+#include "base/time/clock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace chromecast {
+class AlarmManagerTest : public ::testing::Test {
+ protected:
+  class WallClockDependantTask : public AlarmManager::AlarmDelegate {
+   public:
+    WallClockDependantTask() : fired_(false) {}
+    void OnAlarmFire() override { fired_ = true; }
+    bool fired_;
+  };
+  void SetUp() override {
+    message_loop_.reset(new base::MessageLoop);
+    task_runner_ = new base::TestMockTimeTaskRunner;
+    message_loop_->SetTaskRunner(task_runner_);
+  }
+
+  void TearDown() override {
+    task_runner_ = nullptr;
+    message_loop_.reset();
+  }
+
+  std::unique_ptr<base::MessageLoop> message_loop_;
+  scoped_refptr<base::TestMockTimeTaskRunner> task_runner_;
+};
+
+TEST_F(AlarmManagerTest, AddAlarm) {
+  WallClockDependantTask task;
+  ASSERT_FALSE(task.fired_);
+  std::unique_ptr<AlarmManager> manager = base::MakeUnique<AlarmManager>();
+  base::Time alarm_time = base::Time::Now() + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task, alarm_time);
+  manager->RemoveAlarm(&task);
+  ASSERT_FALSE(task.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmNotFire) {
+  WallClockDependantTask task;
+  ASSERT_FALSE(task.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task, alarm_time);
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(9));
+  clock->Advance(base::TimeDelta::FromMinutes(9));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmFire) {
+  WallClockDependantTask task;
+  ASSERT_FALSE(task.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add an alarm.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task, alarm_time);
+  clock->Advance(base::TimeDelta::FromMinutes(10));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(10));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task.fired_);
+
+  // Fires only once.
+  task.fired_ = false;
+  clock->Advance(base::TimeDelta::FromMinutes(10));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(10));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmPast) {
+  WallClockDependantTask task;
+  ASSERT_FALSE(task.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add an alarm in the past. Should fire right away.
+  base::Time alarm_time = base::Time::Now() - base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task, alarm_time);
+  task_runner_->FastForwardBy(base::TimeDelta::FromSeconds(10));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmTimeJump) {
+  WallClockDependantTask task;
+  ASSERT_FALSE(task.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add an alarm. The time jumps to the future.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task, alarm_time);
+  clock->Advance(base::TimeDelta::FromMinutes(10));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmJumpFuture) {
+  WallClockDependantTask task;
+  ASSERT_FALSE(task.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add an alarm. The time jumps far into the future.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task, alarm_time);
+  clock->Advance(base::TimeDelta::FromMinutes(60));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmMultiple) {
+  WallClockDependantTask task1;
+  WallClockDependantTask task2;
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add first task.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add second task.
+  alarm_time = now + base::TimeDelta::FromMinutes(12);
+  manager->AddAlarm(&task2, alarm_time);
+
+  // First task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(10));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+
+  // Reset state;
+  task1.fired_ = false;
+  task2.fired_ = false;
+
+  // Second task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(2));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_TRUE(task2.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmMultipleReverseOrder) {
+  WallClockDependantTask task1;
+  WallClockDependantTask task2;
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add first task.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(12);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add second task.
+  alarm_time = now + base::TimeDelta::FromMinutes(10);
+  manager->AddAlarm(&task2, alarm_time);
+
+  // Second task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(10));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_TRUE(task2.fired_);
+
+  // Reset state;
+  task1.fired_ = false;
+  task2.fired_ = false;
+
+  // First task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(2));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmMultipleSameTime) {
+  WallClockDependantTask task1;
+  WallClockDependantTask task2;
+  WallClockDependantTask task3;
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_FALSE(task3.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add first task.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(12);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add second task.
+  alarm_time = now + base::TimeDelta::FromMinutes(16);
+  manager->AddAlarm(&task2, alarm_time);
+
+  // Add third task.
+  alarm_time = now + base::TimeDelta::FromMinutes(12);
+  manager->AddAlarm(&task3, alarm_time);
+
+  // First and third task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(12));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_TRUE(task3.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmMultipleShuffle) {
+  WallClockDependantTask task1;
+  WallClockDependantTask task2;
+  WallClockDependantTask task3;
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_FALSE(task3.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add first task.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(15);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add second task.
+  alarm_time = now + base::TimeDelta::FromMinutes(16);
+  manager->AddAlarm(&task2, alarm_time);
+
+  // Add third task.
+  alarm_time = now + base::TimeDelta::FromMinutes(11);
+  manager->AddAlarm(&task3, alarm_time);
+
+  // Third task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(12));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_TRUE(task3.fired_);
+
+  clock->Advance(base::TimeDelta::FromMinutes(3));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_TRUE(task3.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmReplace) {
+  WallClockDependantTask task1;
+  WallClockDependantTask task2;
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add first task.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(15);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add it again with less time.
+  alarm_time = now + base::TimeDelta::FromMinutes(1);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add second task.
+  alarm_time = now + base::TimeDelta::FromMinutes(16);
+  manager->AddAlarm(&task2, alarm_time);
+
+  // Third task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(1));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_TRUE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+
+  task1.fired_ = false;
+  task2.fired_ = false;
+
+  // First task should not fire again at original time.
+  clock->Advance(base::TimeDelta::FromMinutes(14));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+}
+
+TEST_F(AlarmManagerTest, AlarmCancel) {
+  WallClockDependantTask task1;
+  WallClockDependantTask task2;
+  WallClockDependantTask task3;
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_FALSE(task3.fired_);
+
+  // Create the AlarmManager.
+  base::Time now = base::Time::Now();
+  std::unique_ptr<base::SimpleTestClock> test_clock =
+      base::MakeUnique<base::SimpleTestClock>();
+  test_clock->SetNow(now);
+  base::SimpleTestClock* clock = test_clock.get();
+  std::unique_ptr<AlarmManager> manager =
+      base::MakeUnique<AlarmManager>(std::move(test_clock));
+
+  // Add first task.
+  base::Time alarm_time = now + base::TimeDelta::FromMinutes(12);
+  manager->AddAlarm(&task1, alarm_time);
+
+  // Add second task.
+  alarm_time = now + base::TimeDelta::FromMinutes(16);
+  manager->AddAlarm(&task2, alarm_time);
+
+  // Add third task.
+  alarm_time = now + base::TimeDelta::FromMinutes(12);
+  manager->AddAlarm(&task3, alarm_time);
+
+  // Remove the first task.
+  manager->RemoveAlarm(&task1);
+
+  // First and third task should fire.
+  clock->Advance(base::TimeDelta::FromMinutes(15));
+  task_runner_->FastForwardBy(base::TimeDelta::FromMinutes(1));
+  task_runner_->RunUntilIdle();
+  ASSERT_FALSE(task1.fired_);
+  ASSERT_FALSE(task2.fired_);
+  ASSERT_TRUE(task3.fired_);
+}
+}