feat(team-mode): add team session registry with tests

2026-04-28 10:45:41 +09:00
parent 3a93cb776f
commit a91023961c
2 changed files with 122 additions and 0 deletions
@@ -0,0 +1,89 @@
+/// <reference types="bun-types" />
+
+import { afterEach, describe, expect, test } from "bun:test"
+
+import {
+  clearTeamSessionRegistry,
+  lookupTeamSession,
+  registerTeamSession,
+  unregisterTeamSession,
+  unregisterTeamSessionsByTeam,
+} from "./team-session-registry"
+
+describe("team-session-registry", () => {
+  afterEach(() => {
+    clearTeamSessionRegistry()
+  })
+
+  test("registers a session and looks it up by sessionId", () => {
+    // given
+    registerTeamSession("ses_alpha", { teamRunId: "team-1", memberName: "worker-1", role: "member" })
+
+    // when
+    const entry = lookupTeamSession("ses_alpha")
+
+    // then
+    expect(entry).toEqual({ teamRunId: "team-1", memberName: "worker-1", role: "member" })
+  })
+
+  test("returns undefined when the sessionId is not registered", () => {
+    // given - nothing registered
+    // when
+    const entry = lookupTeamSession("ses_missing")
+
+    // then
+    expect(entry).toBeUndefined()
+  })
+
+  test("unregisters a single session by sessionId", () => {
+    // given
+    registerTeamSession("ses_alpha", { teamRunId: "team-1", memberName: "lead", role: "lead" })
+    registerTeamSession("ses_beta", { teamRunId: "team-1", memberName: "worker-1", role: "member" })
+
+    // when
+    unregisterTeamSession("ses_alpha")
+
+    // then
+    expect(lookupTeamSession("ses_alpha")).toBeUndefined()
+    expect(lookupTeamSession("ses_beta")).toEqual({ teamRunId: "team-1", memberName: "worker-1", role: "member" })
+  })
+
+  test("unregisters every session that belongs to the given teamRunId", () => {
+    // given
+    registerTeamSession("ses_alpha", { teamRunId: "team-1", memberName: "lead", role: "lead" })
+    registerTeamSession("ses_beta", { teamRunId: "team-1", memberName: "worker-1", role: "member" })
+    registerTeamSession("ses_gamma", { teamRunId: "team-2", memberName: "solo", role: "member" })
+
+    // when
+    unregisterTeamSessionsByTeam("team-1")
+
+    // then
+    expect(lookupTeamSession("ses_alpha")).toBeUndefined()
+    expect(lookupTeamSession("ses_beta")).toBeUndefined()
+    expect(lookupTeamSession("ses_gamma")).toEqual({ teamRunId: "team-2", memberName: "solo", role: "member" })
+  })
+
+  test("clearTeamSessionRegistry removes every entry", () => {
+    // given
+    registerTeamSession("ses_alpha", { teamRunId: "team-1", memberName: "lead", role: "lead" })
+    registerTeamSession("ses_beta", { teamRunId: "team-2", memberName: "worker", role: "member" })
+
+    // when
+    clearTeamSessionRegistry()
+
+    // then
+    expect(lookupTeamSession("ses_alpha")).toBeUndefined()
+    expect(lookupTeamSession("ses_beta")).toBeUndefined()
+  })
+
+  test("registering the same sessionId twice overwrites the previous entry", () => {
+    // given
+    registerTeamSession("ses_alpha", { teamRunId: "team-1", memberName: "worker-1", role: "member" })
+
+    // when
+    registerTeamSession("ses_alpha", { teamRunId: "team-2", memberName: "promoted-lead", role: "lead" })
+
+    // then
+    expect(lookupTeamSession("ses_alpha")).toEqual({ teamRunId: "team-2", memberName: "promoted-lead", role: "lead" })
+  })
+})
@@ -0,0 +1,33 @@
+export type TeamSessionRole = "lead" | "member"
+
+export type TeamSessionEntry = {
+  teamRunId: string
+  memberName: string
+  role: TeamSessionRole
+}
+
+const registry = new Map<string, TeamSessionEntry>()
+
+export function registerTeamSession(sessionId: string, entry: TeamSessionEntry): void {
+  registry.set(sessionId, entry)
+}
+
+export function lookupTeamSession(sessionId: string): TeamSessionEntry | undefined {
+  return registry.get(sessionId)
+}
+
+export function unregisterTeamSession(sessionId: string): void {
+  registry.delete(sessionId)
+}
+
+export function unregisterTeamSessionsByTeam(teamRunId: string): void {
+  for (const [sessionId, entry] of registry.entries()) {
+    if (entry.teamRunId === teamRunId) {
+      registry.delete(sessionId)
+    }
+  }
+}
+
+export function clearTeamSessionRegistry(): void {
+  registry.clear()
+}