"""Lambda test module."""

from unittest.mock import patch

from content_utils.exceptions import InvalidMessageException
import pytest


def test_handler(mock_event):
    """Test handler function."""
    # Importar e inmediatamente parchear los conectores
    from src import app

    with patch.object(app, 'os_connector') as mock_os:

        result = app.handler(mock_event, None)
        assert mock_os.remove_product.call_count == 1
        mock_os.remove_product.asset_called_with(4340075)
        assert result == {'status': 'success'}

        with pytest.raises(InvalidMessageException):
            app.handler({'foo': 'bar'}, None)