boost_mqtt5/test/include/test_common/test_authenticators.hpp

//
// 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_AUTHENTICATORS_HPP
#define BOOST_MQTT5_TEST_TEST_AUTHENTICATORS_HPP

#include <boost/mqtt5/types.hpp>

#include <boost/asio/async_result.hpp>
#include <boost/asio/dispatch.hpp>
#include <boost/asio/prepend.hpp>
#include <boost/system/error_code.hpp>

#include <string>
#include <string_view>

namespace boost::mqtt5::test {

namespace asio = boost::asio;

struct test_authenticator {
    test_authenticator() = default;

    template <typename CompletionToken>
    decltype(auto) async_auth(
        auth_step_e step, std::string data,
        CompletionToken&& token
    ) {
        using error_code = boost::system::error_code;
        using Signature = void (error_code, std::string);

        auto initiate = [](auto handler, auth_step_e, std::string) {
            asio::dispatch(
                asio::prepend(std::move(handler), error_code {}, "")
            );
        };

        return asio::async_initiate<CompletionToken, Signature>(
            initiate, token, step, std::move(data)
        );
    }

    std::string_view method() const {
        return "method";
    }
};

template <auth_step_e fail_on_step>
struct fail_test_authenticator {
    fail_test_authenticator() = default;

    template <typename CompletionToken>
    decltype(auto) async_auth(
        auth_step_e step, std::string data,
        CompletionToken&& token
    ) {
        using error_code = boost::system::error_code;
        using Signature = void (error_code, std::string);

        auto initiate = [](auto handler, auth_step_e step, std::string) {
            error_code ec;
            if (fail_on_step == step)
                ec = asio::error::no_recovery;

            asio::dispatch(
                asio::prepend(std::move(handler), ec, "")
            );
        };

        return asio::async_initiate<CompletionToken, Signature>(
            initiate, token, step, std::move(data)
        );
    }

    std::string_view method() const {
        return "method";
    }
};

} // end namespace boost::mqtt5::test

#endif // BOOST_MQTT5_TEST_TEST_AUTHENTICATORS_HPP