//
// 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)
//

#include "test_common/test_service.hpp"

#include <boost/asio/io_context.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/mqtt5.hpp>
#include <boost/test/unit_test.hpp>

#include <chrono>
#include <cstdint>
#include <string>

using namespace boost::mqtt5;

BOOST_AUTO_TEST_SUITE(session/*, *boost::unit_test::disabled()*/)

BOOST_AUTO_TEST_CASE(session_state_session_present) {
    detail::session_state session_state;

    BOOST_TEST(session_state.session_present() == false);
    session_state.session_present(true);
    BOOST_TEST(session_state.session_present() == true);
    session_state.session_present(false);
    BOOST_TEST(session_state.session_present() == false);

    BOOST_TEST(session_state.subscriptions_present() == false);
    session_state.subscriptions_present(true);
    BOOST_TEST(session_state.subscriptions_present() == true);
    session_state.subscriptions_present(false);
    BOOST_TEST(session_state.subscriptions_present() == false);
}

BOOST_AUTO_TEST_CASE(clear_waiting_on_pubrel) {
    asio::io_context ioc;
    using client_service_type = test::test_service<asio::ip::tcp::socket>;
    auto svc_ptr = std::make_shared<client_service_type>(ioc.get_executor());
    svc_ptr->open_stream();

    decoders::publish_message pub_msg = std::make_tuple(
        "topic", uint16_t(1), uint8_t(0b0100), publish_props {}, "payload"
    );

    detail::publish_rec_op<client_service_type> { svc_ptr }.perform(pub_msg);

    // let publish_rec_op reach wait_on_pubrel stage
    test::test_timer timer(ioc.get_executor());
    timer.expires_after(std::chrono::milliseconds(1));
    timer.async_wait([&svc_ptr](error_code) {
        BOOST_TEST(svc_ptr.use_count() == 2);
        svc_ptr->update_session_state(); // session_present = false
        // publish_rec_op should complete
        BOOST_TEST(svc_ptr.use_count() == 1);
    });

    test::test_broker::run(ioc);
}


BOOST_AUTO_TEST_SUITE_END();