#!/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 json
import unittest
from unittest.mock import patch

import boto3
import pytest
from moto import mock_firehose
from test.test_topic import create_s3_delivery_stream


@mock_firehose
def test_lambda_function_with_topic_event():
    firehose = boto3.client("firehose", region_name="us-east-1")
    create_s3_delivery_stream(firehose, "Topics")
    from lambda_function import handler

    topic_event = {
        "version": "0",
        "id": "de55e880-0f1d-4b1d-982e-23ed13e45aaa",
        "detail-type": "topics",
        "source": "com.analyze.topic.inference.topics",
        "account": "FAKEACCOUNT",
        "time": "2020-06-24T17:16:02Z",
        "region": "us-west-2",
        "resources": [],
        "detail": {
            "000": [
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "000",
                    "term": "health",
                    "weight": "0.09484477",
                },
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "000",
                    "term": "walk",
                    "weight": "0.020982718",
                },
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "000",
                    "term": "place",
                    "weight": "0.004689377",
                    "created_at": "2020-06-24",
                },
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "000",
                    "term": "like",
                    "weight": "0.0056834435",
                },
            ],
            "001": [
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "001",
                    "term": "fun",
                    "weight": "0.13023746",
                },
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "001",
                    "term": "movie",
                    "weight": "0.002189455",
                },
                {
                    "job_id": "1234567890123456789012345",
                    "job_timestamp": "2020-06-26T19:05:16.785Z",
                    "topic": "001",
                    "term": "song",
                    "weight": "0.002034978",
                },
            ],
        },
    }

    with patch.dict(
        "os.environ",
        {"TOPICS_NS": "com.analyze.topic.inference.topics", "TOPIC_MAPPINGS_NS": "com.analyze.inference.mappings"},
    ):
        handler(topic_event, None)


@mock_firehose
def test_lambda_function_with_mapping_event():
    firehose = boto3.client("firehose", region_name="us-east-1")
    create_s3_delivery_stream(firehose, "TopicMappings")
    from lambda_function import handler

    mapping_event = {
        "version": "0",
        "id": "b2123492-5ecc-1a7a-33b6-58e9798e9a27",
        "detail-type": "mappings",
        "source": "com.analyze.topic.inference.mappings",
        "account": "FAKEACCOUNT",
        "time": "2020-06-24T17:16:05Z",
        "region": "us-west-2",
        "resources": [],
        "detail": {
            "platform": "twitter",
            "job_id": "1234567890123456789012345",
            "job_timestamp": "2020-06-26T19:05:16.785Z",
            "id_str": "1274357316737957888",
            "topic": "000",
        },
    }

    with patch.dict(
        "os.environ",
        {"TOPICS_NS": "com.analyze.inference.topics", "TOPIC_MAPPINGS_NS": "com.analyze.topic.inference.mappings"},
    ):
        handler(mapping_event, None)