feat(hooks): add team-mode status injector hook with tests

2026-04-28 10:47:17 +09:00
parent 7f8034c8e2
commit bd768a469b
3 changed files with 224 additions and 0 deletions
@@ -0,0 +1,97 @@
+import { describe, expect, it } from "bun:test"
+
+import { TeamModeConfigSchema } from "../../config/schema/team-mode"
+import { createTeamModeStatusInjector } from "./hook"
+
+function createOutput(sessionID: string): {
+  messages: Array<{
+    info: { role: string; sessionID: string }
+    parts: Array<{ type: string; text?: string; synthetic?: boolean }>
+  }>
+} {
+  return {
+    messages: [
+      {
+        info: {
+          role: "user",
+          sessionID,
+        },
+        parts: [{ type: "text", text: "original message" }],
+      },
+    ],
+  }
+}
+
+describe("createTeamModeStatusInjector", () => {
+  it("injects a one-time team mode enabled message before the latest user message", async () => {
+    // given
+    const hook = createTeamModeStatusInjector(TeamModeConfigSchema.parse({ enabled: true }))
+    const output = createOutput("session-team-mode")
+
+    // when
+    await hook["experimental.chat.messages.transform"]?.(
+      { sessionID: "session-team-mode" },
+      output,
+    )
+
+    // then
+    expect(output.messages).toHaveLength(2)
+    expect(output.messages[0]).toEqual({
+      info: {
+        role: "user",
+        sessionID: "session-team-mode",
+      },
+      parts: [
+        {
+          type: "text",
+          text: expect.stringContaining("Team mode is ENABLED for this session."),
+          synthetic: true,
+        },
+      ],
+    })
+    expect(output.messages[1]?.parts[0]?.text).toBe("original message")
+  })
+
+  it("does not inject again when the team mode status was already added", async () => {
+    // given
+    const hook = createTeamModeStatusInjector(TeamModeConfigSchema.parse({ enabled: true }))
+    const firstOutput = createOutput("session-team-mode")
+    const secondOutput = createOutput("session-team-mode")
+
+    // when
+    await hook["experimental.chat.messages.transform"]?.(
+      { sessionID: "session-team-mode" },
+      firstOutput,
+    )
+    secondOutput.messages = structuredClone(firstOutput.messages)
+    await hook["experimental.chat.messages.transform"]?.(
+      { sessionID: "session-team-mode" },
+      secondOutput,
+    )
+
+    // then
+    expect(firstOutput.messages).toHaveLength(2)
+    expect(secondOutput.messages).toHaveLength(2)
+    expect(
+      secondOutput.messages.filter((message) =>
+        message.parts.some((part) => part.text?.includes("<team_mode_status enabled=\"true\">")),
+      ),
+    ).toHaveLength(1)
+  })
+
+  it("does nothing when team mode is disabled", async () => {
+    // given
+    const hook = createTeamModeStatusInjector(TeamModeConfigSchema.parse({ enabled: false }))
+    const output = createOutput("session-team-mode")
+
+    // when
+    await hook["experimental.chat.messages.transform"]?.(
+      { sessionID: "session-team-mode" },
+      output,
+    )
+
+    // then
+    expect(output.messages).toHaveLength(1)
+    expect(output.messages[0]?.parts[0]?.text).toBe("original message")
+  })
+})