YeonGyu-Kim
98 lines
2.8 KiB
TypeScript
98 lines
2.8 KiB
TypeScript
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")
|
|
})
|
|
})
|