mirror of
https://github.com/claude-code-best/claude-code.git
synced 2026-06-23 00:35:51 +00:00
feat: 工具层及 mcp 大重构 (#252)
* feat: 第一版大重构 * fix: 修复类型问题 * chore: 更新版本到 1.3.2 * Add brave as alternative WebSearchTool * fix: 修正顺序 * fix: 修复对穷鬼模式的 auto dream 和 session memory 越过 * feat: 穷鬼模式去除 session-summary * feat: 创建 builtin-tools 包,搬运所有工具实现 将 src/tools/ 下的全部 60 个工具目录迁移至 packages/builtin-tools/src/tools/, 内部导入路径已更新为 src/ alias 模式。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * refactor: 更新 src/ 中所有工具引用至 builtin-tools 包,删除 src/tools/ - src/tools.ts 及 178 个 src/ 文件的 import 路径从 ./tools/ 改为 builtin-tools/tools/ - 删除 src/tools/ 整个目录(已迁移至 packages/builtin-tools/) Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * chore: 添加 builtin-tools 路径别名至 tsconfig,更新 bun.lock - tsconfig.json 新增 builtin-tools/* 和 builtin-tools 路径映射 - 新增 packages/builtin-tools/src 至 include Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * refactor: 为 builtin-tools、mcp-client、agent-tools 添加 @claude-code-best 作用域前缀 所有包名及 import 路径统一添加 @claude-code-best/ 前缀: - builtin-tools → @claude-code-best/builtin-tools - mcp-client → @claude-code-best/mcp-client - agent-tools → @claude-code-best/agent-tools Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * fix: 修复 node 环境没有 bun 的问题 --------- Co-authored-by: Eric-Guo <eric.guocz@gmail.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
113
packages/mcp-client/src/__tests__/manager.test.ts
Normal file
113
packages/mcp-client/src/__tests__/manager.test.ts
Normal file
@@ -0,0 +1,113 @@
|
||||
import { describe, expect, test, mock } from 'bun:test'
|
||||
import { createMcpManager } from '../manager.js'
|
||||
import type { McpManager } from '../manager.js'
|
||||
import type { McpClientDependencies } from '../interfaces.js'
|
||||
import type { ScopedMcpServerConfig, MCPServerConnection, ConnectedMCPServer } from '../types.js'
|
||||
import type { Client } from '@modelcontextprotocol/sdk/client/index.js'
|
||||
|
||||
function createMockDeps(): McpClientDependencies {
|
||||
return {
|
||||
logger: {
|
||||
debug: mock(() => {}),
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
},
|
||||
httpConfig: {
|
||||
getUserAgent: () => 'test-agent/1.0',
|
||||
getSessionId: () => 'test-session',
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
describe('createMcpManager', () => {
|
||||
test('creates a manager instance', () => {
|
||||
const manager = createMcpManager(createMockDeps())
|
||||
expect(manager).toBeDefined()
|
||||
expect(manager.getConnections).toBeTypeOf('function')
|
||||
expect(manager.connect).toBeTypeOf('function')
|
||||
expect(manager.disconnect).toBeTypeOf('function')
|
||||
expect(manager.getTools).toBeTypeOf('function')
|
||||
expect(manager.getAllTools).toBeTypeOf('function')
|
||||
expect(manager.callTool).toBeTypeOf('function')
|
||||
expect(manager.on).toBeTypeOf('function')
|
||||
expect(manager.off).toBeTypeOf('function')
|
||||
})
|
||||
|
||||
test('connect throws if connectFn not set', async () => {
|
||||
const manager = createMcpManager(createMockDeps())
|
||||
await expect(manager.connect('test', { command: 'npx', args: [] }))
|
||||
.rejects.toThrow('connectFn not set')
|
||||
})
|
||||
|
||||
test('connect calls connectFn and emits connected event', async () => {
|
||||
const manager = createMcpManager(createMockDeps()) as any
|
||||
let connectedEvent: string | null = null
|
||||
manager.on('connected', (name: string) => { connectedEvent = name })
|
||||
|
||||
const mockConnection: ConnectedMCPServer = {
|
||||
type: 'connected',
|
||||
name: 'test-server',
|
||||
client: {
|
||||
request: mock(() => Promise.resolve({ tools: [] })),
|
||||
onclose: null,
|
||||
} as unknown as Client,
|
||||
capabilities: {},
|
||||
config: { command: 'npx', args: [], scope: 'dynamic' } as ScopedMcpServerConfig,
|
||||
cleanup: mock(() => Promise.resolve()),
|
||||
}
|
||||
|
||||
manager.setConnectFn(async (name: string, config: ScopedMcpServerConfig) => {
|
||||
expect(name).toBe('test-server')
|
||||
expect(config.scope).toBe('dynamic')
|
||||
return mockConnection
|
||||
})
|
||||
|
||||
const result = await manager.connect('test-server', { command: 'npx', args: [] })
|
||||
expect(result.type).toBe('connected')
|
||||
expect(connectedEvent).toBe('test-server')
|
||||
})
|
||||
|
||||
test('disconnect calls cleanup and emits disconnected', async () => {
|
||||
const manager = createMcpManager(createMockDeps()) as any
|
||||
let disconnected = false
|
||||
manager.on('disconnected', () => { disconnected = true })
|
||||
|
||||
const mockCleanup = mock(() => Promise.resolve())
|
||||
const mockConnection: ConnectedMCPServer = {
|
||||
type: 'connected',
|
||||
name: 'test-server',
|
||||
client: { request: mock(() => Promise.resolve({ tools: [] })) } as unknown as Client,
|
||||
capabilities: {},
|
||||
config: { command: 'npx', args: [], scope: 'dynamic' } as ScopedMcpServerConfig,
|
||||
cleanup: mockCleanup,
|
||||
}
|
||||
|
||||
manager.setConnectFn(async () => mockConnection)
|
||||
await manager.connect('test-server', { command: 'npx', args: [] })
|
||||
|
||||
await manager.disconnect('test-server')
|
||||
expect(mockCleanup).toHaveBeenCalled()
|
||||
expect(disconnected).toBe(true)
|
||||
expect(manager.getConnections().size).toBe(0)
|
||||
})
|
||||
|
||||
test('on/off event handling', () => {
|
||||
const manager = createMcpManager(createMockDeps()) as any
|
||||
const handler = mock(() => {})
|
||||
manager.on('error', handler)
|
||||
manager.off('error', handler)
|
||||
// No crash — just verifying it works
|
||||
expect(true).toBe(true)
|
||||
})
|
||||
|
||||
test('getTools returns empty array for unknown server', () => {
|
||||
const manager = createMcpManager(createMockDeps())
|
||||
expect(manager.getTools('unknown')).toEqual([])
|
||||
})
|
||||
|
||||
test('getAllTools returns empty array initially', () => {
|
||||
const manager = createMcpManager(createMockDeps())
|
||||
expect(manager.getAllTools()).toEqual([])
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user