#if !defined(BOOST_ALL_NO_LIB)
#define BOOST_ALL_NO_LIB 1
#endif
#include <boost/asio/experimental/coro.hpp>
#include <boost/asio/detached.hpp>
#include <boost/asio/io_context.hpp>
#include <iostream>
#include "../../unit_test.hpp"
using namespace boost::asio::experimental;
namespace coro {
boost::asio::experimental::coro<int()>
stack_generator(boost::asio::any_io_executor, int i = 1)
{
for (;;)
{
co_yield i;
i *= 2;
}
}
boost::asio::experimental::coro<int(int)>
stack_accumulate(boost::asio::any_io_executor exec)
{
auto gen = stack_generator(exec);
int offset = 0;
while (auto next = co_await gen) offset = co_yield *next + offset; }
boost::asio::experimental::coro<int>
main_stack_coro(boost::asio::io_context&, bool & done)
{
auto g = stack_accumulate(co_await boost::asio::this_coro::executor);
BOOST_ASIO_CHECK(g.is_open());
BOOST_ASIO_CHECK(1 == (co_await g(1000)).value_or(-1));
BOOST_ASIO_CHECK(2002 == (co_await g(2000)).value_or(-1));
BOOST_ASIO_CHECK(3004 == (co_await g(3000)).value_or(-1));
BOOST_ASIO_CHECK(4008 == (co_await g(4000)).value_or(-1));
BOOST_ASIO_CHECK(5016 == (co_await g(5000)).value_or(-1));
BOOST_ASIO_CHECK(6032 == (co_await g(6000)).value_or(-1));
BOOST_ASIO_CHECK(7064 == (co_await g(7000)).value_or(-1));
BOOST_ASIO_CHECK(8128 == (co_await g(8000)).value_or(-1));
BOOST_ASIO_CHECK(9256 == (co_await g(9000)).value_or(-1));
BOOST_ASIO_CHECK(511 == (co_await g(-1)).value_or(-1));
done = true;
}
void stack_test()
{
bool done = false;
boost::asio::io_context ctx;
auto k = main_stack_coro(ctx, done);
k.async_resume(boost::asio::detached);
ctx.run();
BOOST_ASIO_CHECK(done);
}
}
BOOST_ASIO_TEST_SUITE
(
"coro/stack_test",
BOOST_ASIO_TEST_CASE(::coro::stack_test)
)