Fix Flaky test.

Removed usage of snapshot in parametrised test as they don't work well together and end up failing the test.

Author: AishwaryaKalloli

tests/client/test_auth.py

@@ -10,7 +10,6 @@
 
 import httpx
 import pytest
-from inline_snapshot import snapshot
 from pydantic import AnyHttpUrl
 
 from mcp.client.auth import OAuthClientProvider
@@ -968,8 +967,7 @@ def test_build_metadata(
         revocation_options=RevocationOptions(enabled=True),
     )
 
-    assert metadata == snapshot(
-        OAuthMetadata(
+    assert metadata == OAuthMetadata(
             issuer=AnyHttpUrl(issuer_url),
             authorization_endpoint=AnyHttpUrl(authorization_endpoint),
             token_endpoint=AnyHttpUrl(token_endpoint),
@@ -982,4 +980,3 @@ def test_build_metadata(
             revocation_endpoint_auth_methods_supported=["client_secret_post"],
             code_challenge_methods_supported=["S256"],
         )
-    )

tests/issues/test_188_concurrency.py

@@ -14,29 +14,38 @@
 @pytest.mark.anyio
 async def test_messages_are_executed_concurrently():
     server = FastMCP("test")
-
+    call_timestamps = []
+    
     @server.tool("sleep")
     async def sleep_tool():
+        call_timestamps.append(("tool_start_time", anyio.current_time()))
         await anyio.sleep(_sleep_time_seconds)
+        call_timestamps.append(("tool_end_time", anyio.current_time()))
         return "done"
 
     @server.resource(_resource_name)
     async def slow_resource():
+        call_timestamps.append(("resource_start_time", anyio.current_time()))
         await anyio.sleep(_sleep_time_seconds)
+        call_timestamps.append(("resource_end_time", anyio.current_time()))
         return "slow"
 
     async with create_session(server._mcp_server) as client_session:
-        start_time = anyio.current_time()
         async with anyio.create_task_group() as tg:
             for _ in range(10):
                 tg.start_soon(client_session.call_tool, "sleep")
                 tg.start_soon(client_session.read_resource, AnyUrl(_resource_name))
 
-        end_time = anyio.current_time()
-
-        duration = end_time - start_time
-        assert duration < 6 * _sleep_time_seconds
-        print(duration)
+        active_calls = 0
+        max_concurrent_calls = 0
+        for call_type, _ in sorted(call_timestamps, key=lambda x: x[1]):
+            if "start" in call_type:
+                active_calls += 1
+                max_concurrent_calls = max(max_concurrent_calls, active_calls)
+            else:
+                active_calls -= 1
+        print(f"Max concurrent calls: {max_concurrent_calls}")
+        assert max_concurrent_calls > 1, "No concurrent calls were executed"
 
 
 def main():