Make tasks cancellable inside the blink scheduler.

third_party/WebKit/Source/platform/scheduler/base/time_domain_unittest.cc

 // Copyright 2015 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 "platform/scheduler/base/time_domain.h"
 
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/test/simple_test_tick_clock.h"
 #include "cc/test/ordered_simple_task_runner.h"
@@ skipping 12 matching lines @@
 namespace scheduler {
 
 class MockTimeDomain : public TimeDomain {
  public:
   explicit MockTimeDomain(TimeDomain::Observer* observer)
       : TimeDomain(observer),
         now_(base::TimeTicks() + base::TimeDelta::FromSeconds(1)) {}
 
   ~MockTimeDomain() override {}
 
+  using TimeDomain::CancelDelayedWork;
   using TimeDomain::ClearExpiredWakeups;
   using TimeDomain::NextScheduledRunTime;
   using TimeDomain::NextScheduledTaskQueue;
   using TimeDomain::ScheduleDelayedWork;
   using TimeDomain::UnregisterQueue;
   using TimeDomain::UpdateWorkQueues;
   using TimeDomain::RegisterAsUpdatableTaskQueue;
 
   // TimeSource implementation:
   LazyNow CreateLazyNow() const override { return LazyNow(now_); }
@@ skipping 146 matching lines @@
   time_domain_->ClearExpiredWakeups();
 
   ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
   EXPECT_EQ(run_time2, next_run_time);
 
   time_domain_->SetNow(run_time2);
   time_domain_->ClearExpiredWakeups();
   ASSERT_FALSE(time_domain_->NextScheduledRunTime(&next_run_time));
 }
 
+TEST_F(TimeDomainTest, CancelDelayedWork) {
+  base::TimeDelta delay = base::TimeDelta::FromMilliseconds(50);
+  EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay));
+  base::TimeTicks now = time_domain_->Now();
+  base::TimeTicks delayed_runtime = now + delay;
+  time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime, now);
+
+  base::TimeTicks next_run_time;
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime, next_run_time);
+
+  time_domain_->CancelDelayedWork(task_queue_.get(), delayed_runtime);
+  EXPECT_FALSE(time_domain_->NextScheduledRunTime(&next_run_time));
+}
+
+TEST_F(TimeDomainTest, CancelDelayedWorkTaskQueueDoesntMatch) {
+  base::TimeDelta delay = base::TimeDelta::FromMilliseconds(50);
+  EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay));
+  base::TimeTicks now = time_domain_->Now();
+  base::TimeTicks delayed_runtime = now + delay;
+  time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime, now);
+
+  base::TimeTicks next_run_time;
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime, next_run_time);
+
+  time_domain_->CancelDelayedWork(nullptr, delayed_runtime);
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime, next_run_time);
+}
+
+TEST_F(TimeDomainTest, CancelDelayedWorkTwoWakeUpsCancelFirst) {
+  base::TimeDelta delay1 = base::TimeDelta::FromMilliseconds(50);
+  base::TimeDelta delay2 = base::TimeDelta::FromMilliseconds(150);
+  EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay1));
+  base::TimeTicks now = time_domain_->Now();
+  base::TimeTicks delayed_runtime1 = now + delay1;
+  base::TimeTicks delayed_runtime2 = now + delay2;
+  time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime1, now);
+  time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime2, now);
+
+  base::TimeTicks next_run_time;
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime1, next_run_time);
+  Mock::VerifyAndClearExpectations(time_domain_.get());
+
+  EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay2));
+  time_domain_->CancelDelayedWork(task_queue_.get(), delayed_runtime1);
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime2, next_run_time);
+}
+
+TEST_F(TimeDomainTest, CancelDelayedWorkTwoWakeUpsCancelSecond) {
+  base::TimeDelta delay1 = base::TimeDelta::FromMilliseconds(50);
+  base::TimeDelta delay2 = base::TimeDelta::FromMilliseconds(150);
+  EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, delay1));
+  base::TimeTicks now = time_domain_->Now();
+  base::TimeTicks delayed_runtime1 = now + delay1;
+  base::TimeTicks delayed_runtime2 = now + delay2;
+  time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime1, now);
+  time_domain_->ScheduleDelayedWork(task_queue_.get(), delayed_runtime2, now);
+
+  base::TimeTicks next_run_time;
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime1, next_run_time);
+  Mock::VerifyAndClearExpectations(time_domain_.get());
+
+  time_domain_->CancelDelayedWork(task_queue_.get(), delayed_runtime2);
+  ASSERT_TRUE(time_domain_->NextScheduledRunTime(&next_run_time));
+  EXPECT_EQ(delayed_runtime1, next_run_time);
+
+  time_domain_->SetNow(delayed_runtime1);
+  LazyNow lazy_now = time_domain_->CreateLazyNow();
+  time_domain_->UpdateWorkQueues(false, nullptr, lazy_now);
+  ASSERT_FALSE(time_domain_->NextScheduledRunTime(&next_run_time));
+}
+
 namespace {
 class MockObserver : public TimeDomain::Observer {
  public:
   ~MockObserver() override {}
 
   MOCK_METHOD0(OnTimeDomainHasImmediateWork, void());
   MOCK_METHOD0(OnTimeDomainHasDelayedWork, void());
 };
 }  // namespace
 
@@ skipping 15 matching lines @@
 TEST_F(TimeDomainWithObserverTest, OnTimeDomainHasDelayedWork) {
   EXPECT_CALL(*observer_, OnTimeDomainHasDelayedWork());
   EXPECT_CALL(*time_domain_.get(), RequestWakeup(_, _));
   base::TimeTicks now = time_domain_->Now();
   time_domain_->ScheduleDelayedWork(
       task_queue_.get(), now + base::TimeDelta::FromMilliseconds(10), now);
 }
 
 }  // namespace scheduler
 }  // namespace blink