1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79


base / task / default_delayed_task_handle_delegate_unittest.cc [blame]

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/task/default_delayed_task_handle_delegate.h"

#include "base/functional/callback_helpers.h"
#include "base/task/delayed_task_handle.h"
#include "base/test/bind.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace base {

// Tests that running the bound callback invalidates the handle.
TEST(DefaultDelayedTaskHandleDelegateTest, RunTask) {
  auto delegate = std::make_unique<DefaultDelayedTaskHandleDelegate>();
  EXPECT_FALSE(delegate->IsValid());

  auto bound_callback = delegate->BindCallback(DoNothing());
  EXPECT_TRUE(delegate->IsValid());

  DelayedTaskHandle delayed_task_handle(std::move(delegate));
  EXPECT_TRUE(delayed_task_handle.IsValid());

  std::move(bound_callback).Run();
  EXPECT_FALSE(delayed_task_handle.IsValid());
}

// Tests that DefaultDelayedTaskHandleDelegate supports CancelTask().
TEST(DefaultDelayedTaskHandleDelegateTest, CancelTask) {
  auto delegate = std::make_unique<DefaultDelayedTaskHandleDelegate>();
  EXPECT_FALSE(delegate->IsValid());

  auto bound_callback = delegate->BindCallback(DoNothing());
  EXPECT_TRUE(delegate->IsValid());

  DelayedTaskHandle delayed_task_handle(std::move(delegate));
  EXPECT_TRUE(delayed_task_handle.IsValid());

  delayed_task_handle.CancelTask();
  EXPECT_FALSE(delayed_task_handle.IsValid());
  EXPECT_TRUE(bound_callback.IsCancelled());
}

// Tests that destroying the bound callback invalidates the handle.
TEST(DefaultDelayedTaskHandleDelegateTest, DestroyTask) {
  auto delegate = std::make_unique<DefaultDelayedTaskHandleDelegate>();
  EXPECT_FALSE(delegate->IsValid());

  auto bound_callback = delegate->BindCallback(DoNothing());
  EXPECT_TRUE(delegate->IsValid());

  DelayedTaskHandle delayed_task_handle(std::move(delegate));
  EXPECT_TRUE(delayed_task_handle.IsValid());

  bound_callback = OnceClosure();
  EXPECT_FALSE(delayed_task_handle.IsValid());
}

// Tests that the handle is invalid while the task is running.
TEST(DefaultDelayedTaskHandleDelegateTest, HandleInvalidInsideCallback) {
  auto delegate = std::make_unique<DefaultDelayedTaskHandleDelegate>();
  EXPECT_FALSE(delegate->IsValid());

  auto bound_callback = delegate->BindCallback(
      BindLambdaForTesting([delegate = delegate.get()]() {
        // Check that the delegate is invalid inside the callback.
        EXPECT_FALSE(delegate->IsValid());
      }));
  EXPECT_TRUE(delegate->IsValid());

  DelayedTaskHandle delayed_task_handle(std::move(delegate));
  EXPECT_TRUE(delayed_task_handle.IsValid());

  std::move(bound_callback).Run();
  EXPECT_FALSE(delayed_task_handle.IsValid());
}

}  // namespace base