"""Lambda test module."""

from unittest.mock import MagicMock, Mock, patch

import pytest

import app


@patch('app.AdjustmentsJsonImportProcessor')
@patch('app.mysql_connection')
def test_handler(mock_mysql_connection, mock_processor, mock_event):
    """Test handler returns OK when records are imported."""
    context = Mock()
    mock_mysql_connection.return_value.__enter__.return_value = MagicMock()
    mock_processor.return_value.process.return_value = True
    result = app.handler(mock_event, context)

    assert result == {'status': 'OK'}
    mock_processor.assert_called_once_with(
        mock_event, mock_mysql_connection().__enter__()
    )


@patch('app.mysql_connection')
def test_handler_exception_propagation(mock_mysql_connection, mock_event):
    """Test that exceptions are logged and re-raised."""
    context = Mock()
    mock_mysql_connection.side_effect = Exception('MySql Connection Failed')

    with pytest.raises(Exception) as exc_info:
        app.handler(mock_event, context)

    assert 'MySql Connection Failed' in str(exc_info.value)