test: example of managing lifespan state for BedrockAgentCo…

pyproject.toml

@@ -34,6 +34,7 @@ dependencies = [
     "typing-extensions>=4.13.2,<5.0.0",
     "uvicorn>=0.34.2",
     "websockets>=13.0",
+    "httpx2>=2.12.0",
 ]
 
 [tool.hatch.metadata]

tests/bedrock_agentcore/runtime/test_app.py

@@ -8,8 +8,10 @@
 import time
 from datetime import datetime
 from decimal import Decimal
+from typing import AsyncIterator, TypedDict
 from unittest.mock import MagicMock, patch
 
+import httpx2
 import pytest
 from starlette.testclient import TestClient
 
@@ -404,6 +406,34 @@ async def lifespan(app):
             assert startup_called is True
         assert shutdown_called is True
 
+    def test_lifespan_state_access(self):
+        """ Test that we can access lifespan state """
+
+        expected_greeting = "hello world"
+
+        class AppState(TypedDict):
+            greeting: str
+
+        @contextlib.asynccontextmanager              
+        async def lifespan(app) -> AsyncIterator[AppState]:
+            async with httpx2.AsyncClient() as client:
+                yield { "greeting": expected_greeting }
+
+        app = BedrockAgentCoreApp(lifespan=lifespan)
+
+        @app.entrypoint
+        async def invoke(payload, context):
+            greet = context.request.state["greeting"]
+            return {"greeting": greet} 
+
+        with TestClient(app) as client:
+            response = client.post("/invocations", json={"input": "test"})
+            assert response.status_code == 200
+
+            payload = response.json()
+            assert "greeting" in payload
+            assert payload["greeting"] == expected_greeting
+
     def test_initialization_without_lifespan(self):
         """Test that BedrockAgentCoreApp still works without lifespan."""
         app = BedrockAgentCoreApp()  # No lifespan parameter