9f086f0600
Bug: webrtc:469327588 Change-Id: I5306d9aaa98e12522d829f85cd405ce76a6a6964 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/461341 Commit-Queue: Jeremy Leconte <jleconte@google.com> Reviewed-by: Jeremy Leconte <jleconte@google.com> Auto-Submit: Evan Shrubsole <eshr@webrtc.org> Cr-Commit-Position: refs/heads/main@{#47305}
83 lines
2.4 KiB
C++
83 lines
2.4 KiB
C++
/*
|
|
* Copyright 2024 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "test/wait_until.h"
|
|
|
|
#include <variant>
|
|
|
|
#include "absl/functional/overload.h"
|
|
#include "api/function_view.h"
|
|
#include "api/test/time_controller.h"
|
|
#include "api/units/time_delta.h"
|
|
#include "api/units/timestamp.h"
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/thread.h"
|
|
#include "rtc_base/time_utils.h"
|
|
#include "system_wrappers/include/clock.h"
|
|
|
|
namespace webrtc {
|
|
|
|
[[nodiscard]] bool WaitUntil(FunctionView<bool()> fn,
|
|
WaitUntilSettings settings) {
|
|
if (std::holds_alternative<std::monostate>(settings.clock)) {
|
|
RTC_CHECK(Thread::Current()) << "A current thread is required. An "
|
|
"webrtc::test::RunLoop can work for tests.";
|
|
}
|
|
|
|
auto now = [&] {
|
|
return std::visit(
|
|
absl::Overload{
|
|
[](const std::monostate&) {
|
|
return Timestamp::Micros(TimeMicros());
|
|
},
|
|
[](SimulatedClock* clock) { return clock->CurrentTime(); },
|
|
[](TimeController* time_controller) {
|
|
return time_controller->GetClock()->CurrentTime();
|
|
},
|
|
[](auto* clock) {
|
|
return Timestamp::Micros(clock->TimeNanos() / 1000);
|
|
},
|
|
},
|
|
settings.clock);
|
|
};
|
|
|
|
auto sleep = [&](TimeDelta delta) {
|
|
std::visit(absl::Overload{
|
|
[&](const std::monostate&) {
|
|
Thread::Current()->ProcessMessages(0);
|
|
Thread::Current()->SleepMs(delta.ms());
|
|
},
|
|
[&](auto* clock) { clock->AdvanceTime(delta); },
|
|
},
|
|
settings.clock);
|
|
};
|
|
|
|
if (fn()) {
|
|
return true;
|
|
}
|
|
|
|
Timestamp deadline = now() + settings.timeout;
|
|
|
|
// Run pending tasks first as they might change result of the `fn` and
|
|
// thus avoid unnecessary advancing time.
|
|
sleep(TimeDelta::Zero());
|
|
|
|
for (;;) {
|
|
if (fn()) {
|
|
return true;
|
|
} else if (now() >= deadline) {
|
|
return false;
|
|
}
|
|
sleep(settings.polling_interval);
|
|
}
|
|
}
|
|
|
|
} // namespace webrtc
|