refactor(e2e-utils): raise upon invocation error

Author: heitorlessa

tests/e2e/utils/data_fetcher/common.py

@@ -14,10 +14,45 @@ def get_lambda_response(
     lambda_arn: str,
     payload: Optional[str] = None,
     client: Optional[LambdaClient] = None,
+    raise_on_error: bool = True,
 ) -> Tuple[InvocationResponseTypeDef, datetime]:
+    """Invoke function synchronously
+
+    Parameters
+    ----------
+    lambda_arn : str
+        Lambda function ARN to invoke
+    payload : Optional[str], optional
+        JSON payload for Lambda invocation, by default None
+    client : Optional[LambdaClient], optional
+        Boto3 Lambda SDK client, by default None
+    raise_on_error : bool, optional
+        Whether to raise exception upon invocation error, by default True
+
+    Returns
+    -------
+    Tuple[InvocationResponseTypeDef, datetime]
+        Function response and approximate execution time
+
+    Raises
+    ------
+    RuntimeError
+        Function invocation error details
+    """
     client = client or boto3.client("lambda")
     payload = payload or ""
     execution_time = datetime.utcnow()
+
+    response: InvocationResponseTypeDef = client.invoke(
+        FunctionName=lambda_arn,
+        InvocationType="RequestResponse",
+        Payload=payload,
+    )
+    has_error = response.get("FunctionError", "") == "Unhandled"
+    if has_error and raise_on_error:
+        error_payload = response["Payload"].read().decode()
+        raise RuntimeError(f"Function failed invocation: {error_payload}")
+
     return client.invoke(FunctionName=lambda_arn, InvocationType="RequestResponse", Payload=payload), execution_time