"""Lambda test module."""

from unittest.mock import patch

import pytest

from src import app
from src.exceptions import ProductNotSubmittedException


@patch('src.app.complete_product')
def test_handler_success(mock_logic, mock_event):
    """Test handler function."""
    result = app.handler(mock_event, None)
    assert result == {'status': 'success'}


@patch('src.app.complete_product')
def test_handler_skip(mock_logic, mock_event):
    """Test skip result."""
    mock_logic.complete_product_digital.side_effect = ProductNotSubmittedException('foo')
    result = app.handler(mock_event, None)
    assert result == {'status': 'skip'}


@patch('src.app.complete_product')
def test_handler_error(mock_logic, mock_event):
    """Test error result."""
    mock_logic.complete_product_digital.side_effect = Exception('foo')
    with pytest.raises(Exception) as e:
        app.handler(mock_event, None)
    assert e.value.args[0] == 'foo'