@@ -36,7 +36,7 @@ const mockOAuthClientInfo = {
36
36
// Mock MCP SDK functions - must be before imports
37
37
jest . mock ( "@modelcontextprotocol/sdk/client/auth.js" , ( ) => ( {
38
38
auth : jest . fn ( ) ,
39
- discoverOAuthMetadata : jest . fn ( ) ,
39
+ discoverAuthorizationServerMetadata : jest . fn ( ) ,
40
40
registerClient : jest . fn ( ) ,
41
41
startAuthorization : jest . fn ( ) ,
42
42
exchangeAuthorization : jest . fn ( ) ,
@@ -46,7 +46,7 @@ jest.mock("@modelcontextprotocol/sdk/client/auth.js", () => ({
46
46
47
47
// Import the functions to get their types
48
48
import {
49
- discoverOAuthMetadata ,
49
+ discoverAuthorizationServerMetadata ,
50
50
registerClient ,
51
51
startAuthorization ,
52
52
exchangeAuthorization ,
@@ -57,9 +57,10 @@ import { OAuthMetadata } from "@modelcontextprotocol/sdk/shared/auth.js";
57
57
import { EMPTY_DEBUGGER_STATE } from "@/lib/auth-types" ;
58
58
59
59
// Type the mocked functions properly
60
- const mockDiscoverOAuthMetadata = discoverOAuthMetadata as jest . MockedFunction <
61
- typeof discoverOAuthMetadata
62
- > ;
60
+ const mockDiscoverAuthorizationServerMetadata =
61
+ discoverAuthorizationServerMetadata as jest . MockedFunction <
62
+ typeof discoverAuthorizationServerMetadata
63
+ > ;
63
64
const mockRegisterClient = registerClient as jest . MockedFunction <
64
65
typeof registerClient
65
66
> ;
@@ -102,7 +103,9 @@ describe("AuthDebugger", () => {
102
103
// Suppress console errors in tests to avoid JSDOM navigation noise
103
104
jest . spyOn ( console , "error" ) . mockImplementation ( ( ) => { } ) ;
104
105
105
- mockDiscoverOAuthMetadata . mockResolvedValue ( mockOAuthMetadata ) ;
106
+ mockDiscoverAuthorizationServerMetadata . mockResolvedValue (
107
+ mockOAuthMetadata ,
108
+ ) ;
106
109
mockRegisterClient . mockResolvedValue ( mockOAuthClientInfo ) ;
107
110
mockDiscoverOAuthProtectedResourceMetadata . mockRejectedValue (
108
111
new Error ( "No protected resource metadata found" ) ,
@@ -203,7 +206,7 @@ describe("AuthDebugger", () => {
203
206
} ) ;
204
207
205
208
// Should first discover and save OAuth metadata
206
- expect ( mockDiscoverOAuthMetadata ) . toHaveBeenCalledWith (
209
+ expect ( mockDiscoverAuthorizationServerMetadata ) . toHaveBeenCalledWith (
207
210
new URL ( "https://example.com/" ) ,
208
211
) ;
209
212
@@ -216,7 +219,7 @@ describe("AuthDebugger", () => {
216
219
} ) ;
217
220
218
221
it ( "should show error when quick OAuth flow fails to discover metadata" , async ( ) => {
219
- mockDiscoverOAuthMetadata . mockRejectedValue (
222
+ mockDiscoverAuthorizationServerMetadata . mockRejectedValue (
220
223
new Error ( "Metadata discovery failed" ) ,
221
224
) ;
222
225
@@ -362,7 +365,7 @@ describe("AuthDebugger", () => {
362
365
fireEvent . click ( screen . getByText ( "Continue" ) ) ;
363
366
} ) ;
364
367
365
- expect ( mockDiscoverOAuthMetadata ) . toHaveBeenCalledWith (
368
+ expect ( mockDiscoverAuthorizationServerMetadata ) . toHaveBeenCalledWith (
366
369
new URL ( "https://example.com/" ) ,
367
370
) ;
368
371
} ) ;
@@ -509,7 +512,9 @@ describe("AuthDebugger", () => {
509
512
mockDiscoverOAuthProtectedResourceMetadata . mockResolvedValue (
510
513
mockResourceMetadata ,
511
514
) ;
512
- mockDiscoverOAuthMetadata . mockResolvedValue ( mockOAuthMetadata ) ;
515
+ mockDiscoverAuthorizationServerMetadata . mockResolvedValue (
516
+ mockOAuthMetadata ,
517
+ ) ;
513
518
514
519
await act ( async ( ) => {
515
520
renderAuthDebugger ( {
@@ -563,7 +568,9 @@ describe("AuthDebugger", () => {
563
568
// Mock failed metadata discovery
564
569
mockDiscoverOAuthProtectedResourceMetadata . mockRejectedValue ( mockError ) ;
565
570
// But OAuth metadata should still work with the original URL
566
- mockDiscoverOAuthMetadata . mockResolvedValue ( mockOAuthMetadata ) ;
571
+ mockDiscoverAuthorizationServerMetadata . mockResolvedValue (
572
+ mockOAuthMetadata ,
573
+ ) ;
567
574
568
575
await act ( async ( ) => {
569
576
renderAuthDebugger ( {
@@ -603,7 +610,7 @@ describe("AuthDebugger", () => {
603
610
} ) ;
604
611
605
612
// Verify that regular OAuth metadata discovery was still called
606
- expect ( mockDiscoverOAuthMetadata ) . toHaveBeenCalledWith (
613
+ expect ( mockDiscoverAuthorizationServerMetadata ) . toHaveBeenCalledWith (
607
614
new URL ( "https://example.com/" ) ,
608
615
) ;
609
616
} ) ;
0 commit comments