Merge pull request #27 from sacha-development-stuff/revert-26-codex/fix-oauthtoken-error-in-test-case

Revert "Fix OAuth authorization flow to use auth code exchange"

Author: SoldierSacha

src/mcp/client/auth/oauth2.py

@@ -433,9 +433,11 @@ def _select_scopes(self, init_response: httpx.Response) -> None:
 
     # Discovery and registration helpers provided by BaseOAuthProvider
 
-    async def _perform_authorization(self) -> tuple[str, str]:
-        """Perform the authorization flow and return authorization code data."""
-        return await self._perform_authorization_code_grant()
+    async def _perform_authorization(self) -> httpx.Request:
+        """Perform the authorization flow."""
+        auth_code, code_verifier = await self._perform_authorization_code_grant()
+        token_request = await self._exchange_token_authorization_code(auth_code, code_verifier)
+        return token_request
 
     async def _perform_authorization_code_grant(self) -> tuple[str, str]:
         """Perform the authorization redirect and get auth code."""
@@ -685,18 +687,14 @@ async def async_auth_flow(self, request: httpx.Request) -> AsyncGenerator[httpx.
                             break  # Non-4XX error, stop trying
 
                     # Step 4: Register client if needed
-                    if self.context.client_info and not self._client_info:
-                        self._client_info = self.context.client_info
                     registration_request = self._create_registration_request(self._metadata)
                     if registration_request:
                         registration_response = yield registration_request
                         await self._handle_registration_response(registration_response)
                         self.context.client_info = self._client_info
 
                     # Step 5: Perform authorization and complete token exchange
-                    auth_code, code_verifier = await self._perform_authorization()
-                    token_request = await self._exchange_token_authorization_code(auth_code, code_verifier)
-                    token_response = yield token_request
+                    token_response = yield await self._perform_authorization()
                     await self._handle_token_response(token_response)
                 except Exception:  # pragma: no cover
                     logger.exception("OAuth flow error")
@@ -717,9 +715,7 @@ async def async_auth_flow(self, request: httpx.Request) -> AsyncGenerator[httpx.
                         self._select_scopes(response)
 
                         # Step 2b: Perform (re-)authorization and token exchange
-                        auth_code, code_verifier = await self._perform_authorization()
-                        token_request = await self._exchange_token_authorization_code(auth_code, code_verifier)
-                        token_response = yield token_request
+                        token_response = yield await self._perform_authorization()
                         await self._handle_token_response(token_response)
                     except Exception:  # pragma: no cover
                         logger.exception("OAuth flow error")