add more tests for parser and resolver

Author: anafalcao

aws_lambda_powertools/utilities/parser/envelopes/__init__.py

@@ -2,7 +2,7 @@
 from .apigw_websocket import ApiGatewayWebSocketEnvelope
 from .apigwv2 import ApiGatewayV2Envelope
 from .base import BaseEnvelope
-from .bedrock_agent import BedrockAgentEnvelope
+from .bedrock_agent import BedrockAgentEnvelope, BedrockAgentFunctionEnvelope
 from .cloudwatch import CloudWatchLogsEnvelope
 from .dynamodb import DynamoDBStreamEnvelope
 from .event_bridge import EventBridgeEnvelope
@@ -20,6 +20,7 @@
     "ApiGatewayV2Envelope",
     "ApiGatewayWebSocketEnvelope",
     "BedrockAgentEnvelope",
+    "BedrockAgentFunctionEnvelope",
     "CloudWatchLogsEnvelope",
     "DynamoDBStreamEnvelope",
     "EventBridgeEnvelope",

tests/functional/event_handler/required_dependencies/test_bedrock_agent_functions.py

@@ -132,3 +132,23 @@ def test_resolve_raises_value_error_on_missing_required_field():
         resolver.resolve(incomplete_event, {})
 
     assert "Missing required field:" in str(excinfo.value)
+
+
+def test_resolve_with_no_registered_function():
+    # GIVEN a Bedrock Agent Function resolver
+    app = BedrockAgentFunctionResolver()
+
+    # AND a valid event but with a non-existent function
+    raw_event = {
+        "messageVersion": "1.0",
+        "agent": {"name": "TestAgent", "id": "test-id", "alias": "test", "version": "1"},
+        "actionGroup": "test_group",
+        "function": "non_existent_function",
+        "parameters": [],
+    }
+
+    # WHEN calling resolve with a non-existent function
+    result = app.resolve(raw_event, {})
+
+    # THEN the response should contain an error message
+    assert "Error: 'non_existent_function'" in result["response"]["functionResponse"]["responseBody"]["TEXT"]["body"]

tests/unit/parser/_pydantic/test_bedrock_agent.py

@@ -108,3 +108,16 @@ def test_bedrock_agent_function_event():
     assert parameters[1].name == "endDate"
     assert parameters[1].type_ == "string"
     assert parameters[1].value == "2024-03-20"
+
+
+def test_bedrock_agent_function_event_with_envelope():
+    raw_event = load_event("bedrockAgentFunctionEvent.json")
+    raw_event["inputText"] = '{"username": "Jane", "name": "Doe"}'
+    parsed_event: MyBedrockAgentBusiness = parse(
+        event=raw_event,
+        model=MyBedrockAgentBusiness,
+        envelope=envelopes.BedrockAgentFunctionEnvelope,
+    )
+
+    assert parsed_event.username == "Jane"
+    assert parsed_event.name == "Doe"