diff options
| author | Brian Ray <[email protected]> | 2025-07-22 09:34:56 -0400 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-07-22 13:34:56 +0000 |
| commit | 258c8489092c0970db0f693ddf0956e17051316c (patch) | |
| tree | 05c4b7040a4b011267a566a31d89b0dde66efafb /packages/core/src/tools/mcp-client.test.ts | |
| parent | 138ff738216e1dd29434bd20868328c96b791ac7 (diff) | |
MCP OAuth Part 2 - MCP Client Integration (#4318)
Co-authored-by: Greg Shikhman <[email protected]>
Diffstat (limited to 'packages/core/src/tools/mcp-client.test.ts')
| -rw-r--r-- | packages/core/src/tools/mcp-client.test.ts | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/packages/core/src/tools/mcp-client.test.ts b/packages/core/src/tools/mcp-client.test.ts index 09614442..fbd2a2d4 100644 --- a/packages/core/src/tools/mcp-client.test.ts +++ b/packages/core/src/tools/mcp-client.test.ts @@ -20,6 +20,8 @@ import * as GenAiLib from '@google/genai'; vi.mock('@modelcontextprotocol/sdk/client/stdio.js'); vi.mock('@modelcontextprotocol/sdk/client/index.js'); vi.mock('@google/genai'); +vi.mock('../mcp/oauth-provider.js'); +vi.mock('../mcp/oauth-token-storage.js'); describe('mcp-client', () => { afterEach(() => { @@ -82,7 +84,7 @@ describe('mcp-client', () => { describe('should connect via httpUrl', () => { it('without headers', async () => { - const transport = createTransport( + const transport = await createTransport( 'test-server', { httpUrl: 'http://test-server', @@ -96,7 +98,7 @@ describe('mcp-client', () => { }); it('with headers', async () => { - const transport = createTransport( + const transport = await createTransport( 'test-server', { httpUrl: 'http://test-server', @@ -117,7 +119,7 @@ describe('mcp-client', () => { describe('should connect via url', () => { it('without headers', async () => { - const transport = createTransport( + const transport = await createTransport( 'test-server', { url: 'http://test-server', @@ -130,7 +132,7 @@ describe('mcp-client', () => { }); it('with headers', async () => { - const transport = createTransport( + const transport = await createTransport( 'test-server', { url: 'http://test-server', @@ -149,10 +151,10 @@ describe('mcp-client', () => { }); }); - it('should connect via command', () => { + it('should connect via command', async () => { const mockedTransport = vi.mocked(SdkClientStdioLib.StdioClientTransport); - createTransport( + await createTransport( 'test-server', { command: 'test-command', |