Removing payload exception

Author: Cavalcante Damascena

aws_lambda_powertools/event_handler/appsync.py

@@ -3,7 +3,7 @@
 import warnings
 from typing import Any, Callable, Dict, List, Optional, Type, Union
 
-from aws_lambda_powertools.event_handler.exceptions_appsync import InconsistentPayloadError, ResolverNotFoundError
+from aws_lambda_powertools.event_handler.graphql_appsync.exceptions import ResolverNotFoundError
 from aws_lambda_powertools.event_handler.graphql_appsync.router import Router
 from aws_lambda_powertools.utilities.data_classes import AppSyncResolverEvent
 from aws_lambda_powertools.utilities.typing import LambdaContext
@@ -269,13 +269,6 @@ def _call_batch_resolver(self, event: List[dict], data_model: Type[AppSyncResolv
         """
         logger.debug("Processing batch resolver event")
 
-        # All events in the batch must have the same fieldName
-        field_names = {field_name["info"]["fieldName"] for field_name in event}
-        if len(field_names) > 1:
-            raise InconsistentPayloadError(
-                f"All events in the batch must have the same fieldName. Found: {field_names}",
-            )
-
         self.current_batch_event = [data_model(e) for e in event]
         type_name, field_name = self.current_batch_event[0].type_name, self.current_batch_event[0].field_name
 

aws_lambda_powertools/event_handler/exceptions_appsync.py

@@ -0,0 +1,4 @@
+class ResolverNotFoundError(Exception):
+    """
+    When a resolver is not found during a lookup.
+    """

aws_lambda_powertools/event_handler/graphql_appsync/exceptions.py

@@ -4,7 +4,7 @@
 import pytest
 
 from aws_lambda_powertools.event_handler import AppSyncResolver
-from aws_lambda_powertools.event_handler.exceptions_appsync import InconsistentPayloadError, ResolverNotFoundError
+from aws_lambda_powertools.event_handler.graphql_appsync.exceptions import ResolverNotFoundError
 from aws_lambda_powertools.event_handler.graphql_appsync.router import Router
 from aws_lambda_powertools.utilities.data_classes import AppSyncResolverEvent
 from aws_lambda_powertools.utilities.typing import LambdaContext
@@ -568,49 +568,6 @@ async def create_something(event: AppSyncResolverEvent) -> Optional[list]:  # no
     assert not app.current_event
 
 
-def test_resolver_batch_resolver_many_fields_with_different_name():
-    # GIVEN
-    app = AppSyncResolver()
-    router = Router()
-
-    @router.batch_resolver(type_name="Query", field_name="listLocations")
-    def get_locations(event: AppSyncResolverEvent, name: str) -> str:
-        return "get_locations#" + name + "#" + event.source["id"]
-
-    app.include_router(router)
-
-    # WHEN
-    mock_event1 = [
-        {
-            "typeName": "Query",
-            "info": {
-                "fieldName": "listLocations",
-                "parentTypeName": "Query",
-            },
-            "fieldName": "listLocations",
-            "arguments": {"name": "value"},
-            "source": {
-                "id": "1",
-            },
-        },
-        {
-            "typeName": "Query",
-            "info": {
-                "fieldName": "listLocationsDifferent",
-                "parentTypeName": "Query",
-            },
-            "fieldName": "listLocations",
-            "arguments": {"name": "value"},
-            "source": {
-                "id": "1",
-            },
-        },
-    ]
-
-    with pytest.raises(InconsistentPayloadError):
-        app.resolve(mock_event1, LambdaContext())
-
-
 def test_resolver_batch_with_resolver_not_found():
     # GIVEN a AppSyncResolver
     app = AppSyncResolver()