// 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.

#ifndef BASE_TEST_SCOPED_ASYNC_TASK_SCHEDULER_H_
#define BASE_TEST_SCOPED_ASYNC_TASK_SCHEDULER_H_

#include "base/macros.h"
#include "base/task_scheduler/lazy_task_runner.h"

namespace base {

class TaskScheduler;

namespace test {

// DEPRECATED. Use ScopedTaskEnvironment instead.
//
// TODO(fdoray): Replace ScopedAsyncTaskScheduler instances by
// ScopedTaskEnvironment. https://crbug.com/708584
//
// Allows usage of the base/task_scheduler/post_task.h API within its scope.
//
// To wait until all tasks posted without a delay have run, use
// TaskScheduler::GetInstance()->FlushForTesting().
class ScopedAsyncTaskScheduler {
 public:
  // Registers a single-threaded TaskScheduler.
  ScopedAsyncTaskScheduler();

  // Shuts down and unregisters the TaskScheduler.
  //
  // It is guaranteed that all BLOCK_SHUTDOWN tasks have run when this returns.
  ~ScopedAsyncTaskScheduler();

 private:
  const TaskScheduler* task_scheduler_ = nullptr;

  // Ensures destruction of lazy TaskRunners when this is destroyed.
  internal::ScopedLazyTaskRunnerListForTesting
      scoped_lazy_task_runner_list_for_testing_;

  DISALLOW_COPY_AND_ASSIGN(ScopedAsyncTaskScheduler);
};

}  // namespace test
}  // namespace base

#endif  // BASE_TEST_SCOPED_ASYNC_TASK_SCHEDULER_H_