Adding subscribe test

Author: leandrodamascena

aws_lambda_powertools/event_handler/__init__.py

@@ -11,8 +11,8 @@
     Response,
 )
 from aws_lambda_powertools.event_handler.appsync import AppSyncResolver
-from aws_lambda_powertools.event_handler.appsync_events import AppSyncEventsResolver
 from aws_lambda_powertools.event_handler.bedrock_agent import BedrockAgentResolver
+from aws_lambda_powertools.event_handler.events_appsync.appsync_events import AppSyncEventsResolver
 from aws_lambda_powertools.event_handler.lambda_function_url import (
     LambdaFunctionUrlResolver,
 )

aws_lambda_powertools/event_handler/events_appsync/__init__.py

@@ -0,0 +1,5 @@
+from aws_lambda_powertools.event_handler.events_appsync.appsync_events import AppSyncEventsResolver
+
+__all__ = [
+    "AppSyncEventsResolver",
+]

aws_lambda_powertools/event_handler/appsync_events.py renamed to aws_lambda_powertools/event_handler/events_appsync/appsync_events.py

@@ -8,6 +8,7 @@
 from aws_lambda_powertools.event_handler.events_appsync.router import Router
 from aws_lambda_powertools.utilities.data_classes.appsync_resolver_events_event import AppSyncResolverEventsEvent
 from aws_lambda_powertools.warnings import PowertoolsUserWarning
+from aws_lambda_powertools.event_handler.events_appsync.exceptions import UnauthorizedException
 
 if TYPE_CHECKING:
     from collections.abc import Callable
@@ -153,6 +154,8 @@ def _subscribe_events(self) -> Any:
         if resolver:
             try:
                 return resolver["func"]()
+            except UnauthorizedException:
+                raise
             except Exception as error:
                 return {"error": self._format_error_response(error)}
 
@@ -239,6 +242,8 @@ def _process_publish_event_sync_resolver(
                     )
 
                 return {"events": response}
+            except UnauthorizedException:
+                raise
             except Exception as error:
                 return {"error": self._format_error_response(error)}
 
@@ -291,6 +296,8 @@ async def _call_publish_event_async_resolver(
                     )
 
                 return {"events": response}
+            except UnauthorizedException:
+                raise
             except Exception as error:
                 return {"error": self._format_error_response(error)}
 

aws_lambda_powertools/event_handler/events_appsync/exceptions.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+
+class UnauthorizedException(Exception):
+    """
+    Error to be thrown to communicate the subscription is unauthorized.
+
+    When this error is raised, the client will receive a 40x error code
+    and the subscription will be closed.
+
+    Attributes:
+        message (str): The error message describing the unauthorized access.
+    """
+
+    def __init__(self, message: str | None = None, *args, **kwargs):
+        """
+        Initialize the UnauthorizedException.
+
+        Args:
+            message (str): A descriptive error message.
+            *args: Variable positional arguments.
+            **kwargs: Variable keyword arguments.
+        """
+        super().__init__(message, *args, **kwargs)
+        self.name = "UnauthorizedException"

tests/functional/event_handler/required_dependencies/appsync/test_appsync_events_resolvers.py

@@ -4,6 +4,7 @@
 import pytest
 
 from aws_lambda_powertools.event_handler import AppSyncEventsResolver
+from aws_lambda_powertools.event_handler.events_appsync.exceptions import UnauthorizedException
 from aws_lambda_powertools.event_handler.events_appsync.router import Router
 from aws_lambda_powertools.warnings import PowertoolsUserWarning
 from tests.functional.utils import load_event
@@ -1571,3 +1572,40 @@ def test_handler():
 
     # THEN we should get an error response
     assert not result
+
+def test_publish_events_throw_unauthorized_exception(lambda_context, mock_event):
+    """Test handling events with an empty payload."""
+    # GIVEN a sample publish event with empty events
+    mock_event["info"]["operation"] = "PUBLISH"
+    mock_event["info"]["channel"]["path"] = "/default/test"
+    mock_event["events"] = [
+        {"id": "123", "payload": {"data": "test data"}},
+    ]
+
+    # GIVEN an AppSyncEventsResolver
+    app = AppSyncEventsResolver()
+
+    @app.on_publish(path="/default/*", aggregate=True)
+    def handle_events(payload):
+        raise UnauthorizedException
+
+    # WHEN we resolve the event with unauthorized route
+    with pytest.raises(UnauthorizedException):
+        app.resolve(mock_event, lambda_context)
+
+def test_subscribe_events_throw_unauthorized_exception(lambda_context, mock_event):
+    """Test handling events with an empty payload."""
+    # GIVEN a sample publish event with empty events
+    mock_event["info"]["operation"] = "SUBSCRIBE"
+    mock_event["info"]["channel"]["path"] = "/default/test"
+
+    # GIVEN an AppSyncEventsResolver
+    app = AppSyncEventsResolver()
+
+    @app.on_subscribe(path="/default/*")
+    def handle_events():
+        raise UnauthorizedException
+
+    # WHEN we resolve the event with unauthorized route
+    with pytest.raises(UnauthorizedException):
+        app.resolve(mock_event, lambda_context)