"""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.smeAnalyticsAccountCreation-0': [
                {
                    'topic': 'event.smeAnalyticsAccountCreation',
                    'partition': 0,
                    'offset': 4,
                    'timestamp': 1626669809539,
                    'timestampType': 'CREATE_TIME',
                    'value': 'eyJjb3JyZWxhdGlvbl9pZCI6ICI4ODdlNWYxZS1jNjBhLTQ4MTMtYWRiYy0zYWJiMzM5NDIxZDMiLCAicGFyZW50X3JlcF9vd25lcl9uYW1lIjogIkNvbHVtYmlhIFJlY29yZHMiLCAicGFyZW50X3JlcF9vd25lcl9rZXkiOiAiTDEyMyIsICJyZXBfb3duZXJfbmFtZSI6ICJDb2x1bWJpYSBSZWNvcmRzIEphenoiLCAicmVwX293bmVyX2tleSI6ICJMNDU2In0='  # 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'}}