forked from boostorg/beast
Vinnie Falco
6baa607295
This adjusts Beast's interfaces and implementation to match the changes in Boost.Asio.
118 lines
3.0 KiB
C++
118 lines
3.0 KiB
C++
//
|
|
// Copyright (c) 2019 Vinnie Falco (vinnie dot falco at gmail 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)
|
|
//
|
|
// Official repository: https://github.com/boostorg/beast
|
|
//
|
|
|
|
#ifndef BOOST_BEAST_TEST_TCP_HPP
|
|
#define BOOST_BEAST_TEST_TCP_HPP
|
|
|
|
#include <boost/beast/core/detail/config.hpp>
|
|
#include <boost/beast/core/detail/get_io_context.hpp>
|
|
#include <boost/beast/_experimental/unit_test/suite.hpp>
|
|
#include <boost/beast/_experimental/test/handler.hpp>
|
|
#include <boost/asio/ip/tcp.hpp>
|
|
#include <chrono>
|
|
|
|
namespace boost {
|
|
namespace beast {
|
|
namespace test {
|
|
|
|
/** Run an I/O context.
|
|
|
|
This function runs and dispatches handlers on the specified
|
|
I/O context, until one of the following conditions is true:
|
|
|
|
@li The I/O context runs out of work.
|
|
|
|
@param ioc The I/O context to run
|
|
*/
|
|
inline
|
|
void
|
|
run(net::io_context& ioc)
|
|
{
|
|
ioc.run();
|
|
ioc.restart();
|
|
}
|
|
|
|
/** Run an I/O context for a certain amount of time.
|
|
|
|
This function runs and dispatches handlers on the specified
|
|
I/O context, until one of the following conditions is true:
|
|
|
|
@li The I/O context runs out of work.
|
|
|
|
@li No completions occur and the specified amount of time has elapsed.
|
|
|
|
@param ioc The I/O context to run
|
|
|
|
@param elapsed The maximum amount of time to run for.
|
|
*/
|
|
template<class Rep, class Period>
|
|
void
|
|
run_for(
|
|
net::io_context& ioc,
|
|
std::chrono::duration<Rep, Period> elapsed)
|
|
{
|
|
ioc.run_for(elapsed);
|
|
ioc.restart();
|
|
}
|
|
|
|
/** Connect two TCP sockets together.
|
|
*/
|
|
template<class Executor>
|
|
bool
|
|
connect(
|
|
net::basic_stream_socket<net::ip::tcp, Executor>& s1,
|
|
net::basic_stream_socket<net::ip::tcp, Executor>& s2)
|
|
|
|
{
|
|
auto ioc1 = beast::detail::get_io_context(s1);
|
|
auto ioc2 = beast::detail::get_io_context(s2);
|
|
if(! BEAST_EXPECT(ioc1 != nullptr))
|
|
return false;
|
|
if(! BEAST_EXPECT(ioc2 != nullptr))
|
|
return false;
|
|
if(! BEAST_EXPECT(ioc1 == ioc2))
|
|
return false;
|
|
auto& ioc = *ioc1;
|
|
try
|
|
{
|
|
net::basic_socket_acceptor<
|
|
net::ip::tcp, Executor> a(s1.get_executor());
|
|
auto ep = net::ip::tcp::endpoint(
|
|
net::ip::make_address_v4("127.0.0.1"), 0);
|
|
a.open(ep.protocol());
|
|
a.set_option(
|
|
net::socket_base::reuse_address(true));
|
|
a.bind(ep);
|
|
a.listen(0);
|
|
ep = a.local_endpoint();
|
|
a.async_accept(s2, test::success_handler());
|
|
s1.async_connect(ep, test::success_handler());
|
|
run(ioc);
|
|
if(! BEAST_EXPECT(
|
|
s1.remote_endpoint() == s2.local_endpoint()))
|
|
return false;
|
|
if(! BEAST_EXPECT(
|
|
s2.remote_endpoint() == s1.local_endpoint()))
|
|
return false;
|
|
}
|
|
catch(std::exception const& e)
|
|
{
|
|
beast::unit_test::suite::this_suite()->fail(
|
|
e.what(), __FILE__, __LINE__);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
} // test
|
|
} // beast
|
|
} // boost
|
|
|
|
#endif
|