2
0
mirror of https://github.com/boostorg/mqtt5.git synced 2026-01-19 04:22:11 +00:00
Files
mqtt5/test/include/test_common/test_service.hpp
Bruno Iljazovic 6198b0e44a Mock timer and resolver in unit tests.
Summary:
* Removed all usage of real timers and resolvers in unit tests
* Moved most of the tests to test/unit folder
* cmake: split boost_mqtt5_tests into boost_mqtt5_unittests and boost_mqtt5_integrationtests

Reviewers: ivica

Reviewed By: ivica

Subscribers: miljen

Differential Revision: https://repo.mireo.local/D38186
2025-11-17 13:10:36 +01:00

102 lines
2.6 KiB
C++

//
// Copyright (c) 2023-2025 Ivica Siladic, Bruno Iljazovic, Korina Simicevic
//
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#ifndef BOOST_MQTT5_TEST_TEST_SERVICE_HPP
#define BOOST_MQTT5_TEST_TEST_SERVICE_HPP
#include "test_broker.hpp"
#include <boost/mqtt5/types.hpp>
#include <boost/mqtt5/impl/client_service.hpp>
#include <boost/asio/any_io_executor.hpp>
#include <boost/asio/async_result.hpp>
#include <boost/asio/post.hpp>
#include <boost/asio/prepend.hpp>
#include <boost/system/error_code.hpp>
#include <cstdint>
#include <variant> // std::monostate
namespace boost::mqtt5::test {
namespace asio = boost::asio;
template <
typename StreamType,
typename TlsContext = std::monostate
>
class test_service : public boost::mqtt5::detail::client_service<StreamType, TlsContext> {
using error_code = boost::system::error_code;
using base = boost::mqtt5::detail::client_service<StreamType, TlsContext>;
asio::any_io_executor _ex;
connack_props _test_props;
public:
explicit test_service(const asio::any_io_executor& ex)
: base(ex), _ex(ex)
{}
test_service(const asio::any_io_executor& ex, connack_props props)
: base(ex), _ex(ex), _test_props(std::move(props))
{}
template <typename BufferType, typename CompletionToken>
decltype(auto) async_send(
const BufferType&, uint32_t, unsigned,
CompletionToken&& token
) {
auto initiation = [this](auto handler) {
asio::post(_ex,
asio::prepend(std::move(handler), error_code {})
);
};
return asio::async_initiate<CompletionToken, void (error_code)>(
std::move(initiation), token
);
}
template <typename Prop>
const auto& connack_property(Prop p) const {
return _test_props[p];
}
size_t max_packet_size() const {
return connack_property(prop::maximum_packet_size)
.value_or(::boost::mqtt5::detail::default_max_send_size);
}
const auto& connack_properties() {
return _test_props;
}
bool was_connected() const { return true; }
};
template <
typename StreamType,
typename TlsContext = std::monostate
>
class overrun_client : public boost::mqtt5::detail::client_service<StreamType, TlsContext> {
public:
explicit overrun_client(const asio::any_io_executor& ex) :
boost::mqtt5::detail::client_service<StreamType, TlsContext>(ex)
{}
uint16_t allocate_pid() {
return 0;
}
};
} // end namespace boost::mqtt5::test
#endif // BOOST_MQTT5_TEST_TEST_SERVICE_HPP