From 21b782de93e8c58dddca3f85837a06e41654ef87 Mon Sep 17 00:00:00 2001
From: YeonGyu-Kim <code.yeon.gyu@gmail.com>
Date: Mon, 18 May 2026 13:09:34 +0900
Subject: [PATCH] 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>
---
 src/plugin/tool-registry.team-mode.test.ts | 96 +++++++++++++++++++++-
 1 file changed, 95 insertions(+), 1 deletion(-)

diff --git a/src/plugin/tool-registry.team-mode.test.ts b/src/plugin/tool-registry.team-mode.test.ts
index d858dee45..787c7012a 100644
--- a/src/plugin/tool-registry.team-mode.test.ts
+++ b/src/plugin/tool-registry.team-mode.test.ts
@@ -1,6 +1,9 @@
 /// <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"
 
@@ -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() {
   return OhMyOpenCodeConfigSchema.parse({
     git_master: {
@@ -30,6 +47,83 @@ function createPluginConfig() {
 }
 
 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", () => {
     // given
     const client = {} as OpencodeClient