# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

import os

import botocore
import pytest

import aws_solutions.core


@pytest.fixture(scope="function", autouse=True)
def reset_botocore_config():
    """remove botocore configuration before test"""
    aws_solutions.core.config._botocore_config = None


@pytest.fixture(
    params=[
        "SO9900",
        "SO9900a",
        "SO9900A",
    ]
)
def solution_id_valid(request):
    solution_id = request.param
    os.environ["SOLUTION_ID"] = solution_id
    yield solution_id
    del os.environ["SOLUTION_ID"]


@pytest.fixture(params=["S0100", "abc", "SO0x3ab"])
def solution_id_invalid(request):
    solution_id = request.param
    os.environ["SOLUTION_ID"] = solution_id
    yield solution_id
    del os.environ["SOLUTION_ID"]


@pytest.fixture(
    params=[
        "v0.0.1-alpha",
        "v0.0.1-alpha.1",
        "v0.0.1-alpha.beta",
        "v0.0.1-beta",
        "v0.0.1-beta.2",
        "v0.0.1-beta.11",
        "v0.0.1-rc.1",
        "v0.0.1",
        "v0.0.1-alpha",
        "v0.0.1-alpha.1",
        "v0.0.1-0.3.7",
        "v0.0.1-x.7.z.92",
        "v0.0.1-x-y-z.-",
        "v0.0.1-alpha+001",
        "v0.0.1+20130313144700",
        "v0.0.1-beta+exp.sha.5114f85",
        "v0.0.1+21AF26D3--117B344092BD",
    ]
)
def solution_version_valid(request):
    solution_version = request.param
    os.environ["SOLUTION_VERSION"] = solution_version
    yield solution_version
    del os.environ["SOLUTION_VERSION"]


@pytest.fixture(params=["a.b.c", "a1.2.3", "v.1.2.3"])
def solution_version_invalid(request):
    solution_version = request.param
    os.environ["SOLUTION_VERSION"] = solution_version
    yield solution_version
    del os.environ["SOLUTION_VERSION"]


def test_valid_solution_id(solution_id_valid):
    config_id = aws_solutions.core.config.id
    assert config_id == solution_id_valid


def test_invalid_solution_id(solution_id_invalid):
    with pytest.raises(ValueError):
        aws_solutions.core.config.id


def test_valid_solution_version(solution_version_valid):
    version = aws_solutions.core.config.version
    assert version == solution_version_valid


def test_invalid_solution_id(solution_version_invalid):
    with pytest.raises(ValueError):
        aws_solutions.core.config.version


def test_valid_botocore_config(solution_id_valid, solution_version_valid):
    boto_config = aws_solutions.core.config.botocore_config
    assert boto_config.user_agent_extra == f"AwsSolution/{solution_id_valid}/{solution_version_valid}"


def test_solution_config_env_reuse():
    aws_solutions.core.config.id = "SO9900"

    id_1 = aws_solutions.core.config.id
    id_2 = aws_solutions.core.config.id

    assert id_1 is id_2


def test_botocore_config_change():
    aws_solutions.core.config.id = "SO9900"
    aws_solutions.core.config.version = "v0.0.1"
    aws_solutions.core.config.botocore_config.read_timeout = 123
    assert aws_solutions.core.config.botocore_config.read_timeout == 123


def test_botocore_config_change_defaults():
    # it is probably better to just set the value directly as per above
    aws_solutions.core.config.id = "SO9900"
    aws_solutions.core.config.version = "v0.0.1"
    aws_solutions.core.config.botocore_config
    cfg_2 = botocore.config.Config(read_timeout=123)
    aws_solutions.core.config.botocore_config = cfg_2

    assert aws_solutions.core.config.botocore_config.read_timeout == 123
    assert aws_solutions.core.config.botocore_config.user_agent_extra == f"AwsSolution/SO9900/v0.0.1"