fix tests

Author: dmandar

.vscode/launch.json

@@ -12,7 +12,12 @@
         "PYTHONPATH": "${workspaceFolder}"
       },
       "cwd": "${workspaceFolder}/examples/helloworld",
-      "args": ["--host", "localhost", "--port", "9999"]
+      "args": [
+        "--host",
+        "localhost",
+        "--port",
+        "9999"
+      ]
     },
     {
       "name": "Debug Currency Agent",
@@ -25,7 +30,24 @@
         "PYTHONPATH": "${workspaceFolder}"
       },
       "cwd": "${workspaceFolder}/examples/langgraph",
-      "args": ["--host", "localhost", "--port", "10000"]
+      "args": [
+        "--host",
+        "localhost",
+        "--port",
+        "10000"
+      ]
+    },
+    {
+      "name": "Pytest All",
+      "type": "debugpy",
+      "request": "launch",
+      "module": "pytest",
+      "args": [
+        "-v",
+        "-s"
+      ],
+      "console": "integratedTerminal",
+      "justMyCode": true // Or false if you need to step into library code
     }
   ]
-}
+}

src/a2a/client/auth/credentials.py

@@ -22,31 +22,30 @@ async def get_credentials(
 
 class InMemoryContextCredentialStore(CredentialService):
     """
-    A simple in-memory store for context-keyed credentials.
+    A simple in-memory store for session-keyed credentials.
 
-    This class uses the 'contextId' from the ClientCallContext state to
-    store and retrieve credentials, providing a simple, user-specific
-    credential mechanism without requiring a full user authentication system.
+    This class uses the 'sessionId' from the ClientCallContext state to
+    store and retrieve credentials...
     """
 
     def __init__(self):
-        # {context_id: {scheme_name: credential}}
+        # {session_id: {scheme_name: credential}}
         self._store: dict[str, dict[str, str]] = {}
 
     async def get_credentials(
         self,
         security_scheme_name: str,
         context: ClientCallContext | None,
     ) -> str | None:
-        if not context or 'contextId' not in context.state:
+        if not context or 'sessionId' not in context.state:
             return None
-        context_id = context.state['contextId']
-        return self._store.get(context_id, {}).get(security_scheme_name)
+        session_id = context.state['sessionId']
+        return self._store.get(session_id, {}).get(security_scheme_name)
 
-    async def set_credentials(
-        self, context_id: str, security_scheme_name: str, credential: str
+    async def set_credential(
+        self, session_id: str, security_scheme_name: str, credential: str
     ):
         """Method to populate the store."""
-        if context_id not in self._store:
-            self._store[context_id] = {}
-        self._store[context_id][security_scheme_name] = credential
+        if session_id not in self._store:
+            self._store[session_id] = {}
+        self._store[session_id][security_scheme_name] = credential

src/a2a/client/auth/interceptor.py

@@ -5,7 +5,8 @@
 
 from a2a.client.auth.credentials import CredentialService
 from a2a.client.middleware import ClientCallContext, ClientCallInterceptor
-from a2a.types import AgentCard, APIKeySecurityScheme, HTTPAuthSecurityScheme, In, OAuth2SecurityScheme, OpenIdConnectSecurityScheme
+from a2a.types import (AgentCard, APIKeySecurityScheme, HTTPAuthSecurityScheme,
+                       In, OAuth2SecurityScheme, OpenIdConnectSecurityScheme)
 
 logger = logging.getLogger(__name__)
 

src/a2a/client/client.py

@@ -270,7 +270,7 @@ async def send_message_streaming(
             self.httpx_client,
             'POST',
             self.url,
-            json=payload
+            json=payload,
             **modified_kwargs,
         ) as event_source:
             try:

tests/client/test_auth_middleware.py

@@ -6,29 +6,15 @@
 import pytest
 import respx
 
-from a2a.client import (
-    A2AClient,
-    ClientCallContext,
-    ClientCallInterceptor,
-)
-from a2a.client.auth import (
-    AuthInterceptor,
-    CredentialService,
-    InMemoryContextCredentialStore,
-)
-from a2a.types import (
-    AgentCard,
-    AgentCapabilities,
-    APIKeySecurityScheme,
-    AuthorizationCodeOAuthFlow,
-    HTTPAuthSecurityScheme,
-    In,
-    OAuth2SecurityScheme,
-    OAuthFlows,
-    OpenIdConnectSecurityScheme,
-    SecurityScheme,
-    SendMessageRequest,
-)
+from a2a.client import A2AClient, ClientCallContext, ClientCallInterceptor
+from a2a.client.auth import (AuthInterceptor, CredentialService,
+                             InMemoryContextCredentialStore)
+from a2a.types import (AgentCapabilities, AgentCard, APIKeySecurityScheme,
+                       AuthorizationCodeOAuthFlow, HTTPAuthSecurityScheme, In,
+                       OAuth2SecurityScheme, OAuthFlows,
+                       OpenIdConnectSecurityScheme, SecurityScheme,
+                       SendMessageRequest)
+
 
 # A simple mock interceptor for testing basic middleware functionality
 class HeaderInterceptor(ClientCallInterceptor):
@@ -60,7 +46,6 @@ async def test_client_with_simple_interceptor():
     # Arrange
     test_url = "http://fake-agent.com/rpc"
     header_interceptor = HeaderInterceptor("X-Test-Header", "Test-Value-123")
-    
     async with httpx.AsyncClient() as http_client:
         client = A2AClient(
             httpx_client=http_client,