blob: 8f05dcb01fbf09a59739189bb456cf6fce5974ce [file] [log] [blame]
// Copyright 2017 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "peridot/bin/ledger/testing/test_with_environment.h"
#include <lib/fit/function.h>
#include <lib/timekeeper/test_loop_test_clock.h>
#include "peridot/lib/rng/test_random.h"
namespace ledger {
namespace {
// Wrapper around a real CoroutineHandler for test.
//
// The wrapper allows to delay re-entering the coroutine body when the run loop
// is running. When |Resume| is called, it quits the loop, and the main method
// calls |ResumeIfNeeded| when the loop exits.
class TestCoroutineHandler : public coroutine::CoroutineHandler {
public:
explicit TestCoroutineHandler(coroutine::CoroutineHandler* delegate,
fit::closure quit_callback)
: delegate_(delegate), quit_callback_(std::move(quit_callback)) {}
coroutine::ContinuationStatus Yield() override { return delegate_->Yield(); }
void Resume(coroutine::ContinuationStatus status) override {
// If interrupting, no need to delay the call as the test will not run the
// loop itself.
if (status == coroutine::ContinuationStatus::INTERRUPTED) {
delegate_->Resume(status);
return;
}
quit_callback_();
need_to_continue_ = true;
}
// Re-enters the coroutine body if the handler delayed the call.
void ResumeIfNeeded() {
if (need_to_continue_) {
need_to_continue_ = false;
delegate_->Resume(coroutine::ContinuationStatus::OK);
}
}
private:
coroutine::CoroutineHandler* delegate_;
fit::closure quit_callback_;
bool need_to_continue_ = false;
};
} // namespace
TestWithEnvironment::TestWithEnvironment()
: environment_(
EnvironmentBuilder()
.SetAsync(dispatcher())
.SetIOAsync(dispatcher())
.SetClock(
std::make_unique<timekeeper::TestLoopTestClock>(&test_loop()))
.SetRandom(std::make_unique<rng::TestRandom>(
test_loop().initial_state()))
.Build()) {}
void TestWithEnvironment::RunInCoroutine(
fit::function<void(coroutine::CoroutineHandler*)> run_test) {
std::unique_ptr<TestCoroutineHandler> test_handler;
volatile bool ended = false;
environment_.coroutine_service()->StartCoroutine(
[&](coroutine::CoroutineHandler* handler) {
test_handler = std::make_unique<TestCoroutineHandler>(
handler, [this] { QuitLoop(); });
run_test(test_handler.get());
ended = true;
});
while (!ended) {
test_handler->ResumeIfNeeded();
RunLoopUntilIdle();
}
}
} // namespace ledger