YeonGyu-Kim
ef09880e26
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
85 lines
2.6 KiB
TypeScript
85 lines
2.6 KiB
TypeScript
import { afterEach, describe, expect, mock, test } from "bun:test"
|
|
|
|
afterEach(() => {
|
|
mock.restore()
|
|
})
|
|
|
|
function mockLocalMcps(): void {
|
|
mock.module("../lsp", () => ({
|
|
createLspMcpConfig: () => ({ type: "local", command: ["node", "dist/cli.js", "mcp"], enabled: true }),
|
|
}))
|
|
mock.module("../ast-grep", () => ({
|
|
createAstGrepMcpConfig: () => ({ type: "local", command: ["node", "ast-grep-mcp", "mcp"], enabled: true }),
|
|
}))
|
|
}
|
|
|
|
describe("createBuiltinMcps", () => {
|
|
test("should return all MCPs when disabled_mcps is empty", () => {
|
|
// given
|
|
mockLocalMcps()
|
|
const { createBuiltinMcps } = require("../index") as typeof import("../index")
|
|
const disabledMcps: string[] = []
|
|
|
|
// when
|
|
const result = createBuiltinMcps(disabledMcps)
|
|
|
|
// then
|
|
expect(Object.keys(result).length).toBeGreaterThan(0)
|
|
expect(result.websearch).toBeDefined()
|
|
expect(result.context7).toBeDefined()
|
|
expect(result.grep_app).toBeDefined()
|
|
expect(result.lsp).toBeDefined()
|
|
expect(result.ast_grep).toBeDefined()
|
|
})
|
|
|
|
test("should filter out disabled MCPs", () => {
|
|
// given
|
|
mockLocalMcps()
|
|
const { createBuiltinMcps } = require("../index") as typeof import("../index")
|
|
const disabledMcps = ["websearch"]
|
|
|
|
// when
|
|
const result = createBuiltinMcps(disabledMcps)
|
|
|
|
// then
|
|
expect(result.websearch).toBeUndefined()
|
|
expect(result.context7).toBeDefined()
|
|
expect(result.grep_app).toBeDefined()
|
|
expect(result.lsp).toBeDefined()
|
|
expect(result.ast_grep).toBeDefined()
|
|
})
|
|
|
|
test("should keep lsp when it uses a bootstrap command", () => {
|
|
// given
|
|
mock.module("../lsp", () => ({
|
|
createLspMcpConfig: () => ({ type: "local", command: ["node", "-e", "bootstrap", "/repo"], enabled: true }),
|
|
}))
|
|
const { createBuiltinMcps } = require("../index") as typeof import("../index")
|
|
|
|
// when
|
|
const result = createBuiltinMcps([])
|
|
|
|
// then
|
|
expect(result.lsp).toBeDefined()
|
|
})
|
|
|
|
test("should return empty array when all MCPs are disabled", () => {
|
|
// given
|
|
mockLocalMcps()
|
|
const { createBuiltinMcps } = require("../index") as typeof import("../index")
|
|
const disabledMcps = ["websearch", "context7", "grep_app", "lsp", "ast_grep"]
|
|
|
|
// when
|
|
const result = createBuiltinMcps(disabledMcps)
|
|
|
|
// then
|
|
const remainingMcpNames = Object.keys(result)
|
|
expect(remainingMcpNames).not.toContain("websearch")
|
|
expect(remainingMcpNames).not.toContain("context7")
|
|
expect(remainingMcpNames).not.toContain("grep_app")
|
|
expect(remainingMcpNames).not.toContain("lsp")
|
|
expect(remainingMcpNames).not.toContain("ast_grep")
|
|
expect(remainingMcpNames).toEqual([])
|
|
})
|
|
})
|