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

import os

import boto3
import pytest

from moto import mock_es


@pytest.fixture
def test_aes():
    with mock_es():
        region = os.environ.get("AWS_REGION")
        domain_name = os.environ.get("DOMAIN_NAME")
        client = boto3.client("es", region_name=region)
        client.create_elasticsearch_domain(
            DomainName=domain_name,
            ElasticsearchVersion="7.10",
            ElasticsearchClusterConfig={
                "InstanceType": "r6g.large.elasticsearch",
                "InstanceCount": 1,
                "DedicatedMasterEnabled": False,
                "ZoneAwarenessEnabled": False,
                "WarmEnabled": False,
                "ColdStorageOptions": {"Enabled": False},
            },
            EBSOptions={
                "EBSEnabled": True,
                "VolumeType": "gp2",
                "VolumeSize": 20,
            },
            VPCOptions={"SubnetIds": ["s1"], "SecurityGroupIds": ["sg1"]},
            CognitoOptions={"Enabled": False},
            EncryptionAtRestOptions={"Enabled": True},
            NodeToNodeEncryptionOptions={"Enabled": True},
            DomainEndpointOptions={
                "EnforceHTTPS": True,
                "CustomEndpointEnabled": False,
            },
            AdvancedSecurityOptions={"Enabled": False},
            AutoTuneOptions={"DesiredState": "ENABLED"},
        )

        yield


@pytest.fixture
def test_aos():
    with mock_es():
        region = os.environ.get("AWS_REGION")
        domain_name = os.environ.get("DOMAIN_NAME")
        client = boto3.client("es", region_name=region)
        client.create_elasticsearch_domain(
            DomainName=domain_name,
            ElasticsearchVersion="OpenSearch_1.2",
            ElasticsearchClusterConfig={
                "InstanceType": "r6g.large.elasticsearch",
                "InstanceCount": 2,
                "DedicatedMasterEnabled": True,
                "DedicatedMasterType": "m6g.large.elasticsearch",
                "DedicatedMasterCount": 3,
                "ZoneAwarenessEnabled": False,
                "WarmEnabled": False,
                "ColdStorageOptions": {"Enabled": False},
            },
            EBSOptions={
                "EBSEnabled": True,
                "VolumeType": "gp2",
                "VolumeSize": 20,
            },
            VPCOptions={"SubnetIds": ["s1"], "SecurityGroupIds": ["sg1"]},
            CognitoOptions={"Enabled": False},
            EncryptionAtRestOptions={"Enabled": True},
            NodeToNodeEncryptionOptions={"Enabled": True},
            DomainEndpointOptions={
                "EnforceHTTPS": True,
                "CustomEndpointEnabled": False,
            },
            AdvancedSecurityOptions={"Enabled": True},
            AutoTuneOptions={"DesiredState": "ENABLED"},
        )

        yield


class Response:
    def __init__(self, status_code, text=""):
        self.status_code = status_code
        self.text = text


@pytest.mark.parametrize(
    "engine,client,expected",
    [
        ("OpenSearch", test_aos, "OK"),
        ("Elasticsearch", test_aes, "OK"),
    ],
)
def test_lambda_handler(mocker, engine, client, expected):
    # Can only import here, as the environment variables need to be set first.
    import lambda_function

    mocker.patch(
        "util.osutil.OpenSearch.put_index_template", return_value=Response(201, "OK")
    )
    mocker.patch(
        "util.osutil.OpenSearch.import_saved_objects", return_value=Response(201, "OK")
    )

    mocker.patch(
        "util.osutil.OpenSearch.create_ism_policy", return_value=Response(201, "OK")
    )

    mocker.patch(
        "util.osutil.OpenSearch.create_index", return_value=Response(201, "OK")
    )

    result = lambda_function.lambda_handler(None, None)
    assert result == expected


@pytest.mark.parametrize(
    "status,client,expected",
    [
        (403, test_aos, "OK"),
        (429, test_aos, "OK"),
    ],
)
def test_lambda_handler_with_aos_error(mocker, status, client, expected):
    # Can only import here, as the environment variables need to be set first.
    import lambda_function

    mocker.patch(
        "util.osutil.OpenSearch.put_index_template",
        return_value=Response(status, "resp"),
    )
    mocker.patch(
        "util.osutil.OpenSearch.import_saved_objects", return_value=Response(201, "OK")
    )

    mocker.patch(
        "util.osutil.OpenSearch.create_ism_policy", return_value=Response(201, "OK")
    )

    mocker.patch(
        "util.osutil.OpenSearch.create_index", return_value=Response(201, "OK")
    )

    result = lambda_function.lambda_handler(None, None)
    assert result == expected


def test_lambda_handler2(mocker, test_aos):
    # Can only import here, as the environment variables need to be set first.
    import lambda_function

    mocker.patch(
        "util.osutil.OpenSearch.put_index_template", return_value=Response(201, "OK")
    )
    mocker.patch(
        "util.osutil.OpenSearch.create_index", return_value=Response(201, "OK")
    )

    event = {"hello": "world"}

    assert lambda_function.lambda_handler(event, None) == "OK"


@pytest.fixture
def test_event():
    return {
        "action": "CreateIndexTemplate",
        "props": {
            "log_type": "nignx",
            "createDashboard": "Yes",
            "mappings": {"a": "b"},
        },
    }


def test_lambda_handler2_with_event(mocker, test_aos, test_event):
    # Can only import here, as the environment variables need to be set first.
    import lambda_function

    mocker.patch(
        "util.osutil.OpenSearch.import_saved_objects", return_value=Response(201, "OK")
    )
    mocker.patch(
        "util.osutil.OpenSearch.create_ism_policy", return_value=Response(201, "OK")
    )
    mocker.patch(
        "util.osutil.OpenSearch.create_index", return_value=Response(201, "OK")
    )

    assert lambda_function.lambda_handler(test_event, None) == "OK"