diff options
Diffstat (limited to '3rdparty/asio/src/tests/unit/bind_executor.cpp')
-rw-r--r-- | 3rdparty/asio/src/tests/unit/bind_executor.cpp | 168 |
1 files changed, 166 insertions, 2 deletions
diff --git a/3rdparty/asio/src/tests/unit/bind_executor.cpp b/3rdparty/asio/src/tests/unit/bind_executor.cpp index 74f24638a62..59be1cd14f3 100644 --- a/3rdparty/asio/src/tests/unit/bind_executor.cpp +++ b/3rdparty/asio/src/tests/unit/bind_executor.cpp @@ -2,7 +2,7 @@ // bind_executor.cpp // ~~~~~~~~~~~~~~~~~ // -// Copyright (c) 2003-2016 Christopher M. Kohlhoff (chris at kohlhoff dot com) +// Copyright (c) 2003-2024 Christopher M. Kohlhoff (chris at kohlhoff dot com) // // Distributed under the Boost Software License, Version 1.0. (See accompanying // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) @@ -16,10 +16,174 @@ // Test that header file is self-contained. #include "asio/bind_executor.hpp" +#include <functional> +#include "asio/io_context.hpp" +#include "asio/steady_timer.hpp" #include "unit_test.hpp" +#if defined(ASIO_HAS_BOOST_DATE_TIME) +# include "asio/deadline_timer.hpp" +#else // defined(ASIO_HAS_BOOST_DATE_TIME) +# include "asio/steady_timer.hpp" +#endif // defined(ASIO_HAS_BOOST_DATE_TIME) + +using namespace asio; +namespace bindns = std; + +#if defined(ASIO_HAS_BOOST_DATE_TIME) +typedef deadline_timer timer; +namespace chronons = boost::posix_time; +#else // defined(ASIO_HAS_BOOST_DATE_TIME) +typedef steady_timer timer; +namespace chronons = asio::chrono; +#endif // defined(ASIO_HAS_BOOST_DATE_TIME) + +void increment(int* count) +{ + ++(*count); +} + +void bind_executor_to_function_object_test() +{ + io_context ioc1; + io_context ioc2; + + int count = 0; + + timer t(ioc1, chronons::seconds(1)); + t.async_wait( + bind_executor( + ioc2.get_executor(), + bindns::bind(&increment, &count))); + + ioc1.run(); + + ASIO_CHECK(count == 0); + + ioc2.run(); + + ASIO_CHECK(count == 1); + + t.async_wait( + bind_executor( + ioc2.get_executor(), + bind_executor( + asio::system_executor(), + bindns::bind(&increment, &count)))); + + ioc1.restart(); + ioc1.run(); + + ASIO_CHECK(count == 1); + + ioc2.restart(); + ioc2.run(); + + ASIO_CHECK(count == 2); +} + +struct incrementer_token_v1 +{ + explicit incrementer_token_v1(int* c) : count(c) {} + int* count; +}; + +struct incrementer_handler_v1 +{ + explicit incrementer_handler_v1(incrementer_token_v1 t) : count(t.count) {} + void operator()(asio::error_code){ increment(count); } + int* count; +}; + +namespace asio { + +template <> +class async_result<incrementer_token_v1, void(asio::error_code)> +{ +public: + typedef incrementer_handler_v1 completion_handler_type; + typedef void return_type; + explicit async_result(completion_handler_type&) {} + return_type get() {} +}; + +} // namespace asio + +void bind_executor_to_completion_token_v1_test() +{ + io_context ioc1; + io_context ioc2; + + int count = 0; + + timer t(ioc1, chronons::seconds(1)); + t.async_wait( + bind_executor( + ioc2.get_executor(), + incrementer_token_v1(&count))); + + ioc1.run(); + + ASIO_CHECK(count == 0); + + ioc2.run(); + + ASIO_CHECK(count == 1); +} + +struct incrementer_token_v2 +{ + explicit incrementer_token_v2(int* c) : count(c) {} + int* count; +}; + +namespace asio { + +template <> +class async_result<incrementer_token_v2, void(asio::error_code)> +{ +public: +#if !defined(ASIO_HAS_RETURN_TYPE_DEDUCTION) + typedef void return_type; +#endif // !defined(ASIO_HAS_RETURN_TYPE_DEDUCTION) + + template <typename Initiation, typename... Args> + static void initiate(Initiation initiation, + incrementer_token_v2 token, Args&&... args) + { + initiation(bindns::bind(&increment, token.count), + static_cast<Args&&>(args)...); + } +}; + +} // namespace asio + +void bind_executor_to_completion_token_v2_test() +{ + io_context ioc1; + io_context ioc2; + + int count = 0; + + timer t(ioc1, chronons::seconds(1)); + t.async_wait( + bind_executor( + ioc2.get_executor(), + incrementer_token_v2(&count))); + + ioc1.run(); + + ASIO_CHECK(count == 0); + + ioc2.run(); + + ASIO_CHECK(count == 1); +} + ASIO_TEST_SUITE ( "bind_executor", - ASIO_TEST_CASE(null_test) + ASIO_TEST_CASE(bind_executor_to_function_object_test) + ASIO_TEST_CASE(bind_executor_to_completion_token_v1_test) + ASIO_TEST_CASE(bind_executor_to_completion_token_v2_test) ) |