import pytest
from click import ClickException

from datahub.cli.quickstart_versioning import (
    QuickstartExecutionPlan,
    QuickstartVersionMappingConfig,
)

example_version_mapper = QuickstartVersionMappingConfig.parse_obj(
    {
        "quickstart_version_map": {
            "default": {
                "composefile_git_ref": "master",
                "docker_tag": "head",
                "mysql_tag": "8.2",
            },
            "v0.9.6": {
                "composefile_git_ref": "v0.9.6.1",
                "docker_tag": "v0.9.6.1",
                "mysql_tag": "5.7",
            },
            "v2.0.0": {
                "composefile_git_ref": "v2.0.1",
                "docker_tag": "v2.0.0",
                "mysql_tag": "8.2",
            },
            "v1.2.0": {
                "composefile_git_ref": "v1.2.0",
                "docker_tag": "v1.2.0",
                "mysql_tag": "8.2",
            },
            "stable": {
                "composefile_git_ref": "v1.2.1",
                "docker_tag": "v1.2.1",
                "mysql_tag": "8.2",
            },
        },
    }
)


def test_quickstart_version_config():
    execution_plan = example_version_mapper.get_quickstart_execution_plan("v1.2.0")
    expected = QuickstartExecutionPlan(
        docker_tag="v1.2.0",
        composefile_git_ref="v1.2.0",
        mysql_tag="8.2",
    )
    assert execution_plan == expected


def test_quickstart_version_config_default():
    execution_plan = example_version_mapper.get_quickstart_execution_plan("v2.0.0")
    expected = QuickstartExecutionPlan(
        docker_tag="v2.0.0",
        composefile_git_ref="v2.0.1",
        mysql_tag="8.2",
    )
    assert execution_plan == expected


def test_quickstart_version_config_stable():
    execution_plan = example_version_mapper.get_quickstart_execution_plan("stable")
    expected = QuickstartExecutionPlan(
        docker_tag="v1.2.1", composefile_git_ref="v1.2.1", mysql_tag="8.2"
    )
    assert execution_plan == expected


def test_quickstart_forced_stable():
    example_version_mapper.quickstart_version_map["default"] = QuickstartExecutionPlan(
        composefile_git_ref="v1.2.0",
        docker_tag="head",
        mysql_tag="8.2",
    )
    execution_plan = example_version_mapper.get_quickstart_execution_plan(None)
    expected = QuickstartExecutionPlan(
        docker_tag="head",
        composefile_git_ref="v1.2.0",
        mysql_tag="8.2",
    )
    assert execution_plan == expected


def test_quickstart_forced_not_a_version_tag():
    """
    If the user specifies a version that is not in the version mapping, we should still use the default version for checking.

    This is because the user may be using a version of datahub that is not in the version mapping. Which is most likely
    a recent change, otherwise it should be on an older version of datahub.
    """
    execution_plan = example_version_mapper.get_quickstart_execution_plan(
        "NOT A VERSION"
    )
    expected = QuickstartExecutionPlan(
        docker_tag="NOT A VERSION",
        composefile_git_ref="NOT A VERSION",
        mysql_tag="8.2",
    )
    assert execution_plan == expected


def test_quickstart_get_older_version():
    with pytest.raises(ClickException, match="Minimum supported version not met"):
        example_version_mapper.get_quickstart_execution_plan("v0.9.6")


def test_quickstart_version_older_than_v1_2_0_uses_commit_hash():
    """
    Test that versions older than v1.2.0 get their composefile_git_ref set to the specific commit hash
    that contains the profiles based resolved compose file.
    This exercises line 168 in quickstart_versioning.py.
    """
    # Create a version mapping with a version older than v1.2.0
    version_mapper = QuickstartVersionMappingConfig.parse_obj(
        {
            "quickstart_version_map": {
                "v1.1.0": {
                    "composefile_git_ref": "v1.1.0",
                    "docker_tag": "v1.1.0",
                    "mysql_tag": "8.2",
                },
            },
        }
    )

    execution_plan = version_mapper.get_quickstart_execution_plan("v1.1.0")
    expected = QuickstartExecutionPlan(
        docker_tag="v1.1.0",
        composefile_git_ref="21726bc3341490f4182b904626c793091ac95edd",  # This should be the commit hash from line 168
        mysql_tag="8.2",
    )
    assert execution_plan == expected