"""Lambda test module."""

from unittest.mock import patch

import config
from src import app as index


@patch('src.app.boto3')
@patch('src.app.boto3')
def test_handler(mock_boto3_stepfunction, mock_boto3_dynamo):
    """Test handler function."""
    # stub test for lambda handler skeleton
    event = {
        'eventSource': 'aws:kafka',
        'records': {
            'event.awalPlusAccountCreation-0': [
                {
                    'topic': 'event.awalPlusAccountCreation',
                    'partition': 0,
                    'offset': 4,
                    'timestamp': 1626669809539,
                    'timestampType': 'CREATE_TIME',
                    'value': 'eyJ2ZW5kb3JfaWQiOiA0MjA1NiwgImN1c3RvbWVyX3R5cGUiOiAiYXdhbF9kaXN0cm8iLCAiZmlyc3RfbmFtZSI6ICJBbGV4YW5kZXIiLCAibGFzdF9uYW1lIjogIklseWluIiwgImVtYWlsIjogIm1sb3JlbnorYXdhbF9iYWNrZmlsbF8yNUB0aGVvcmNoYXJkLmNvbSIsICJzb3VyY2UiOiAiZXZlbnQuYXdhbFBsdXNBY2NvdW50Q3JlYXRpb24iLCAic2VydmljZV90aWVyX3V1aWQiOiAiNWYyYmQ0ZmMtZGY5NC00ZjM1LTk3ZDMtZWYyM2Y4NTczMjc5IiwgImNvcnJlbGF0aW9uX2lkIjogIjdhZDc4NmQwLWQ5M2ItMTVkNS1lMDA0LTJhZmY4ZWI0YWEzZSJ9'  # noqa
                }
            ]
        }
    }
    config.STEP_FN_ARN = 'test-step-function-arn'
    mock_boto3_dynamo.client().get_item.return_value = {}
    execute_response = {'step_fn': 'result'}
    mock_boto3_stepfunction.client().start_execution.return_value = execute_response
    result = index.handler(event, None)
    assert result == {'status': {4: 'success'}}