|
1 | 1 | import { describe, test, expect, vi, afterAll, beforeAll } from 'vitest'; |
2 | | -import { NodeMCPServerStdio } from '../../../src/shims/mcp-server/node'; |
| 2 | +import { |
| 3 | + NodeMCPServerStdio, |
| 4 | + NodeMCPServerSSE, |
| 5 | +} from '../../../src/shims/mcp-server/node'; |
3 | 6 | import { TransportSendOptions } from '@modelcontextprotocol/sdk/shared/transport'; |
4 | 7 | import { JSONRPCMessage } from '@modelcontextprotocol/sdk/types'; |
5 | 8 |
|
@@ -86,3 +89,104 @@ class MockClient { |
86 | 89 | return Promise.resolve(); |
87 | 90 | } |
88 | 91 | } |
| 92 | + |
| 93 | +let capturedFetch: any = undefined; |
| 94 | + |
| 95 | +class MockSSEClientTransport { |
| 96 | + url: URL; |
| 97 | + options: { |
| 98 | + authProvider?: any; |
| 99 | + requestInit?: any; |
| 100 | + eventSourceInit?: any; |
| 101 | + fetch?: any; |
| 102 | + }; |
| 103 | + |
| 104 | + constructor( |
| 105 | + url: URL, |
| 106 | + options: { |
| 107 | + authProvider?: any; |
| 108 | + requestInit?: any; |
| 109 | + eventSourceInit?: any; |
| 110 | + fetch?: any; |
| 111 | + }, |
| 112 | + ) { |
| 113 | + this.url = url; |
| 114 | + this.options = options; |
| 115 | + capturedFetch = options.fetch; |
| 116 | + } |
| 117 | + |
| 118 | + start(): Promise<void> { |
| 119 | + return Promise.resolve(); |
| 120 | + } |
| 121 | + |
| 122 | + send( |
| 123 | + _message: JSONRPCMessage, |
| 124 | + _options?: TransportSendOptions, |
| 125 | + ): Promise<void> { |
| 126 | + return Promise.resolve(); |
| 127 | + } |
| 128 | + |
| 129 | + close(): Promise<void> { |
| 130 | + return Promise.resolve(); |
| 131 | + } |
| 132 | +} |
| 133 | + |
| 134 | +describe('NodeMCPServerSSE', () => { |
| 135 | + beforeAll(() => { |
| 136 | + vi.mock( |
| 137 | + '@modelcontextprotocol/sdk/client/sse.js', |
| 138 | + async (importOriginal) => { |
| 139 | + return { |
| 140 | + ...(await importOriginal()), |
| 141 | + SSEClientTransport: MockSSEClientTransport, |
| 142 | + }; |
| 143 | + }, |
| 144 | + ); |
| 145 | + vi.mock( |
| 146 | + '@modelcontextprotocol/sdk/client/index.js', |
| 147 | + async (importOriginal) => { |
| 148 | + return { |
| 149 | + ...(await importOriginal()), |
| 150 | + Client: MockClient, |
| 151 | + }; |
| 152 | + }, |
| 153 | + ); |
| 154 | + }); |
| 155 | + |
| 156 | + test('should forward custom fetch to SSEClientTransport', async () => { |
| 157 | + const customFetch = vi.fn(async (_input, _init) => { |
| 158 | + return new Response('{}', { status: 200 }); |
| 159 | + }); |
| 160 | + |
| 161 | + const server = new NodeMCPServerSSE({ |
| 162 | + url: 'https://example.com/sse', |
| 163 | + name: 'test-sse-server', |
| 164 | + fetch: customFetch, |
| 165 | + }); |
| 166 | + |
| 167 | + expect(server).toBeDefined(); |
| 168 | + expect(server.name).toBe('test-sse-server'); |
| 169 | + |
| 170 | + await server.connect(); |
| 171 | + |
| 172 | + expect(capturedFetch).toBe(customFetch); |
| 173 | + |
| 174 | + await server.close(); |
| 175 | + }); |
| 176 | + |
| 177 | + test('should accept SSE server without custom fetch', async () => { |
| 178 | + const server = new NodeMCPServerSSE({ |
| 179 | + url: 'https://example.com/sse', |
| 180 | + name: 'test-sse-server-no-fetch', |
| 181 | + }); |
| 182 | + |
| 183 | + expect(server).toBeDefined(); |
| 184 | + await server.connect(); |
| 185 | + await server.close(); |
| 186 | + }); |
| 187 | + |
| 188 | + afterAll(() => { |
| 189 | + vi.clearAllMocks(); |
| 190 | + capturedFetch = undefined; |
| 191 | + }); |
| 192 | +}); |
0 commit comments