mirror of
https://github.com/XRPLF/clio.git
synced 2025-11-19 19:25:53 +00:00
304
tests/unit/util/async/AnyExecutionContextTests.cpp
Normal file
304
tests/unit/util/async/AnyExecutionContextTests.cpp
Normal file
@@ -0,0 +1,304 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
This file is part of clio: https://github.com/XRPLF/clio
|
||||
Copyright (c) 2024, the clio developers.
|
||||
|
||||
Permission to use, copy, modify, and distribute this software for any
|
||||
purpose with or without fee is hereby granted, provided that the above
|
||||
copyright notice and this permission notice appear in all copies.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
//==============================================================================
|
||||
|
||||
#include "util/MockExecutionContext.hpp"
|
||||
#include "util/MockOperation.hpp"
|
||||
#include "util/MockStrand.hpp"
|
||||
#include "util/async/AnyExecutionContext.hpp"
|
||||
#include "util/async/AnyOperation.hpp"
|
||||
#include "util/async/AnyStopToken.hpp"
|
||||
#include "util/async/AnyStrand.hpp"
|
||||
|
||||
#include <gmock/gmock.h>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <any>
|
||||
#include <chrono>
|
||||
#include <functional>
|
||||
#include <type_traits>
|
||||
|
||||
using namespace util::async;
|
||||
using namespace ::testing;
|
||||
|
||||
struct AnyExecutionContextTests : Test {
|
||||
using StrandType = NiceMock<MockStrand>;
|
||||
|
||||
template <typename T>
|
||||
using OperationType = NiceMock<MockOperation<T>>;
|
||||
|
||||
template <typename T>
|
||||
using StoppableOperationType = NiceMock<MockStoppableOperation<T>>;
|
||||
|
||||
template <typename T>
|
||||
using ScheduledOperationType = NiceMock<MockScheduledOperation<T>>;
|
||||
|
||||
NiceMock<MockExecutionContext> mockExecutionContext;
|
||||
AnyExecutionContext ctx{static_cast<MockExecutionContext&>(mockExecutionContext)};
|
||||
};
|
||||
|
||||
TEST_F(AnyExecutionContextTests, ExecuteWithoutTokenAndVoid)
|
||||
{
|
||||
auto mockOp = OperationType<std::any>{};
|
||||
EXPECT_CALL(mockExecutionContext, execute(An<std::function<std::any()>>())).WillOnce(ReturnRef(mockOp));
|
||||
EXPECT_CALL(mockOp, get());
|
||||
|
||||
auto op = ctx.execute([] { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<void>>);
|
||||
|
||||
ASSERT_TRUE(op.get());
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, ExecuteWithoutTokenAndVoidThrowsException)
|
||||
{
|
||||
auto mockOp = OperationType<std::any>{};
|
||||
EXPECT_CALL(mockExecutionContext, execute(An<std::function<std::any()>>()))
|
||||
.WillOnce([](auto&&) -> OperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = ctx.execute([] { throw 0; }));
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, ExecuteWithStopTokenAndVoid)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
EXPECT_CALL(mockExecutionContext, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce(ReturnRef(mockOp));
|
||||
EXPECT_CALL(mockOp, get());
|
||||
|
||||
auto op = ctx.execute([](auto) { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<void>>);
|
||||
|
||||
ASSERT_TRUE(op.get());
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, ExecuteWithStopTokenAndVoidThrowsException)
|
||||
{
|
||||
EXPECT_CALL(mockExecutionContext, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = ctx.execute([](auto) { throw 0; }));
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, ExecuteWithStopTokenAndReturnValue)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(mockExecutionContext, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto op = ctx.execute([](auto) -> int { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<int>>);
|
||||
|
||||
ASSERT_EQ(op.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, ExecuteWithStopTokenAndReturnValueThrowsException)
|
||||
{
|
||||
EXPECT_CALL(mockExecutionContext, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = ctx.execute([](auto) -> int { throw 0; }));
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, TimerCancellation)
|
||||
{
|
||||
auto mockScheduledOp = ScheduledOperationType<std::any>{};
|
||||
EXPECT_CALL(mockScheduledOp, cancel());
|
||||
EXPECT_CALL(
|
||||
mockExecutionContext,
|
||||
scheduleAfter(An<std::chrono::milliseconds>(), An<std::function<std::any(AnyStopToken)>>())
|
||||
)
|
||||
.WillOnce(ReturnRef(mockScheduledOp));
|
||||
|
||||
auto timer = ctx.scheduleAfter(std::chrono::milliseconds{12}, [](auto) { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(timer), AnyOperation<void>>);
|
||||
|
||||
timer.abort();
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, TimerExecuted)
|
||||
{
|
||||
auto mockScheduledOp = ScheduledOperationType<std::any>{};
|
||||
EXPECT_CALL(mockScheduledOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(
|
||||
mockExecutionContext,
|
||||
scheduleAfter(An<std::chrono::milliseconds>(), An<std::function<std::any(AnyStopToken)>>())
|
||||
)
|
||||
.WillOnce([&mockScheduledOp](auto, auto&&) -> ScheduledOperationType<std::any> const& {
|
||||
return mockScheduledOp;
|
||||
});
|
||||
|
||||
auto timer = ctx.scheduleAfter(std::chrono::milliseconds{12}, [](auto) -> int { throw 0; });
|
||||
|
||||
static_assert(std::is_same_v<decltype(timer), AnyOperation<int>>);
|
||||
EXPECT_EQ(timer.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, TimerWithBoolHandlerCancellation)
|
||||
{
|
||||
auto mockScheduledOp = ScheduledOperationType<std::any>{};
|
||||
EXPECT_CALL(mockScheduledOp, cancel());
|
||||
EXPECT_CALL(
|
||||
mockExecutionContext,
|
||||
scheduleAfter(An<std::chrono::milliseconds>(), An<std::function<std::any(AnyStopToken, bool)>>())
|
||||
)
|
||||
.WillOnce(ReturnRef(mockScheduledOp));
|
||||
|
||||
auto timer = ctx.scheduleAfter(std::chrono::milliseconds{12}, [](auto, bool) { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(timer), AnyOperation<void>>);
|
||||
|
||||
timer.abort();
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, TimerWithBoolHandlerExecuted)
|
||||
{
|
||||
auto mockScheduledOp = ScheduledOperationType<std::any>{};
|
||||
EXPECT_CALL(mockScheduledOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(
|
||||
mockExecutionContext,
|
||||
scheduleAfter(An<std::chrono::milliseconds>(), An<std::function<std::any(AnyStopToken, bool)>>())
|
||||
)
|
||||
.WillOnce([&mockScheduledOp](auto, auto&&) -> ScheduledOperationType<std::any> const& {
|
||||
return mockScheduledOp;
|
||||
});
|
||||
|
||||
auto timer = ctx.scheduleAfter(std::chrono::milliseconds{12}, [](auto, bool) -> int { throw 0; });
|
||||
|
||||
static_assert(std::is_same_v<decltype(timer), AnyOperation<int>>);
|
||||
EXPECT_EQ(timer.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithVoid)
|
||||
{
|
||||
auto mockOp = OperationType<std::any>{};
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockOp, get());
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any()>>())).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
auto op = strand.execute([] { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<void>>);
|
||||
|
||||
ASSERT_TRUE(op.get());
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithVoidThrowsException)
|
||||
{
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any()>>()))
|
||||
.WillOnce([](auto&&) -> OperationType<std::any> const& { throw 0; });
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([] {}));
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithReturnValue)
|
||||
{
|
||||
auto mockOp = OperationType<std::any>{};
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any()>>())).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
auto op = strand.execute([]() -> int { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<int>>);
|
||||
|
||||
EXPECT_EQ(op.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithReturnValueThrowsException)
|
||||
{
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any()>>()))
|
||||
.WillOnce([](auto&&) -> OperationType<std::any> const& { throw 0; });
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([]() -> int { throw 0; }));
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithStopTokenAndVoid)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockOp, get());
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _)).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
auto op = strand.execute([](auto) { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<void>>);
|
||||
|
||||
ASSERT_TRUE(op.get());
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithStopTokenAndVoidThrowsException)
|
||||
{
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([](auto) { throw 0; }));
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithStopTokenAndReturnValue)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _)).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
auto op = strand.execute([](auto) -> int { throw 0; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<int>>);
|
||||
|
||||
EXPECT_EQ(op.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyExecutionContextTests, StrandExecuteWithStopTokenAndReturnValueThrowsException)
|
||||
{
|
||||
auto mockStrand = StrandType{};
|
||||
EXPECT_CALL(mockExecutionContext, makeStrand()).WillOnce(ReturnRef(mockStrand));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
auto strand = ctx.makeStrand();
|
||||
static_assert(std::is_same_v<decltype(strand), AnyStrand>);
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([](auto) -> int { throw 0; }));
|
||||
}
|
||||
110
tests/unit/util/async/AnyOperationTests.cpp
Normal file
110
tests/unit/util/async/AnyOperationTests.cpp
Normal file
@@ -0,0 +1,110 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
This file is part of clio: https://github.com/XRPLF/clio
|
||||
Copyright (c) 2024, the clio developers.
|
||||
|
||||
Permission to use, copy, modify, and distribute this software for any
|
||||
purpose with or without fee is hereby granted, provided that the above
|
||||
copyright notice and this permission notice appear in all copies.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
//==============================================================================
|
||||
|
||||
#include "util/MockOperation.hpp"
|
||||
#include "util/async/AnyOperation.hpp"
|
||||
#include "util/async/Error.hpp"
|
||||
|
||||
#include <gmock/gmock.h>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <any>
|
||||
#include <expected>
|
||||
#include <string>
|
||||
|
||||
using namespace util::async;
|
||||
using namespace ::testing;
|
||||
|
||||
struct AnyOperationTests : Test {
|
||||
using OperationType = MockOperation<std::expected<std::any, ExecutionError>>;
|
||||
using StoppableOperationType = MockStoppableOperation<std::expected<std::any, ExecutionError>>;
|
||||
using ScheduledOperationType = MockScheduledOperation<std::expected<std::any, ExecutionError>>;
|
||||
|
||||
NaggyMock<OperationType> mockOp;
|
||||
NaggyMock<StoppableOperationType> mockStoppableOp;
|
||||
NaggyMock<ScheduledOperationType> mockScheduledOp;
|
||||
|
||||
AnyOperation<void> voidOp{impl::ErasedOperation(static_cast<OperationType&>(mockOp))};
|
||||
AnyOperation<void> voidStoppableOp{impl::ErasedOperation(static_cast<StoppableOperationType&>(mockStoppableOp))};
|
||||
AnyOperation<int> intOp{impl::ErasedOperation(static_cast<OperationType&>(mockOp))};
|
||||
AnyOperation<void> scheduledVoidOp{impl::ErasedOperation(static_cast<ScheduledOperationType&>(mockScheduledOp))};
|
||||
};
|
||||
using AnyOperationDeathTest = AnyOperationTests;
|
||||
|
||||
TEST_F(AnyOperationTests, VoidDataYieldsNoError)
|
||||
{
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::any{}));
|
||||
auto res = voidOp.get();
|
||||
ASSERT_TRUE(res);
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationTests, GetIntData)
|
||||
{
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
auto res = intOp.get();
|
||||
EXPECT_EQ(res.value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationTests, WaitCallPropagated)
|
||||
{
|
||||
StrictMock<MockFunction<void()>> callback;
|
||||
EXPECT_CALL(callback, Call());
|
||||
EXPECT_CALL(mockOp, wait()).WillOnce([&] { callback.Call(); });
|
||||
voidOp.wait();
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationTests, CancelAndRequestStopCallPropagated)
|
||||
{
|
||||
StrictMock<MockFunction<void()>> callback;
|
||||
EXPECT_CALL(callback, Call()).Times(2);
|
||||
EXPECT_CALL(mockScheduledOp, cancel()).WillOnce([&] { callback.Call(); });
|
||||
EXPECT_CALL(mockScheduledOp, requestStop()).WillOnce([&] { callback.Call(); });
|
||||
scheduledVoidOp.abort();
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationTests, RequestStopCallPropagatedOnStoppableOperation)
|
||||
{
|
||||
StrictMock<MockFunction<void()>> callback;
|
||||
EXPECT_CALL(callback, Call());
|
||||
EXPECT_CALL(mockStoppableOp, requestStop()).WillOnce([&] { callback.Call(); });
|
||||
voidStoppableOp.abort();
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationTests, GetPropagatesError)
|
||||
{
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::unexpected(ExecutionError{"tid", "Not good"})));
|
||||
auto res = intOp.get();
|
||||
ASSERT_FALSE(res);
|
||||
EXPECT_TRUE(res.error().message.ends_with("Not good"));
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationTests, GetIncorrectDataReturnsError)
|
||||
{
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<double>(4.2)));
|
||||
auto res = intOp.get();
|
||||
|
||||
ASSERT_FALSE(res);
|
||||
EXPECT_TRUE(res.error().message.ends_with("Bad any cast"));
|
||||
EXPECT_TRUE(std::string{res.error()}.ends_with("Bad any cast"));
|
||||
}
|
||||
|
||||
TEST_F(AnyOperationDeathTest, CallAbortOnNonStoppableOrCancellableOperation)
|
||||
{
|
||||
EXPECT_DEATH(voidOp.abort(), ".*");
|
||||
}
|
||||
68
tests/unit/util/async/AnyStopTokenTests.cpp
Normal file
68
tests/unit/util/async/AnyStopTokenTests.cpp
Normal file
@@ -0,0 +1,68 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
This file is part of clio: https://github.com/XRPLF/clio
|
||||
Copyright (c) 2024, the clio developers.
|
||||
|
||||
Permission to use, copy, modify, and distribute this software for any
|
||||
purpose with or without fee is hereby granted, provided that the above
|
||||
copyright notice and this permission notice appear in all copies.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
//==============================================================================
|
||||
|
||||
#include "util/async/AnyStopToken.hpp"
|
||||
|
||||
#include <boost/asio/spawn.hpp>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
using namespace util::async;
|
||||
using namespace ::testing;
|
||||
|
||||
namespace {
|
||||
struct FakeStopToken {
|
||||
bool isStopRequested_ = false;
|
||||
bool
|
||||
isStopRequested() const
|
||||
{
|
||||
return isStopRequested_;
|
||||
}
|
||||
};
|
||||
} // namespace
|
||||
|
||||
struct AnyStopTokenTests : public TestWithParam<bool> {};
|
||||
using AnyStopTokenDeathTest = AnyStopTokenTests;
|
||||
|
||||
INSTANTIATE_TEST_CASE_P(AnyStopTokenGroup, AnyStopTokenTests, ValuesIn({true, false}), [](auto const& info) {
|
||||
return info.param ? "true" : "false";
|
||||
});
|
||||
|
||||
TEST_P(AnyStopTokenTests, CanCopy)
|
||||
{
|
||||
AnyStopToken const stopToken{FakeStopToken{GetParam()}};
|
||||
AnyStopToken const token = stopToken;
|
||||
|
||||
EXPECT_EQ(token, stopToken);
|
||||
}
|
||||
|
||||
TEST_P(AnyStopTokenTests, IsStopRequestedCallPropagated)
|
||||
{
|
||||
auto const flag = GetParam();
|
||||
AnyStopToken const stopToken{FakeStopToken{flag}};
|
||||
|
||||
EXPECT_EQ(stopToken.isStopRequested(), flag);
|
||||
EXPECT_EQ(stopToken, flag);
|
||||
}
|
||||
|
||||
TEST_F(AnyStopTokenDeathTest, ConversionToYieldContextAssertsIfUnsupported)
|
||||
{
|
||||
EXPECT_DEATH(
|
||||
[[maybe_unused]] auto unused = static_cast<boost::asio::yield_context>(AnyStopToken{FakeStopToken{}}), ".*"
|
||||
);
|
||||
}
|
||||
128
tests/unit/util/async/AnyStrandTests.cpp
Normal file
128
tests/unit/util/async/AnyStrandTests.cpp
Normal file
@@ -0,0 +1,128 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
This file is part of clio: https://github.com/XRPLF/clio
|
||||
Copyright (c) 2024, the clio developers.
|
||||
|
||||
Permission to use, copy, modify, and distribute this software for any
|
||||
purpose with or without fee is hereby granted, provided that the above
|
||||
copyright notice and this permission notice appear in all copies.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
//==============================================================================
|
||||
|
||||
#include "util/MockOperation.hpp"
|
||||
#include "util/MockStrand.hpp"
|
||||
#include "util/async/AnyOperation.hpp"
|
||||
#include "util/async/AnyStopToken.hpp"
|
||||
#include "util/async/AnyStrand.hpp"
|
||||
|
||||
#include <gmock/gmock.h>
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <any>
|
||||
#include <chrono>
|
||||
#include <expected>
|
||||
#include <functional>
|
||||
#include <type_traits>
|
||||
|
||||
using namespace util::async;
|
||||
using namespace ::testing;
|
||||
|
||||
struct AnyStrandTests : ::testing::Test {
|
||||
template <typename T>
|
||||
using OperationType = ::testing::NiceMock<MockOperation<T>>;
|
||||
|
||||
template <typename T>
|
||||
using StoppableOperationType = ::testing::NiceMock<MockStoppableOperation<T>>;
|
||||
|
||||
::testing::NaggyMock<MockStrand> mockStrand;
|
||||
AnyStrand strand{static_cast<MockStrand&>(mockStrand)};
|
||||
};
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithoutTokenAndVoid)
|
||||
{
|
||||
auto mockOp = OperationType<std::any>{};
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any()>>())).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto op = strand.execute([] {});
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<void>>);
|
||||
|
||||
ASSERT_TRUE(op.get());
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithoutTokenAndVoidThrowsException)
|
||||
{
|
||||
auto mockOp = OperationType<std::any>{};
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any()>>()))
|
||||
.WillOnce([](auto&&) -> OperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([] {}));
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithStopTokenAndVoid)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _)).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto op = strand.execute([](auto) {});
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<void>>);
|
||||
|
||||
ASSERT_TRUE(op.get());
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithStopTokenAndVoidThrowsException)
|
||||
{
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([](auto) {}));
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithStopTokenAndReturnValue)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _)).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto op = strand.execute([](auto) { return 42; });
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<int>>);
|
||||
|
||||
ASSERT_EQ(op.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithStopTokenAndReturnValueThrowsException)
|
||||
{
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW([[maybe_unused]] auto unused = strand.execute([](auto) { return 42; }));
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithTimeoutAndStopTokenAndReturnValue)
|
||||
{
|
||||
auto mockOp = StoppableOperationType<std::any>{};
|
||||
EXPECT_CALL(mockOp, get()).WillOnce(Return(std::make_any<int>(42)));
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _)).WillOnce(ReturnRef(mockOp));
|
||||
|
||||
auto op = strand.execute([](auto) { return 42; }, std::chrono::milliseconds{1});
|
||||
static_assert(std::is_same_v<decltype(op), AnyOperation<int>>);
|
||||
|
||||
ASSERT_EQ(op.get().value(), 42);
|
||||
}
|
||||
|
||||
TEST_F(AnyStrandTests, ExecuteWithTimoutAndStopTokenAndReturnValueThrowsException)
|
||||
{
|
||||
EXPECT_CALL(mockStrand, execute(An<std::function<std::any(AnyStopToken)>>(), _))
|
||||
.WillOnce([](auto&&, auto) -> StoppableOperationType<std::any> const& { throw 0; });
|
||||
|
||||
EXPECT_ANY_THROW(
|
||||
[[maybe_unused]] auto unused = strand.execute([](auto) { return 42; }, std::chrono::milliseconds{1})
|
||||
);
|
||||
}
|
||||
239
tests/unit/util/async/AsyncExecutionContextTests.cpp
Normal file
239
tests/unit/util/async/AsyncExecutionContextTests.cpp
Normal file
@@ -0,0 +1,239 @@
|
||||
//------------------------------------------------------------------------------
|
||||
/*
|
||||
This file is part of clio: https://github.com/XRPLF/clio
|
||||
Copyright (c) 2024, the clio developers.
|
||||
|
||||
Permission to use, copy, modify, and distribute this software for any
|
||||
purpose with or without fee is hereby granted, provided that the above
|
||||
copyright notice and this permission notice appear in all copies.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
*/
|
||||
//==============================================================================
|
||||
|
||||
#include "util/async/context/BasicExecutionContext.hpp"
|
||||
#include "util/async/context/SyncExecutionContext.hpp"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
#include <chrono>
|
||||
#include <semaphore>
|
||||
#include <stdexcept>
|
||||
#include <string>
|
||||
|
||||
using namespace util::async;
|
||||
using ::testing::Types;
|
||||
|
||||
using ExecutionContextTypes = Types<CoroExecutionContext, PoolExecutionContext, SyncExecutionContext>;
|
||||
|
||||
template <typename T>
|
||||
struct ExecutionContextTests : public ::testing::Test {
|
||||
using ExecutionContextType = T;
|
||||
ExecutionContextType ctx{2};
|
||||
};
|
||||
|
||||
TYPED_TEST_CASE(ExecutionContextTests, ExecutionContextTypes);
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, execute)
|
||||
{
|
||||
auto res = this->ctx.execute([]() { return 42; });
|
||||
EXPECT_EQ(res.get().value(), 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, executeVoid)
|
||||
{
|
||||
auto value = 0;
|
||||
auto res = this->ctx.execute([&value]() { value = 42; });
|
||||
|
||||
res.wait();
|
||||
ASSERT_EQ(value, 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, executeStdException)
|
||||
{
|
||||
auto res = this->ctx.execute([]() { throw std::runtime_error("test"); });
|
||||
|
||||
auto const err = res.get().error();
|
||||
EXPECT_TRUE(err.message.ends_with("test"));
|
||||
EXPECT_TRUE(std::string{err}.ends_with("test"));
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, executeUnknownException)
|
||||
{
|
||||
auto res = this->ctx.execute([]() { throw 0; });
|
||||
|
||||
auto const err = res.get().error();
|
||||
EXPECT_TRUE(err.message.ends_with("unknown"));
|
||||
EXPECT_TRUE(std::string{err}.ends_with("unknown"));
|
||||
}
|
||||
|
||||
// note: this fails on pool context with 1 thread
|
||||
TYPED_TEST(ExecutionContextTests, executeWithTimeout)
|
||||
{
|
||||
auto res = this->ctx.execute(
|
||||
[](auto stopRequested) {
|
||||
while (not stopRequested)
|
||||
;
|
||||
return 42;
|
||||
},
|
||||
std::chrono::milliseconds{1}
|
||||
);
|
||||
|
||||
EXPECT_EQ(res.get().value(), 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, timer)
|
||||
{
|
||||
auto res =
|
||||
this->ctx.scheduleAfter(std::chrono::milliseconds(1), []([[maybe_unused]] auto stopRequested, auto cancelled) {
|
||||
if (not cancelled)
|
||||
return 42;
|
||||
return 0;
|
||||
});
|
||||
|
||||
EXPECT_EQ(res.get().value(), 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, timerWithStopToken)
|
||||
{
|
||||
auto res = this->ctx.scheduleAfter(std::chrono::milliseconds(1), [](auto stopRequested) {
|
||||
while (not stopRequested)
|
||||
;
|
||||
|
||||
return 42;
|
||||
});
|
||||
|
||||
res.requestStop();
|
||||
EXPECT_EQ(res.get().value(), 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, timerCancel)
|
||||
{
|
||||
auto value = 0;
|
||||
std::binary_semaphore sem{0};
|
||||
|
||||
auto res = this->ctx.scheduleAfter(
|
||||
std::chrono::milliseconds(10),
|
||||
[&value, &sem]([[maybe_unused]] auto stopRequested, auto cancelled) {
|
||||
if (cancelled)
|
||||
value = 42;
|
||||
|
||||
sem.release();
|
||||
}
|
||||
);
|
||||
|
||||
res.cancel();
|
||||
sem.acquire();
|
||||
EXPECT_EQ(value, 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, timerStdException)
|
||||
{
|
||||
auto res =
|
||||
this->ctx.scheduleAfter(std::chrono::milliseconds(1), []([[maybe_unused]] auto stopRequested, auto cancelled) {
|
||||
if (not cancelled)
|
||||
throw std::runtime_error("test");
|
||||
return 0;
|
||||
});
|
||||
|
||||
auto const err = res.get().error();
|
||||
EXPECT_TRUE(err.message.ends_with("test"));
|
||||
EXPECT_TRUE(std::string{err}.ends_with("test"));
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, timerUnknownException)
|
||||
{
|
||||
auto res =
|
||||
this->ctx.scheduleAfter(std::chrono::milliseconds(1), []([[maybe_unused]] auto stopRequested, auto cancelled) {
|
||||
if (not cancelled)
|
||||
throw 0;
|
||||
return 0;
|
||||
});
|
||||
|
||||
auto const err = res.get().error();
|
||||
EXPECT_TRUE(err.message.ends_with("unknown"));
|
||||
EXPECT_TRUE(std::string{err}.ends_with("unknown"));
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, strand)
|
||||
{
|
||||
auto strand = this->ctx.makeStrand();
|
||||
auto res = strand.execute([] { return 42; });
|
||||
|
||||
EXPECT_EQ(res.get().value(), 42);
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, strandStdException)
|
||||
{
|
||||
auto strand = this->ctx.makeStrand();
|
||||
auto res = strand.execute([]() { throw std::runtime_error("test"); });
|
||||
|
||||
auto const err = res.get().error();
|
||||
EXPECT_TRUE(err.message.ends_with("test"));
|
||||
EXPECT_TRUE(std::string{err}.ends_with("test"));
|
||||
}
|
||||
|
||||
TYPED_TEST(ExecutionContextTests, strandUnknownException)
|
||||
{
|
||||
auto strand = this->ctx.makeStrand();
|
||||
auto res = strand.execute([]() { throw 0; });
|
||||
|
||||
auto const err = res.get().error();
|
||||
EXPECT_TRUE(err.message.ends_with("unknown"));
|
||||
EXPECT_TRUE(std::string{err}.ends_with("unknown"));
|
||||
}
|
||||
|
||||
// note: this fails on pool context with 1 thread
|
||||
TYPED_TEST(ExecutionContextTests, strandWithTimeout)
|
||||
{
|
||||
auto strand = this->ctx.makeStrand();
|
||||
auto res = strand.execute(
|
||||
[](auto stopRequested) {
|
||||
while (not stopRequested)
|
||||
;
|
||||
return 42;
|
||||
},
|
||||
std::chrono::milliseconds{1}
|
||||
);
|
||||
|
||||
EXPECT_EQ(res.get().value(), 42);
|
||||
}
|
||||
|
||||
using NoErrorHandlerSyncExecutionContext = BasicExecutionContext<
|
||||
impl::SameThreadContext,
|
||||
impl::BasicStopSource,
|
||||
impl::SyncDispatchStrategy,
|
||||
impl::SelfContextProvider,
|
||||
impl::NoErrorHandler>;
|
||||
|
||||
TEST(NoErrorHandlerSyncExecutionContextTests, executeStdException)
|
||||
{
|
||||
auto ctx = NoErrorHandlerSyncExecutionContext{};
|
||||
EXPECT_THROW(ctx.execute([] { throw std::runtime_error("test"); }).wait(), std::runtime_error);
|
||||
}
|
||||
|
||||
TEST(NoErrorHandlerSyncExecutionContextTests, executeUnknownException)
|
||||
{
|
||||
auto ctx = NoErrorHandlerSyncExecutionContext{};
|
||||
EXPECT_ANY_THROW(ctx.execute([] { throw 0; }).wait());
|
||||
}
|
||||
|
||||
TEST(NoErrorHandlerSyncExecutionContextTests, executeStdExceptionInStrand)
|
||||
{
|
||||
auto ctx = NoErrorHandlerSyncExecutionContext{};
|
||||
auto strand = ctx.makeStrand();
|
||||
EXPECT_THROW(strand.execute([] { throw std::runtime_error("test"); }).wait(), std::runtime_error);
|
||||
}
|
||||
|
||||
TEST(NoErrorHandlerSyncExecutionContextTests, executeUnknownExceptionInStrand)
|
||||
{
|
||||
auto ctx = NoErrorHandlerSyncExecutionContext{};
|
||||
auto strand = ctx.makeStrand();
|
||||
EXPECT_ANY_THROW(strand.execute([] { throw 0; }).wait());
|
||||
}
|
||||
Reference in New Issue
Block a user