Fix OAuth example test collection

Author: kinocode3

tests/examples/simple_auth/conftest.py

@@ -0,0 +1,23 @@
+from __future__ import annotations
+
+import importlib
+import sys
+from collections.abc import Callable
+from pathlib import Path
+from types import ModuleType
+
+import pytest
+
+@pytest.fixture
+def load_example_module() -> Callable[[Path, str], ModuleType]:
+    """Import a workspace example without requiring it in the root test environment."""
+
+    def load(package_root: Path, module_name: str) -> ModuleType:
+        original_path = sys.path.copy()
+        try:
+            sys.path.insert(0, str(package_root))
+            return importlib.import_module(module_name)
+        finally:
+            sys.path[:] = original_path
+
+    return load

…-client/tests/test_oauth_resource_url.py …s/simple_auth/test_oauth_resource_url.pyexamples/clients/simple-auth-client/tests/test_oauth_resource_url.py renamed to tests/examples/simple_auth/test_oauth_resource_url.py examples/clients/simple-auth-client/tests/test_oauth_resource_url.py renamed to tests/examples/simple_auth/test_oauth_resource_url.py

@@ -1,19 +1,43 @@
-from collections.abc import AsyncIterator
+from __future__ import annotations
+
+from collections.abc import AsyncIterator, Callable
 from contextlib import asynccontextmanager
+from pathlib import Path
+from types import ModuleType
+from typing import Protocol, cast
 
 import anyio
 import pytest
 from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
+
 from mcp.client.auth import OAuthClientProvider
 from mcp.shared.message import SessionMessage
 
-from mcp_simple_auth_client import main as client_module
-from mcp_simple_auth_client.main import SimpleAuthClient
+CLIENT_ROOT = Path(__file__).parents[3] / "examples" / "clients" / "simple-auth-client"
+
+
+class SimpleAuthClient(Protocol):
+    def __init__(
+        self,
+        server_url: str,
+        transport_type: str = "streamable-http",
+        client_metadata_url: str | None = None,
+    ) -> None: ...
+
+    async def connect(self) -> None: ...
+
+
+class ClientModule(Protocol):
+    SimpleAuthClient: type[SimpleAuthClient]
 
 
 @pytest.mark.anyio
-async def test_oauth_client_preserves_the_complete_connection_url(monkeypatch: pytest.MonkeyPatch) -> None:
+async def test_oauth_client_preserves_the_complete_connection_url(
+    monkeypatch: pytest.MonkeyPatch,
+    load_example_module: Callable[[Path, str], ModuleType],
+) -> None:
     """The example passes the opaque MCP endpoint unchanged to its OAuth provider."""
+    client_module = cast(ClientModule, load_example_module(CLIENT_ROOT, "mcp_simple_auth_client.main"))
     resource_url = "https://mcp.example.com/prefix/mcp?tenant=mcp"
     providers: list[OAuthClientProvider] = []
     sessions = 0
@@ -49,9 +73,9 @@ async def record_session(
 
     monkeypatch.setattr(client_module, "CallbackServer", FakeCallbackServer)
     monkeypatch.setattr(client_module, "sse_client", fake_sse_client)
-    monkeypatch.setattr(SimpleAuthClient, "_run_session", record_session)
+    monkeypatch.setattr(client_module.SimpleAuthClient, "_run_session", record_session)
 
-    await SimpleAuthClient(resource_url, transport_type="sse").connect()
+    await client_module.SimpleAuthClient(resource_url, transport_type="sse").connect()
 
     assert sessions == 1
     assert [str(provider.context.server_url) for provider in providers] == [resource_url]

…-auth/tests/test_resource_server_urls.py …simple_auth/test_resource_server_urls.pyexamples/servers/simple-auth/tests/test_resource_server_urls.py renamed to tests/examples/simple_auth/test_resource_server_urls.py examples/servers/simple-auth/tests/test_resource_server_urls.py renamed to tests/examples/simple_auth/test_resource_server_urls.py

@@ -1,22 +1,35 @@
-from typing import Literal
+from __future__ import annotations
+
+from collections.abc import Callable
+from pathlib import Path
+from types import ModuleType
+from typing import Literal, Protocol, cast
 
 import pytest
+from click import Command
 from click.testing import CliRunner
-from mcp_simple_auth import server
 
 from mcp.server.mcpserver.server import MCPServer
 
+SERVER_ROOT = Path(__file__).parents[3] / "examples" / "servers" / "simple-auth"
+
+
+class ServerModule(Protocol):
+    main: Command
+
 
 @pytest.mark.parametrize(
     ("transport", "endpoint"),
     [("streamable-http", "/mcp"), ("sse", "/sse")],
 )
 def test_selected_transport_uses_one_resource_path(
     monkeypatch: pytest.MonkeyPatch,
+    load_example_module: Callable[[Path, str], ModuleType],
     transport: Literal["sse", "streamable-http"],
     endpoint: str,
 ) -> None:
     """The example advertises and serves the selected transport path."""
+    server = cast(ServerModule, load_example_module(SERVER_ROOT, "mcp_simple_auth.server"))
     created: list[MCPServer] = []
     run_arguments: list[dict[str, object]] = []