YeonGyu-Kim
94 lines
3.5 KiB
TypeScript
94 lines
3.5 KiB
TypeScript
/// <reference types="bun-types" />
|
|
|
|
import { afterEach, describe, expect, test } from "bun:test"
|
|
import { mkdir, rm, writeFile } from "node:fs/promises"
|
|
import { randomUUID } from "node:crypto"
|
|
import { tmpdir } from "node:os"
|
|
import path from "node:path"
|
|
|
|
import { TeamModeConfigSchema } from "../../../config/schema/team-mode"
|
|
import { resolveCallerTeamLead } from "../resolve-caller-team-lead"
|
|
import { loadTeamSpec } from "./loader"
|
|
|
|
async function createTemporaryRoot(): Promise<string> {
|
|
const directoryPath = path.join(tmpdir(), `team-mode-loader-${randomUUID()}`)
|
|
await mkdir(directoryPath, { recursive: true })
|
|
return directoryPath
|
|
}
|
|
|
|
function getFixturePaths(rootDirectory: string, teamName: string) {
|
|
const projectRoot = path.join(rootDirectory, "project")
|
|
const userBaseDir = path.join(rootDirectory, "home", ".omo")
|
|
|
|
return {
|
|
projectRoot,
|
|
userBaseDir,
|
|
userConfigPath: path.join(userBaseDir, "teams", teamName, "config.json"),
|
|
}
|
|
}
|
|
|
|
async function writeJsonFile(filePath: string, value: unknown): Promise<void> {
|
|
await mkdir(path.dirname(filePath), { recursive: true })
|
|
await writeFile(filePath, `${JSON.stringify(value, null, 2)}\n`)
|
|
}
|
|
|
|
describe("loadTeamSpec member name normalization", () => {
|
|
const temporaryDirectories: string[] = []
|
|
|
|
afterEach(async () => {
|
|
await Promise.all(temporaryDirectories.splice(0).map(async (directoryPath) => {
|
|
await rm(directoryPath, { recursive: true, force: true })
|
|
}))
|
|
})
|
|
|
|
test("auto-assigns missing member names for specs on disk", async () => {
|
|
// given
|
|
const rootDirectory = await createTemporaryRoot()
|
|
temporaryDirectories.push(rootDirectory)
|
|
const fixturePaths = getFixturePaths(rootDirectory, "autoname")
|
|
await writeJsonFile(fixturePaths.userConfigPath, {
|
|
name: "autoname",
|
|
lead: { kind: "subagent_type", subagent_type: "sisyphus" },
|
|
members: [
|
|
{ kind: "category", category: "quick", prompt: "Quick scout the workspace structure." },
|
|
{ kind: "category", category: "deep", prompt: "Deep dive the runtime setup." },
|
|
{ kind: "category", category: "deep", prompt: "Deep dive the mailbox implementation." },
|
|
{ kind: "subagent_type", subagent_type: "atlas" },
|
|
],
|
|
})
|
|
|
|
// when
|
|
const teamSpec = await loadTeamSpec("autoname", TeamModeConfigSchema.parse({ base_dir: fixturePaths.userBaseDir }), fixturePaths.projectRoot)
|
|
|
|
// then
|
|
expect(teamSpec.leadAgentId).toBe("lead")
|
|
expect(teamSpec.members.map((member) => member.name)).toEqual(["lead", "quick-1", "deep-1", "deep-2", "atlas-1"])
|
|
})
|
|
|
|
test("injects the caller as lead for preset specs without explicit lead metadata", async () => {
|
|
// given
|
|
const rootDirectory = await createTemporaryRoot()
|
|
temporaryDirectories.push(rootDirectory)
|
|
const fixturePaths = getFixturePaths(rootDirectory, "caller-lead")
|
|
await writeJsonFile(fixturePaths.userConfigPath, {
|
|
name: "caller-lead",
|
|
members: [
|
|
{ kind: "category", category: "quick", prompt: "Quick scout the workspace structure." },
|
|
{ kind: "subagent_type", subagent_type: "atlas" },
|
|
],
|
|
})
|
|
|
|
// when
|
|
const teamSpec = await loadTeamSpec(
|
|
"caller-lead",
|
|
TeamModeConfigSchema.parse({ base_dir: fixturePaths.userBaseDir }),
|
|
fixturePaths.projectRoot,
|
|
{ callerTeamLead: resolveCallerTeamLead("\u200BSisyphus - Ultraworker") },
|
|
)
|
|
|
|
// then
|
|
expect(teamSpec.leadAgentId).toBe("lead")
|
|
expect(teamSpec.members.map((member) => member.name)).toEqual(["lead", "quick-1", "atlas-1"])
|
|
})
|
|
})
|