"""Test event handling."""

from unittest.mock import MagicMock
from unittest.mock import patch

import boto3

from src.logic import event_handling


@patch('src.logic.event_handling.boto3')
def test_execution_already_exists_handling(mock_boto3):
    """Test handling an exception when an execution already exists."""
    fake_client = boto3.client('stepfunctions', 'us-east-1')
    mock_client = MagicMock()
    mock_client.exceptions = fake_client.exceptions
    mock_client.start_execution.side_effect = fake_client.exceptions.ExecutionAlreadyExists(
        {'Error': {'Code': 'foo', 'Message': 'test'}}, 'op')
    mock_boto3.client.return_value = mock_client
    assert event_handling.trigger_stfn({'asset_final_id': 1}) == {
        'executionArn': 'An error occurred (foo) when calling the op operation: test'}