From 46da3fa2bdc00c3ffcfe75a86700f9f21d88d6ad Mon Sep 17 00:00:00 2001 From: zzstoatzz Date: Tue, 21 Oct 2025 20:32:56 -0500 Subject: [PATCH] test: acknowledge oauth in-memory storage warnings MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit add pytest.warns assertions for remaining OAuth tests that trigger the in-memory token storage warning in test_client.py and test_mcp_config.py 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude --- tests/client/test_client.py | 42 ++++++++++++++++++++++++++++++------- tests/test_mcp_config.py | 7 ++++++- 2 files changed, 40 insertions(+), 9 deletions(-) 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)