"""Lambda test module."""

from unittest.mock import MagicMock, patch

import app
import pytest


@patch('app.CommitRoyaltyProcessor')
@patch('app.SnowflakeSQLExecutor')
def test_handler(mock_snowflake, mock_processor, mock_event):
    """Test handler function."""
    mock_snowflake.return_value.__enter__.return_value = MagicMock()
    result = app.handler(mock_event, None)
    assert result == {'status': 'OK'}
    mock_processor.assert_called_once_with(mock_event, mock_snowflake().__enter__())


@patch('app.CommitRoyaltyProcessor')
@patch('app.sentry_sdk')
@patch('app.SnowflakeSQLExecutor')
def test_handler_exception(mock_snowflake, mock_sentry, mock_processor, mock_event):
    """Test sentry is called."""
    mock_sentry.return_value = True
    mock_processor.return_value.process.side_effect = Exception('key error')
    with pytest.raises(Exception) as e:
        app.handler(mock_event, None)
        mock_sentry.assert_called_once_with(e)