#!/usr/bin/env python
######################################################################################################################
#  Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.                                           #
#                                                                                                                    #
#  Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance    #
#  with the License. A copy of the License is located at                                                             #
#                                                                                                                    #
#      http://www.apache.org/licenses/LICENSE-2.0                                                                     #
#                                                                                                                    #
#  or in the 'license' file accompanying this file. This file is distributed on an 'AS IS' BASIS, WITHOUT WARRANTIES #
#  OR CONDITIONS OF ANY KIND, express or implied. See the License for the specific language governing permissions    #
#  and limitations under the License.                                                                                #
######################################################################################################################

import logging
import test.logger_test_helper
from test.fixtures.quicksight_dashboard_fixtures import DashboardStubber
from test.fixtures.quicksight_dataset_fixtures import (
    data_set_type,
    mininmal_data_sets_stub,
    quicksight_create_data_set_stubber,
    quicksight_data_set_stubber,
    quicksight_delete_data_set_stubber,
)
from test.fixtures.quicksight_template_fixtures import template_arn
from test.fixtures.quicksight_test_fixture import quicksight_application_stub
from test.logger_test_helper import dump_state

import pytest
from moto import mock_sts
from util.quicksight_application import QuicksightApplication
from util.source_entity import SourceEntity

logger = logging.getLogger(__name__)


def test_source_entity_init_invalid(quicksight_application_stub, mininmal_data_sets_stub):
    template_arn_param = template_arn

    obj = None
    with pytest.raises(ValueError):
        # Function under test
        obj = SourceEntity(
            data_sets=mininmal_data_sets_stub.data_sets_stub,
            source_obj_arn=template_arn_param,
            config_data=None,
            source_entity_type="InjectErrorInvalidSource",
        )

    assert not obj


def test_source_entity_get_map_invalid_sub_type(quicksight_application_stub, mininmal_data_sets_stub):
    template_arn_param = template_arn
    mock_data_set_references = {
        "MainInjectError": {
            "SourceEntity": {
                "SourceTemplate": {
                    "DataSetReferences": [
                        {
                            "DataSetPlaceholder": "image-text",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/image-text",
                        },
                        {
                            "DataSetPlaceholder": "image-moderation-label",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/image-moderation-label",
                        },
                        {
                            "DataSetPlaceholder": "topic",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/topic",
                        },
                        {
                            "DataSetPlaceholder": "geo",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/geo",
                        },
                        {
                            "DataSetPlaceholder": "topic-mapping",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/topic-mapping",
                        },
                        {
                            "DataSetPlaceholder": "feed",
                            "DataSetArn": "arn:aws:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/feed",
                        },
                    ],
                    "Arn": "{self.source_template_arn}",
                }
            }
        }
    }
    obj = SourceEntity(
        data_sets=mininmal_data_sets_stub.data_sets_stub,
        source_obj_arn=template_arn_param,
        config_data=mock_data_set_references,
        source_entity_type="SourceTemplate",
    )

    source_entity = None
    with pytest.raises(ValueError):
        # Function under test
        source_entity = obj.get_source_entity()

    assert not source_entity


def test_source_entity_get_map_missing_source_entity(quicksight_application_stub, mininmal_data_sets_stub):
    template_arn_param = template_arn
    mock_data_set_references = {
        "main": {
            "SourceEntityInjectError": {
                "SourceTemplate": {
                    "DataSetReferences": [
                        {
                            "DataSetPlaceholder": "image-text",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/image-text",
                        },
                        {
                            "DataSetPlaceholder": "image-moderation-label",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/image-moderation-label",
                        },
                        {
                            "DataSetPlaceholder": "topic",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/topic",
                        },
                        {
                            "DataSetPlaceholder": "geo",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/geo",
                        },
                        {
                            "DataSetPlaceholder": "topic-mapping",
                            "DataSetArn": "arn:{Aws.PARTITION}:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/topic-mapping",
                        },
                        {
                            "DataSetPlaceholder": "feed",
                            "DataSetArn": "arn:aws:quicksight:{Aws.REGION}:{Aws.ACCOUNT_ID}:dataset/feed",
                        },
                    ],
                    "Arn": "{self.source_template_arn}",
                }
            }
        }
    }
    obj = SourceEntity(
        data_sets=mininmal_data_sets_stub.data_sets_stub,
        source_obj_arn=template_arn_param,
        config_data=mock_data_set_references,
        source_entity_type="SourceTemplate",
    )

    source_entity = None
    with pytest.raises(ValueError):
        # Function under test
        source_entity = obj.get_source_entity()

    assert not source_entity