test(plugin): cover fresh install team mode tools
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
@@ -1,6 +1,9 @@
|
|||||||
/// <reference types="bun-types" />
|
/// <reference types="bun-types" />
|
||||||
|
|
||||||
import { describe, expect, mock, test } from "bun:test"
|
import { afterEach, describe, expect, mock, test } from "bun:test"
|
||||||
|
import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs"
|
||||||
|
import { tmpdir } from "node:os"
|
||||||
|
import { join } from "node:path"
|
||||||
|
|
||||||
import { tool } from "@opencode-ai/plugin"
|
import { tool } from "@opencode-ai/plugin"
|
||||||
|
|
||||||
@@ -16,6 +19,20 @@ const fakeTool = tool({
|
|||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
|
const tempDirs: string[] = []
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
delete process.env.OPENCODE_CONFIG_DIR
|
||||||
|
|
||||||
|
for (const tempDir of tempDirs.splice(0)) {
|
||||||
|
rmSync(tempDir, { recursive: true, force: true })
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
async function importFreshPluginConfigModule(): Promise<typeof import("../plugin-config")> {
|
||||||
|
return import(`../plugin-config?team-mode-fresh-install=${Date.now()}-${Math.random()}`)
|
||||||
|
}
|
||||||
|
|
||||||
function createPluginConfig() {
|
function createPluginConfig() {
|
||||||
return OhMyOpenCodeConfigSchema.parse({
|
return OhMyOpenCodeConfigSchema.parse({
|
||||||
git_master: {
|
git_master: {
|
||||||
@@ -30,6 +47,83 @@ function createPluginConfig() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
describe("team-mode tool registry wiring", () => {
|
describe("team-mode tool registry wiring", () => {
|
||||||
|
test("registers team tools from a fresh-install minimal user config", async () => {
|
||||||
|
// given
|
||||||
|
const rootDir = mkdtempSync(join(tmpdir(), "omo-team-mode-fresh-install-"))
|
||||||
|
tempDirs.push(rootDir)
|
||||||
|
const userConfigDir = join(rootDir, "home", ".config", "opencode")
|
||||||
|
const projectDir = join(rootDir, "project")
|
||||||
|
|
||||||
|
mkdirSync(userConfigDir, { recursive: true })
|
||||||
|
mkdirSync(projectDir, { recursive: true })
|
||||||
|
writeFileSync(
|
||||||
|
join(userConfigDir, "oh-my-openagent.json"),
|
||||||
|
JSON.stringify({ team_mode: { enabled: true } }),
|
||||||
|
)
|
||||||
|
process.env.OPENCODE_CONFIG_DIR = userConfigDir
|
||||||
|
|
||||||
|
const { loadPluginConfig } = await importFreshPluginConfigModule()
|
||||||
|
const pluginConfig = loadPluginConfig(projectDir, {})
|
||||||
|
|
||||||
|
// when
|
||||||
|
const result = createToolRegistry({
|
||||||
|
ctx: { directory: projectDir, client: {} } as Parameters<typeof createToolRegistry>[0]["ctx"],
|
||||||
|
pluginConfig,
|
||||||
|
managers: {
|
||||||
|
backgroundManager: {},
|
||||||
|
tmuxSessionManager: {},
|
||||||
|
skillMcpManager: {},
|
||||||
|
} as Parameters<typeof createToolRegistry>[0]["managers"],
|
||||||
|
skillContext: {
|
||||||
|
mergedSkills: [],
|
||||||
|
availableSkills: [],
|
||||||
|
browserProvider: "playwright",
|
||||||
|
disabledSkills: new Set(),
|
||||||
|
},
|
||||||
|
availableCategories: [],
|
||||||
|
toolFactories: {
|
||||||
|
builtinTools: { bash: fakeTool, read: fakeTool },
|
||||||
|
createBackgroundTools: mock(() => ({})),
|
||||||
|
createCallOmoAgent: mock(() => fakeTool),
|
||||||
|
createLookAt: mock(() => fakeTool),
|
||||||
|
createSkillMcpTool: mock(() => fakeTool),
|
||||||
|
createSkillTool: mock(() => fakeTool),
|
||||||
|
createGrepTools: mock(() => ({})),
|
||||||
|
createGlobTools: mock(() => ({})),
|
||||||
|
createAstGrepTools: mock(() => ({})),
|
||||||
|
createSessionManagerTools: mock(() => ({})),
|
||||||
|
createDelegateTask: mock(() => fakeTool),
|
||||||
|
discoverCommandsSync: mock(() => []),
|
||||||
|
interactive_bash: fakeTool,
|
||||||
|
createTaskCreateTool: mock(() => fakeTool),
|
||||||
|
createTaskGetTool: mock(() => fakeTool),
|
||||||
|
createTaskList: mock(() => fakeTool),
|
||||||
|
createTaskUpdateTool: mock(() => fakeTool),
|
||||||
|
createHashlineEditTool: mock(() => fakeTool),
|
||||||
|
createTeamCreateTool: mock(() => fakeTool),
|
||||||
|
createTeamDeleteTool: mock(() => fakeTool),
|
||||||
|
createTeamShutdownRequestTool: mock(() => fakeTool),
|
||||||
|
createTeamApproveShutdownTool: mock(() => fakeTool),
|
||||||
|
createTeamRejectShutdownTool: mock(() => fakeTool),
|
||||||
|
createTeamSendMessageTool: mock(() => fakeTool),
|
||||||
|
createTeamTaskCreateTool: mock(() => fakeTool),
|
||||||
|
createTeamTaskListTool: mock(() => fakeTool),
|
||||||
|
createTeamTaskUpdateTool: mock(() => fakeTool),
|
||||||
|
createTeamTaskGetTool: mock(() => fakeTool),
|
||||||
|
createTeamStatusTool: mock(() => fakeTool),
|
||||||
|
createTeamListTool: mock(() => fakeTool),
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
// then
|
||||||
|
expect(pluginConfig.team_mode?.enabled).toBe(true)
|
||||||
|
expect(result.filteredTools).toHaveProperty("team_create")
|
||||||
|
expect(result.filteredTools).toHaveProperty("team_send_message")
|
||||||
|
expect(result.filteredTools).toHaveProperty("team_task_create")
|
||||||
|
expect(result.filteredTools).toHaveProperty("team_status")
|
||||||
|
expect(Object.keys(result.filteredTools).filter((toolName) => toolName.startsWith("team_"))).toHaveLength(12)
|
||||||
|
})
|
||||||
|
|
||||||
test("passes ctx.client into every team tool factory", () => {
|
test("passes ctx.client into every team tool factory", () => {
|
||||||
// given
|
// given
|
||||||
const client = {} as OpencodeClient
|
const client = {} as OpencodeClient
|
||||||
|
|||||||
Reference in New Issue
Block a user