"""Lambda test module."""

from unittest.mock import patch

import pytest

from src import app


@patch('src.app.get_formatted_product_status_data')
def test_handler_success(mock_fetch, mock_event):
    """Test handler function."""
    mock_fetch.return_value = {
        'release_status': 'foo',
        'error_correction_status': 'bar',
        'release_correction_id': 123
    }
    result = app.handler(mock_event, None)
    assert result == {
        'release_status': 'foo',
        'error_correction_status': 'bar',
        'release_correction_id': 123
    }

    mock_fetch.return_value = {
        'release_status': 'baz',
        'error_correction_status': None,
        'release_correction_id': None
    }
    result = app.handler(mock_event, None)
    assert result == {
        'release_status': 'baz',
        'error_correction_status': None,
        'release_correction_id': None
    }


@patch('src.app.get_formatted_product_status_data')
def test_handler_error(mock_fetch, mock_event):
    """Test handler function."""
    mock_fetch.side_effect = Exception('foo')
    with pytest.raises(Exception):
        app.handler(mock_event, None)
    assert mock_fetch.call_count == 1