import os
from unittest import mock

from datahub.cli import cli_utils
from datahub.cli.config_utils import _get_config_from_env


def test_first_non_null():
    assert cli_utils.first_non_null([]) is None
    assert cli_utils.first_non_null([None]) is None
    assert cli_utils.first_non_null([None, "1"]) == "1"
    assert cli_utils.first_non_null([None, "1", "2"]) == "1"
    assert cli_utils.first_non_null(["3", "1", "2"]) == "3"
    assert cli_utils.first_non_null(["", "1", "2"]) == "1"
    assert cli_utils.first_non_null([" ", "1", "2"]) == "1"


@mock.patch.dict(os.environ, {"DATAHUB_GMS_HOST": "http://localhost:9092"})
def test_correct_url_when_gms_host_in_old_format():
    assert _get_config_from_env() == ("http://localhost:9092", None)


@mock.patch.dict(
    os.environ, {"DATAHUB_GMS_HOST": "localhost", "DATAHUB_GMS_PORT": "8080"}
)
def test_correct_url_when_gms_host_and_port_set():
    assert _get_config_from_env() == ("http://localhost:8080", None)


@mock.patch.dict(
    os.environ,
    {
        "DATAHUB_GMS_URL": "https://example.com",
        "DATAHUB_GMS_HOST": "localhost",
        "DATAHUB_GMS_PORT": "8080",
    },
)
def test_correct_url_when_gms_host_port_url_set():
    assert _get_config_from_env() == ("http://localhost:8080", None)


@mock.patch.dict(
    os.environ,
    {
        "DATAHUB_GMS_URL": "https://example.com",
        "DATAHUB_GMS_HOST": "localhost",
        "DATAHUB_GMS_PORT": "8080",
        "DATAHUB_GMS_PROTOCOL": "https",
    },
)
def test_correct_url_when_gms_host_port_url_protocol_set():
    assert _get_config_from_env() == ("https://localhost:8080", None)


@mock.patch.dict(
    os.environ,
    {
        "DATAHUB_GMS_URL": "https://example.com",
    },
)
def test_correct_url_when_url_set():
    assert _get_config_from_env() == ("https://example.com", None)


def test_fixup_gms_url():
    assert cli_utils.fixup_gms_url("http://localhost:8080") == "http://localhost:8080"
    assert cli_utils.fixup_gms_url("http://localhost:8080/") == "http://localhost:8080"
    assert cli_utils.fixup_gms_url("http://abc.acryl.io") == "https://abc.acryl.io/gms"
    assert (
        cli_utils.fixup_gms_url("http://abc.acryl.io/api/gms")
        == "https://abc.acryl.io/gms"
    )
    assert (
        cli_utils.fixup_gms_url("http://abcd.acryl.io:8080")
        == "https://abcd.acryl.io/gms"
    )


def test_guess_frontend_url_from_gms_url():
    assert (
        cli_utils.guess_frontend_url_from_gms_url("http://localhost:8080")
        == "http://localhost:9002"
    )
    assert (
        cli_utils.guess_frontend_url_from_gms_url("http://localhost:8080/")
        == "http://localhost:9002"
    )
    assert (
        cli_utils.guess_frontend_url_from_gms_url("https://abc.acryl.io/gms")
        == "https://abc.acryl.io"
    )