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


base / test / run_until.cc [blame]

// Copyright 2023 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/test/run_until.h"

#include <functional>

#include "base/callback_list.h"
#include "base/functional/callback.h"
#include "base/task/current_thread.h"
#include "base/test/scoped_run_loop_timeout.h"
#include "base/test/test_future.h"
#include "base/time/time_override.h"

namespace base::test {

void TestPredicateOrRegisterOnNextIdleCallback(
    base::FunctionRef<bool(void)> condition,
    CallbackListSubscription* on_idle_callback_subscription,
    OnceClosure ready_callback) {
  if (condition()) {
    // Invoke `ready_callback` if `condition` evaluates to true.
    std::move(ready_callback).Run();
  } else {
    // Otherwise try again the next time the thread is idle.
    *on_idle_callback_subscription =
        CurrentThread::Get().RegisterOnNextIdleCallback(
            {},
            BindOnce(TestPredicateOrRegisterOnNextIdleCallback, condition,
                     on_idle_callback_subscription, std::move(ready_callback)));
  }
}

bool RunUntil(base::FunctionRef<bool(void)> condition) {
  // We expect a RunLoop timeout except under MOCK_TIME where TaskEnvironment
  // disables TaskEnvironment::mock_time_domain_ after fast-forwarding into the
  // timeout.
  CHECK(test::ScopedRunLoopTimeout::ExistsForCurrentThread() ||
        subtle::ScopedTimeClockOverrides::overrides_active())
      << "No RunLoop timeout set, meaning `RunUntil` will hang forever on "
         "failure.";

  test::TestFuture<void> ready_signal;

  CallbackListSubscription on_idle_callback_subscription;
  on_idle_callback_subscription =
      CurrentThread::Get().RegisterOnNextIdleCallback(
          {},
          BindOnce(TestPredicateOrRegisterOnNextIdleCallback, condition,
                   &on_idle_callback_subscription, ready_signal.GetCallback()));

  return ready_signal.Wait();
}

}  // namespace base::test