#include "migration/MigratiorStatus.hpp"
#include "migration/TestMigrators.hpp"
#include "migration/impl/MigratorsRegister.hpp"
#include "util/MockMigrationBackend.hpp"
#include "util/MockMigrationBackendFixture.hpp"
#include "util/MockPrometheus.hpp"
#include "util/config/ConfigConstraints.hpp"
#include "util/config/ConfigDefinition.hpp"
#include "util/config/ConfigValue.hpp"
#include "util/config/Types.hpp"

#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include <algorithm>
#include <optional>
#include <string>
#include <tuple>

using EmptyMigratorRegister = migration::impl::MigratorsRegister<MockMigrationBackend>;

namespace {
util::config::ClioConfigDefinition gCfg{
    {{"migration.full_scan_threads",
      util::config::ConfigValue{util::config::ConfigType::Integer}.defaultValue(2).withConstraint(
          util::config::gValidateUint32
      )},
     {"migration.full_scan_jobs",
      util::config::ConfigValue{util::config::ConfigType::Integer}.defaultValue(4).withConstraint(
          util::config::gValidateUint32
      )},
     {"migration.cursors_per_job",
      util::config::ConfigValue{util::config::ConfigType::Integer}.defaultValue(100).withConstraint(
          util::config::gValidateUint32
      )}}
};
}  // namespace

struct MigratorRegisterTests : public util::prometheus::WithMockPrometheus,
                               public MockMigrationBackendTest {};

TEST_F(MigratorRegisterTests, EmptyMigratorRegister)
{
    EmptyMigratorRegister migratorRegister(backend_);
    EXPECT_EQ(migratorRegister.getMigratorsStatus().size(), 0);
    EXPECT_EQ(migratorRegister.getMigratorNames().size(), 0);
    EXPECT_EQ(
        migratorRegister.getMigratorStatus("unknown"), migration::MigratorStatus::Status::NotKnown
    );
    EXPECT_NO_THROW(migratorRegister.runMigrator("unknown", gCfg.getObject("migration")));
    EXPECT_EQ(migratorRegister.getMigratorDescription("unknown"), "No Description");
}

using MultipleMigratorRegister = migration::impl::MigratorsRegister<
    MockMigrationBackend,
    SimpleTestMigrator,
    SimpleTestMigrator2,
    SimpleTestMigrator3>;

struct MultipleMigratorRegisterTests : public util::prometheus::WithMockPrometheus,
                                       public MockMigrationBackendTest {
    std::optional<MultipleMigratorRegister> migratorRegister;

    MultipleMigratorRegisterTests()
    {
        migratorRegister.emplace(backend_);
    }
};

TEST_F(MultipleMigratorRegisterTests, GetMigratorsStatusWhenError)
{
    EXPECT_CALL(*backend_, fetchMigratorStatus(testing::_, testing::_))
        .Times(3)
        .WillRepeatedly(testing::Return(std::nullopt));

    ASSERT_TRUE(migratorRegister.has_value());
    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    auto const status = migratorRegister->getMigratorsStatus();
    EXPECT_EQ(status.size(), 3);
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator2", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator3", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
}

TEST_F(MultipleMigratorRegisterTests, GetMigratorsStatusWhenReturnInvalidStatus)
{
    EXPECT_CALL(*backend_, fetchMigratorStatus(testing::_, testing::_))
        .Times(3)
        .WillRepeatedly(testing::Return("Invalid"));

    ASSERT_TRUE(migratorRegister.has_value());
    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    auto const status = migratorRegister->getMigratorsStatus();
    EXPECT_EQ(status.size(), 3);
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator2", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator3", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
}

TEST_F(MultipleMigratorRegisterTests, GetMigratorsStatusWhenOneMigrated)
{
    EXPECT_CALL(*backend_, fetchMigratorStatus("SimpleTestMigrator", testing::_))
        .WillOnce(testing::Return("Migrated"));
    EXPECT_CALL(*backend_, fetchMigratorStatus("SimpleTestMigrator2", testing::_))
        .WillOnce(testing::Return("NotMigrated"));
    EXPECT_CALL(*backend_, fetchMigratorStatus("SimpleTestMigrator3", testing::_))
        .WillOnce(testing::Return("NotMigrated"));

    ASSERT_TRUE(migratorRegister.has_value());
    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    auto const status = migratorRegister->getMigratorsStatus();
    EXPECT_EQ(status.size(), 3);
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator", migration::MigratorStatus::Status::Migrated)
        ) != status.end()
    );
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator2", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
    EXPECT_TRUE(
        std::ranges::find(
            status,
            std::make_tuple("SimpleTestMigrator3", migration::MigratorStatus::Status::NotMigrated)
        ) != status.end()
    );
}

