fix(team-mode): cleanup team runs on shutdown

src/create-managers.test.ts

@@ -8,11 +8,23 @@ import { createManagers } from "./create-managers"
 import * as openclawRuntimeDispatch from "./openclaw/runtime-dispatch"
 import { createModelCacheState } from "./plugin-state"
 
+type CleanupRegistration = {
+  shutdown: () => void | Promise<void>
+}
+
+type CleanupSessionTeamRunsFn = typeof import("./features/team-mode/team-runtime/session-cleanup").cleanupSessionTeamRuns
+
 const markServerRunningInProcess = mock(() => {})
 let backgroundManagerOptions: {
   onSubagentSessionCreated?: (event: { sessionID: string; parentID: string; title: string }) => Promise<void>
 } | null = null
 const trackedPaneBySession = new Map<string, string>()
+const registeredCleanupManagers: CleanupRegistration[] = []
+const cleanupSessionTeamRunsMock = mock(async () => ({
+  cleanedTeamRunIds: [],
+  removedLayoutTeamRunIds: [],
+  errors: [],
+}))
 
 class MockBackgroundManager {
   constructor(config: {
@@ -51,7 +63,9 @@ function initTaskToastManager(): ReturnType<typeof import("./features/task-toast
   return {} as ReturnType<typeof import("./features/task-toast-manager").initTaskToastManager>
 }
 
-function registerManagerForCleanup(): void {}
+function registerManagerForCleanup(manager: CleanupRegistration): void {
+  registeredCleanupManagers.push(manager)
+}
 
 function createDeps(): NonNullable<Parameters<typeof createManagers>[0]["deps"]> {
   return {
@@ -60,6 +74,7 @@ function createDeps(): NonNullable<Parameters<typeof createManagers>[0]["deps"]>
     TmuxSessionManagerClass: MockTmuxSessionManager as typeof import("./features/tmux-subagent").TmuxSessionManager,
     initTaskToastManagerFn: initTaskToastManager,
     registerManagerForCleanupFn: registerManagerForCleanup,
+    cleanupSessionTeamRunsFn: cleanupSessionTeamRunsMock as CleanupSessionTeamRunsFn,
     createConfigHandlerFn: createConfigHandler,
     markServerRunningInProcessFn: markServerRunningInProcess,
   }
@@ -122,6 +137,8 @@ describe("createManagers", () => {
     dispatchOpenClawEvent.mockReset()
     backgroundManagerOptions = null
     trackedPaneBySession.clear()
+    registeredCleanupManagers.length = 0
+    cleanupSessionTeamRunsMock.mockClear()
   })
 
   afterEach(() => {
@@ -193,4 +210,32 @@ describe("createManagers", () => {
       },
     })
   })
+
+  it("#given team mode is enabled #when process cleanup runs #then session team runs are cleaned with tmux visualization dependencies", async () => {
+    const args = {
+      ctx: createContext("/tmp/project"),
+      pluginConfig: OhMyOpenCodeConfigSchema.parse({
+        team_mode: {
+          enabled: true,
+          tmux_visualization: true,
+        },
+      }),
+      tmuxConfig: createTmuxConfig(true),
+      modelCacheState: createModelCacheState(),
+      backgroundNotificationHookEnabled: false,
+      deps: createDeps(),
+    }
+
+    createManagers(args)
+
+    await registeredCleanupManagers[0]?.shutdown()
+
+    expect(cleanupSessionTeamRunsMock).toHaveBeenCalledTimes(1)
+    const cleanupArgs = cleanupSessionTeamRunsMock.mock.calls[0]?.[0]
+    expect(cleanupArgs).toMatchObject({
+      config: args.pluginConfig.team_mode,
+    })
+    expect(cleanupArgs?.tmuxMgr).toBeInstanceOf(MockTmuxSessionManager)
+    expect(cleanupArgs?.bgMgr).toBeInstanceOf(MockBackgroundManager)
+  })
 })