"""Lambda test module."""

from unittest.mock import patch

import pytest

from src import app

URL = 'https://www.youtube.com/channel/UC3IZKseVpdzPSBaWxBxundA'


@patch('src.app.youtube_api_utils')
def test_handler(youtube_api_utils_mock, mock_event):
    """Test handler successful execution."""
    youtube_api_utils_mock.get_youtube_subscribers.return_value = 100

    result = app.handler(mock_event, {})
    youtube_api_utils_mock.get_youtube_subscribers.assert_called_with(URL)
    assert result == {'youtube_subscribers': 100}


@pytest.mark.parametrize(
    'error,raised_error',
    [
        (OSError(), app.FatalException())
    ])
def test_handler_failure(mock_event, error, raised_error):
    """Test handler successful execution."""
    with patch('src.app.youtube_api_utils') as youtube_api_utils_mock:
        youtube_api_utils_mock.get_youtube_subscribers.side_effect = error
        with pytest.raises(type(raised_error)):
            app.handler(mock_event, {})