diff --git a/tests/client/test_client.py b/tests/client/test_client.py index a32e92073..e47b15260 100644 --- a/tests/client/test_client.py +++ b/tests/client/test_client.py @@ -968,26 +968,52 @@ class TestAuth: Client(transport=StdioTransport("echo", ["hello"]), auth="oauth") def test_oauth_literal_sets_up_oauth_shttp(self): - client = Client( - transport=StreamableHttpTransport("http://localhost:8000"), auth="oauth" - ) + # Expect warning about in-memory token storage + with pytest.warns( + UserWarning, + match="Using in-memory token storage is not recommended for production use", + ): + client = Client( + transport=StreamableHttpTransport("http://localhost:8000"), auth="oauth" + ) assert isinstance(client.transport, StreamableHttpTransport) assert isinstance(client.transport.auth, OAuthClientProvider) def test_oauth_literal_pass_direct_to_transport(self): - client = Client( - transport=StreamableHttpTransport("http://localhost:8000", auth="oauth"), - ) + # Expect warning about in-memory token storage + with pytest.warns( + UserWarning, + match="Using in-memory token storage is not recommended for production use", + ): + client = Client( + transport=StreamableHttpTransport( + "http://localhost:8000", auth="oauth" + ), + ) assert isinstance(client.transport, StreamableHttpTransport) assert isinstance(client.transport.auth, OAuthClientProvider) def test_oauth_literal_sets_up_oauth_sse(self): - client = Client(transport=SSETransport("http://localhost:8000"), auth="oauth") + # Expect warning about in-memory token storage + with pytest.warns( + UserWarning, + match="Using in-memory token storage is not recommended for production use", + ): + client = Client( + transport=SSETransport("http://localhost:8000"), auth="oauth" + ) assert isinstance(client.transport, SSETransport) assert isinstance(client.transport.auth, OAuthClientProvider) def test_oauth_literal_pass_direct_to_transport_sse(self): - client = Client(transport=SSETransport("http://localhost:8000", auth="oauth")) + # Expect warning about in-memory token storage + with pytest.warns( + UserWarning, + match="Using in-memory token storage is not recommended for production use", + ): + client = Client( + transport=SSETransport("http://localhost:8000", auth="oauth") + ) assert isinstance(client.transport, SSETransport) assert isinstance(client.transport.auth, OAuthClientProvider) diff --git a/tests/test_mcp_config.py b/tests/test_mcp_config.py index d0cac215e..f1f9eb564 100644 --- a/tests/test_mcp_config.py +++ b/tests/test_mcp_config.py @@ -461,7 +461,12 @@ async def test_remote_config_with_oauth_literal(): } } } - client = Client(config) + # Expect warning about in-memory token storage + with pytest.warns( + UserWarning, + match="Using in-memory token storage is not recommended for production use", + ): + client = Client(config) assert isinstance(client.transport.transport, StreamableHttpTransport) assert isinstance(client.transport.transport.auth, OAuthClientProvider)