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:
162
packages/mcp-client/src/__tests__/discovery.test.ts
Normal file
162
packages/mcp-client/src/__tests__/discovery.test.ts
Normal file
@@ -0,0 +1,162 @@
|
||||
import { describe, expect, test, mock } from 'bun:test'
|
||||
import { discoverTools, createCachedToolDiscovery } from '../discovery.js'
|
||||
import type { DiscoveryOptions } from '../discovery.js'
|
||||
import type { ConnectedMCPServer } from '../types.js'
|
||||
import type { McpClientDependencies } from '../interfaces.js'
|
||||
|
||||
function createMockDeps(): McpClientDependencies {
|
||||
return {
|
||||
logger: {
|
||||
debug: mock(() => {}),
|
||||
info: mock(() => {}),
|
||||
warn: mock(() => {}),
|
||||
error: mock(() => {}),
|
||||
},
|
||||
httpConfig: {
|
||||
getUserAgent: () => 'test-agent/1.0',
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
describe('discoverTools', () => {
|
||||
test('returns empty array when capabilities.tools is missing', async () => {
|
||||
const result = await discoverTools({
|
||||
serverName: 'test',
|
||||
client: {} as any,
|
||||
capabilities: {},
|
||||
deps: createMockDeps(),
|
||||
})
|
||||
expect(result).toEqual([])
|
||||
})
|
||||
|
||||
test('fetches and transforms tools from server', async () => {
|
||||
const mockClient = {
|
||||
request: mock(() =>
|
||||
Promise.resolve({
|
||||
tools: [
|
||||
{
|
||||
name: 'search',
|
||||
description: 'Search for items',
|
||||
inputSchema: { type: 'object' },
|
||||
annotations: { readOnlyHint: true, title: 'Search Items' },
|
||||
},
|
||||
],
|
||||
}),
|
||||
),
|
||||
}
|
||||
|
||||
const result = await discoverTools({
|
||||
serverName: 'my-server',
|
||||
client: mockClient as any,
|
||||
capabilities: { tools: {} },
|
||||
deps: createMockDeps(),
|
||||
})
|
||||
|
||||
expect(result).toHaveLength(1)
|
||||
const tool = result[0]
|
||||
expect(tool.name).toBe('mcp__my-server__search')
|
||||
expect(tool.mcpInfo).toEqual({ serverName: 'my-server', toolName: 'search' })
|
||||
expect(tool.isMcp).toBe(true)
|
||||
expect(tool.isReadOnly()).toBe(true)
|
||||
expect(tool.userFacingName()).toBe('Search Items')
|
||||
expect(await tool.description()).toBe('Search for items')
|
||||
})
|
||||
|
||||
test('respects skipPrefix option', async () => {
|
||||
const mockClient = {
|
||||
request: mock(() =>
|
||||
Promise.resolve({
|
||||
tools: [{ name: 'search', description: 'Search' }],
|
||||
}),
|
||||
),
|
||||
}
|
||||
|
||||
const result = await discoverTools({
|
||||
serverName: 'my-server',
|
||||
client: mockClient as any,
|
||||
capabilities: { tools: {} },
|
||||
skipPrefix: true,
|
||||
deps: createMockDeps(),
|
||||
})
|
||||
|
||||
expect(result[0].name).toBe('search')
|
||||
})
|
||||
|
||||
test('returns empty array on fetch error', async () => {
|
||||
const mockClient = {
|
||||
request: mock(() => Promise.reject(new Error('Connection lost'))),
|
||||
}
|
||||
const deps = createMockDeps()
|
||||
|
||||
const result = await discoverTools({
|
||||
serverName: 'failing-server',
|
||||
client: mockClient as any,
|
||||
capabilities: { tools: {} },
|
||||
deps,
|
||||
})
|
||||
|
||||
expect(result).toEqual([])
|
||||
expect(deps.logger.warn).toHaveBeenCalled()
|
||||
})
|
||||
|
||||
test('sanitizes tool data', async () => {
|
||||
const mockClient = {
|
||||
request: mock(() =>
|
||||
Promise.resolve({
|
||||
tools: [
|
||||
{
|
||||
name: 'tool\x00with\x07control',
|
||||
description: 'desc',
|
||||
},
|
||||
],
|
||||
}),
|
||||
),
|
||||
}
|
||||
|
||||
const result = await discoverTools({
|
||||
serverName: 'test',
|
||||
client: mockClient as any,
|
||||
capabilities: { tools: {} },
|
||||
deps: createMockDeps(),
|
||||
})
|
||||
|
||||
expect(result[0].name).not.toContain('\x00')
|
||||
})
|
||||
})
|
||||
|
||||
describe('createCachedToolDiscovery', () => {
|
||||
test('caches results by server name', async () => {
|
||||
const deps = createMockDeps()
|
||||
const { discover, cache } = createCachedToolDiscovery(deps)
|
||||
|
||||
const mockConn = {
|
||||
type: 'connected' as const,
|
||||
name: 'cached-server',
|
||||
client: {
|
||||
request: mock(() =>
|
||||
Promise.resolve({
|
||||
tools: [{ name: 'tool1', description: 'Tool 1' }],
|
||||
}),
|
||||
),
|
||||
},
|
||||
capabilities: { tools: {} },
|
||||
} as unknown as ConnectedMCPServer
|
||||
|
||||
// First call — should fetch
|
||||
const result1 = await discover(mockConn)
|
||||
expect(result1).toHaveLength(1)
|
||||
|
||||
// Second call — should use cache
|
||||
const result2 = await discover(mockConn)
|
||||
expect(result2).toHaveLength(1)
|
||||
|
||||
// Request was called only once
|
||||
expect(mockConn.client.request).toHaveBeenCalledTimes(1)
|
||||
|
||||
// Cache delete works
|
||||
cache.delete('cached-server')
|
||||
const result3 = await discover(mockConn)
|
||||
expect(result3).toHaveLength(1)
|
||||
expect(mockConn.client.request).toHaveBeenCalledTimes(2)
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user