test: mock StreamableHTTPClientTransport for faster, deterministic tests

Add mocks for HTTP transport to avoid real network calls during tests.
This addresses reviewer feedback about test reliability:
- Tests are now faster (no network latency)
- Tests are deterministic across environments
- Test intent is clearer (unit testing error handling logic)

The mock throws immediately with a controlled error message,
allowing tests to validate error handling without network dependencies.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
stranger2904
2026-01-14 00:57:38 -05:00
committed by Aleksey Bragin
parent 570b51d07b
commit c9ef648c60
2 changed files with 112 additions and 0 deletions
@@ -3,11 +3,47 @@ import { SkillMcpManager } from "./manager"
import type { SkillMcpClientInfo, SkillMcpServerContext } from "./types"
import type { ClaudeCodeMcpServer } from "../claude-code-mcp-loader/types"
// Mock the MCP SDK transports to avoid network calls
const mockHttpConnect = mock(() => Promise.reject(new Error("Mocked HTTP connection failure")))
const mockHttpClose = mock(() => Promise.resolve())
let lastTransportInstance: { url?: URL; options?: { requestInit?: RequestInit } } = {}
mock.module("@modelcontextprotocol/sdk/client/streamableHttp.js", () => ({
StreamableHTTPClientTransport: class MockStreamableHTTPClientTransport {
constructor(public url: URL, public options?: { requestInit?: RequestInit }) {
lastTransportInstance = { url, options }
}
async start() {
await mockHttpConnect()
}
async close() {
await mockHttpClose()
}
},
}))
describe("SkillMcpManager", () => {
let manager: SkillMcpManager
beforeEach(() => {
manager = new SkillMcpManager()
mockHttpConnect.mockClear()
mockHttpClose.mockClear()
})
afterEach(async () => {
@@ -226,6 +262,30 @@ describe("SkillMcpManager", () => {
/Hints[\s\S]*Verify the URL[\s\S]*authentication headers[\s\S]*MCP over HTTP/
)
})
it("calls mocked transport connect for HTTP connections", async () => {
// #given
const info: SkillMcpClientInfo = {
serverName: "mock-test-server",
skillName: "test-skill",
sessionID: "session-1",
}
const config: ClaudeCodeMcpServer = {
url: "https://example.com/mcp",
}
// #when
try {
await manager.getOrCreateClient(info, config)
} catch {
// Expected to fail
}
// #then - verify mock was called (transport was instantiated)
// The connection attempt happens through the Client.connect() which
// internally calls transport.start()
expect(mockHttpConnect).toHaveBeenCalled()
})
})
describe("stdio connection (backward compatibility)", () => {
@@ -415,6 +475,14 @@ describe("SkillMcpManager", () => {
// Headers are passed through to the transport
await expect(manager.getOrCreateClient(info, config)).rejects.toThrow(
/Failed to connect/
// Verify headers were forwarded to transport
expect(lastTransportInstance.options?.requestInit?.headers).toEqual({
Authorization: "Bearer test-token",
"X-Custom-Header": "custom-value",
})
)
})