TEST_F(MultipleMigratorRegisterTests, GetMigratorStatus)
{
    EXPECT_CALL(*backend_, fetchMigratorStatus("SimpleTestMigrator", testing::_))
        .WillOnce(testing::Return("Migrated"));
    EXPECT_CALL(*backend_, fetchMigratorStatus("SimpleTestMigrator2", testing::_))
        .WillOnce(testing::Return("NotMigrated"));

    ASSERT_TRUE(migratorRegister.has_value());
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorStatus("unknown"),
        migration::MigratorStatus::Status::NotKnown
    );
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorStatus("SimpleTestMigrator"),
        migration::MigratorStatus::Status::Migrated
    );
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorStatus("SimpleTestMigrator2"),
        migration::MigratorStatus::Status::NotMigrated
    );
}

TEST_F(MultipleMigratorRegisterTests, GetMigratorStatusWhenError)
{
    EXPECT_CALL(*backend_, fetchMigratorStatus(testing::_, testing::_))
        .Times(2)
        .WillRepeatedly(testing::Return(std::nullopt));

    ASSERT_TRUE(migratorRegister.has_value());
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorStatus("unknown"),
        migration::MigratorStatus::Status::NotKnown
    );
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorStatus("SimpleTestMigrator"),
        migration::MigratorStatus::Status::NotMigrated
    );
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorStatus("SimpleTestMigrator2"),
        migration::MigratorStatus::Status::NotMigrated
    );
}

TEST_F(MultipleMigratorRegisterTests, Names)
{
    ASSERT_TRUE(migratorRegister.has_value());
    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    auto names = migratorRegister->getMigratorNames();
    EXPECT_EQ(names.size(), 3);
    EXPECT_TRUE(std::ranges::find(names, "SimpleTestMigrator") != names.end());
    EXPECT_TRUE(std::ranges::find(names, "SimpleTestMigrator2") != names.end());
    EXPECT_TRUE(std::ranges::find(names, "SimpleTestMigrator3") != names.end());
}

TEST_F(MultipleMigratorRegisterTests, Description)
{
    ASSERT_TRUE(migratorRegister.has_value());
    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    EXPECT_EQ(migratorRegister->getMigratorDescription("unknown"), "No Description");
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorDescription("SimpleTestMigrator"),
        "The migrator for version 0 -> 1"
    );
    EXPECT_EQ(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->getMigratorDescription("SimpleTestMigrator2"),
        "The migrator for version 1 -> 2"
    );
}

TEST_F(MultipleMigratorRegisterTests, RunUnknownMigrator)
{
    EXPECT_CALL(*backend_, writeMigratorStatus(testing::_, testing::_)).Times(0);
    EXPECT_NO_THROW(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->runMigrator("unknown", gCfg.getObject("migration"))
    );
}

TEST_F(MultipleMigratorRegisterTests, MigrateNormalMigrator)
{
    EXPECT_CALL(*backend_, writeMigratorStatus("SimpleTestMigrator", "Migrated")).Times(1);
    EXPECT_NO_THROW(
        // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
        migratorRegister->runMigrator("SimpleTestMigrator", gCfg.getObject("migration"))
    );
}

TEST_F(MultipleMigratorRegisterTests, canBlock)
{
    ASSERT_TRUE(migratorRegister.has_value());
    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    auto canBlock = migratorRegister->canMigratorBlockClio("SimpleTestMigrator");
    ASSERT_TRUE(canBlock.has_value());
    EXPECT_TRUE(*canBlock);  // NOLINT(bugprone-unchecked-optional-access)

    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    canBlock = migratorRegister->canMigratorBlockClio("SimpleTestMigrator2");
    ASSERT_TRUE(canBlock.has_value());
    EXPECT_FALSE(*canBlock);  // NOLINT(bugprone-unchecked-optional-access)

    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    canBlock = migratorRegister->canMigratorBlockClio("SimpleTestMigrator3");
    ASSERT_TRUE(canBlock.has_value());
    EXPECT_FALSE(*canBlock);  // NOLINT(bugprone-unchecked-optional-access)

    // NOLINTNEXTLINE(bugprone-unchecked-optional-access)
    canBlock = migratorRegister->canMigratorBlockClio("NotAMigrator");
    EXPECT_FALSE(canBlock.has_value());
}