// Copyright 2018 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 WEBRUNNER_COMMON_TEST_RUN_WITH_TIMEOUT_H_
#define WEBRUNNER_COMMON_TEST_RUN_WITH_TIMEOUT_H_

#include "base/macros.h"
#include "base/run_loop.h"
#include "base/threading/thread_task_runner_handle.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace webrunner {

void CheckRunWithTimeout(base::RunLoop* run_loop,
                         const base::TimeDelta& timeout) {
  bool did_timeout = false;
  base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
      FROM_HERE,
      base::BindOnce(
          [](bool* did_timeout, base::OnceClosure quit_closure) {
            *did_timeout = true;
            std::move(quit_closure).Run();
          },
          base::Unretained(&did_timeout), run_loop->QuitClosure()),
      timeout);
  run_loop->Run();
  if (did_timeout)
    ADD_FAILURE() << "Timeout: RunLoop did not Quit() within "
                  << timeout.InSecondsF() << " seconds.";
}

}  // namespace webrunner

#endif  // WEBRUNNER_COMMON_RUNLOOP_WITH_DEADLINE_H_