#pragma once

#include "util/LoggerFixtures.hpp"
#include "util/MockAmendmentBlockHandler.hpp"
#include "util/MockETLService.hpp"
#include "util/MockLedgerFetcher.hpp"
#include "util/MockLoadBalancer.hpp"

#include <gmock/gmock.h>

#include <memory>

/**
 * @brief Fixture with a mock etl service
 */
template <template <typename> typename MockType = ::testing::NiceMock>
struct MockETLServiceTestBase : virtual public ::testing::Test {
    using Mock = MockType<MockETLService>;

protected:
    std::shared_ptr<Mock> mockETLServicePtr_ = std::make_shared<Mock>();
};

/**
 * @brief Fixture with a "nice" ETLService mock.
 *
 * Use @see MockETLServiceTestNaggy during development to get unset call expectation warnings from
 * the embedded mock. Once the test is ready and you are happy you can switch to this fixture to
 * mute the warnings.
 */
using MockETLServiceTest = MockETLServiceTestBase<::testing::NiceMock>;

/**
 * @brief Fixture with a "naggy" ETLService mock.
 *
 * Use this during development to get unset call expectation warnings from the embedded mock.
 */
using MockETLServiceTestNaggy = MockETLServiceTestBase<::testing::NaggyMock>;

/**
 * @brief Fixture with a "strict" ETLService mock.
 */
using MockETLServiceTestStrict = MockETLServiceTestBase<::testing::StrictMock>;

/**
 * @brief Fixture with a mock etl balancer
 */
struct MockLoadBalancerTest : virtual public ::testing::Test {
protected:
    std::shared_ptr<MockLoadBalancer> mockLoadBalancerPtr_ = std::make_shared<MockLoadBalancer>();
};

/**
 * @brief Fixture with a mock ledger fetcher
 */
struct MockLedgerFetcherTest : virtual public ::testing::Test {
protected:
    std::shared_ptr<MockLedgerFetcher> mockLedgerFetcherPtr_ =
        std::make_shared<MockLedgerFetcher>();
};

/**
 * @brief Fixture with a mock ledger fetcher
 */
struct MockAmendmentBlockHandlerTest : virtual public ::testing::Test {
protected:
    std::shared_ptr<MockAmendmentBlockHandler> mockAmendmentBlockHandlerPtr_ =
        std::make_shared<MockAmendmentBlockHandler>();
};