Skip to content

Commit c33cc00

Browse files
committed
Fix pyright issues in OAuth tests
1 parent 3aef782 commit c33cc00

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

src/mcp/client/auth/oauth2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -883,7 +883,7 @@ def __init__(
883883
self.actor_token_supplier = actor_token_supplier
884884
self.actor_token_type = actor_token_type
885885
self.audience = audience
886-
self.resource = resource or resource_url_from_server_url(server_url)
886+
self.resource: str | None = resource or resource_url_from_server_url(server_url)
887887
self._current_tokens: OAuthToken | None = None
888888
self._token_expiry_time: float | None = None
889889
self._token_lock = anyio.Lock()

tests/unit/client/test_oauth2_providers.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -695,7 +695,7 @@ async def post(self, url: str, *, data: dict[str, str], headers: dict[str, str])
695695
self.last_data = data
696696
return await super().post(url, data=data, headers=headers)
697697

698-
clients = [RecordingAsyncClient()]
698+
clients: list[DummyAsyncClient] = [RecordingAsyncClient()]
699699
monkeypatch.setattr("mcp.client.auth.oauth2.httpx.AsyncClient", AsyncClientFactory(clients))
700700

701701
await provider._request_token()
@@ -1111,7 +1111,10 @@ async def fake_handle_token(self: OAuthClientProvider, response: httpx.Response)
11111111
self.context.current_tokens = token
11121112
await self.context.storage.set_tokens(token)
11131113

1114-
provider._select_scopes = MethodType(lambda self, response: None, provider)
1114+
def fake_select_scopes(self: OAuthClientProvider, response: httpx.Response) -> None:
1115+
return None
1116+
1117+
provider._select_scopes = MethodType(fake_select_scopes, provider)
11151118
monkeypatch.setattr(provider, "_build_protected_resource_discovery_urls", MethodType(fake_build_resource_urls, provider))
11161119
monkeypatch.setattr(provider, "_handle_protected_resource_response", MethodType(fake_handle_resource, provider))
11171120
monkeypatch.setattr(provider, "_get_discovery_urls", MethodType(fake_get_discovery_urls, provider))

tests/unit/server/auth/test_token_handler.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
AuthorizationCodeRequest,
1313
ClientCredentialsRequest,
1414
TokenErrorResponse,
15-
TokenExchangeRequest,
1615
TokenHandler,
1716
TokenSuccessResponse,
1817
)

0 commit comments

Comments
 (0)