|
| 1 | +// tests/beman/execution/exec-spawn-future.test.cpp -*-C++-*- |
| 2 | +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| 3 | + |
| 4 | +#include <beman/execution/detail/spawn_future.hpp> |
| 5 | +#include <beman/execution/detail/queryable.hpp> |
| 6 | +#include <beman/execution/detail/sender.hpp> |
| 7 | +#include <beman/execution/detail/async_scope_token.hpp> |
| 8 | +#include <test/execution.hpp> |
| 9 | +#include <concepts> |
| 10 | + |
| 11 | +// ---------------------------------------------------------------------------- |
| 12 | + |
| 13 | +namespace |
| 14 | +{ |
| 15 | + struct non_env { ~non_env() = delete; }; |
| 16 | + static_assert(not test_detail::queryable<non_env>); |
| 17 | + |
| 18 | + struct env {}; |
| 19 | + static_assert(test_detail::queryable<env>); |
| 20 | + |
| 21 | + struct non_sender { |
| 22 | + }; |
| 23 | + static_assert(not test_std::sender<non_sender>); |
| 24 | + |
| 25 | + struct sender { |
| 26 | + using sender_concept = test_std::sender_t; |
| 27 | + }; |
| 28 | + static_assert(test_std::sender<sender>); |
| 29 | + |
| 30 | + template <bool Noexcept> |
| 31 | + struct token { |
| 32 | + auto try_associate() -> bool { return {}; } |
| 33 | + auto disassociate() noexcept(Noexcept) -> void {} |
| 34 | + template <test_std::sender Sender> |
| 35 | + auto wrap(Sender&& sender) -> Sender { return std::forward<Sender>(sender); } |
| 36 | + }; |
| 37 | + static_assert(test_std::async_scope_token<token<true>>); |
| 38 | + static_assert(not test_std::async_scope_token<token<false>>); |
| 39 | + |
| 40 | + struct throws { |
| 41 | + throws() = default; |
| 42 | + throws(throws&&) noexcept(false) = default; |
| 43 | + }; |
| 44 | + static_assert(!std::is_nothrow_constructible_v<std::decay_t<throws>, throws>); |
| 45 | + |
| 46 | + template <bool Expect, typename Sender, typename Token, typename Env = ::env> |
| 47 | + auto test_spawn_future_interface(Sender&& sndr, Token&& tok, Env&& e = Env{}) -> void { |
| 48 | + if constexpr (!std::same_as<std::remove_cvref_t<Env>, non_env>){ |
| 49 | + static_assert(Expect == requires{ test_std::spawn_future(std::forward<Sender>(sndr), std::forward<Token>(tok)); }); |
| 50 | + } |
| 51 | + static_assert(Expect == requires{ test_std::spawn_future(std::forward<Sender>(sndr), std::forward<Token>(tok), std::forward<Env>(e)); }); |
| 52 | + } |
| 53 | + |
| 54 | + template <typename Completions> |
| 55 | + struct state_base: test_detail::spawn_future_state_base<Completions> { |
| 56 | + auto complete() noexcept -> void override {} |
| 57 | + }; |
| 58 | + auto test_state_base() { |
| 59 | + static_assert(noexcept(std::declval<test_detail::spawn_future_state_base<test_std::completion_signatures<>>&>().complete())); |
| 60 | + using state0_t = state_base<test_std::completion_signatures<>>; |
| 61 | + [[maybe_unused]] state0_t b0; |
| 62 | + static_assert(std::same_as<std::variant<std::monostate>, state0_t::variant_t>); |
| 63 | + static_assert(std::same_as<state0_t::variant_t, decltype(b0.result)>); |
| 64 | + |
| 65 | + using state1_t = state_base<test_std::completion_signatures<test_std::set_value_t(int)>>; |
| 66 | + [[maybe_unused]] state1_t b1; |
| 67 | + static_assert(std::same_as<std::variant<std::monostate, std::tuple<test_std::set_value_t, int>>, state1_t::variant_t>); |
| 68 | + static_assert(std::same_as<state1_t::variant_t, decltype(b1.result)>); |
| 69 | + |
| 70 | + using state2_t = state_base<test_std::completion_signatures<test_std::set_value_t(throws)>>; |
| 71 | + [[maybe_unused]] state2_t b2; |
| 72 | + static_assert(std::same_as<std::variant<std::monostate, std::tuple<test_std::set_error_t, std::exception_ptr>, std::tuple<test_std::set_value_t, throws>>, typename state2_t::variant_t>); |
| 73 | + static_assert(std::same_as<state2_t::variant_t, decltype(b2.result)>); |
| 74 | + |
| 75 | + using state3_t = state_base<test_std::completion_signatures<test_std::set_value_t(throws), test_std::set_error_t(std::exception_ptr)>>; |
| 76 | + [[maybe_unused]] state3_t b3; |
| 77 | + static_assert(std::same_as<std::variant<std::monostate, std::tuple<test_std::set_error_t, std::exception_ptr>, std::tuple<test_std::set_value_t, throws>>, typename state3_t::variant_t>); |
| 78 | + static_assert(std::same_as<state3_t::variant_t, decltype(b3.result)>); |
| 79 | + |
| 80 | + using state4_t = state_base<test_std::completion_signatures<test_std::set_value_t(throws), test_std::set_value_t(throws const&)>>; |
| 81 | + [[maybe_unused]] state4_t b4; |
| 82 | + static_assert(std::same_as<std::variant<std::monostate, std::tuple<test_std::set_error_t, std::exception_ptr>, std::tuple<test_std::set_value_t, throws>>, typename state4_t::variant_t>); |
| 83 | + static_assert(std::same_as<state4_t::variant_t, decltype(b4.result)>); |
| 84 | + } |
| 85 | +} |
| 86 | + |
| 87 | +TEST(exec_spawn_future) { |
| 88 | + static_assert(std::same_as<test_std::spawn_future_t const, decltype(test_std::spawn_future)>); |
| 89 | + test_spawn_future_interface<true>(sender{}, token<true>{}); |
| 90 | + test_spawn_future_interface<false>(non_sender{}, token<true>{}); |
| 91 | + test_spawn_future_interface<false>(sender{}, token<false>{}); |
| 92 | + test_spawn_future_interface<false>(sender{}, token<true>{}, *new non_env{}); |
| 93 | + |
| 94 | + test_state_base(); |
| 95 | +} |
0 commit comments