"""Lambda test module."""

import pytest

from src import app
from src.constants import service_name


def test_handler_success(mock_event, request_engine):
    """Test handler execution success."""
    request_engine[service_name.OWS_PRODUCT_REVIEW].add_spec(
        'POST', '/review/20580/apply-revisions-legacy',
        status=200,
        response={'applied_in_legacy': True}
    )
    result = app.handler(mock_event, None)
    assert result is True


def test_handler_error(mock_event, request_engine):
    """Test handler execution failure."""
    request_engine[service_name.OWS_PRODUCT_REVIEW].add_spec(
        'POST', '/review/20580/apply-revisions-legacy',
        status=403,
        response={'foo': 'bar'}
    )
    expected_error_message = (
        'Failed to apply revisions in vapi for review_queue_id 20580.\n'
        'Status code: 403, Error: {"foo": "bar"}'
    )
    with pytest.raises(Exception) as excinfo:
        app.handler(mock_event, None)
    assert excinfo.value.args[0] == expected_error_message