import pytest

from hello_world import app
from {{ cookiecutter.AWS_Schema_root }} import AWSEvent
from {{ cookiecutter.AWS_Schema_root }} import {{ cookiecutter.AWS_Schema_name }}
from {{ cookiecutter.AWS_Schema_root }} import Marshaller

@pytest.fixture()
def eventBridgeEvent():
    """ Generates EventBridge Event"""

    return {
            "version":"0",
            "id":"7bf73129-1428-4cd3-a780-95db273d1602",
            "detail-type":"{{ cookiecutter.AWS_Schema_detail_type }}",
            "source":"{{ cookiecutter.AWS_Schema_source }}",
            "account":"123456789012",
            "time":"2015-11-11T21:29:54Z",
            "region":"us-east-1",
            "resources":[
              "arn:aws:ec2:us-east-1:123456789012:instance/i-abcd1111"
            ],
            "detail":{
              "ADD-YOUR-FIELDS-HERE":""
            }
    }


def test_lambda_handler(eventBridgeEvent, mocker):

    ret = app.lambda_handler(eventBridgeEvent, "")

    awsEventRet:AWSEvent = Marshaller.unmarshall(ret, AWSEvent)
    detailRet:{{ cookiecutter.AWS_Schema_name }} = awsEventRet.detail

    assert awsEventRet.detail_type.startswith("HelloWorldFunction updated event of ")