YeonGyu-Kim
99 lines
2.4 KiB
TypeScript
99 lines
2.4 KiB
TypeScript
import { afterEach, describe, expect, it } from "bun:test"
|
|
|
|
import { _resetForTesting } from "../../features/claude-code-session-state"
|
|
import { clearSessionModel, setSessionModel } from "../../shared/session-model-state"
|
|
import { clearSessionTools } from "../../shared/session-tools-store"
|
|
import {
|
|
resolveLatestSessionPromptConfig,
|
|
resolveSessionPromptConfig,
|
|
} from "./session-prompt-config-resolver"
|
|
|
|
type SessionMessage = {
|
|
info?: {
|
|
agent?: string
|
|
model?: {
|
|
providerID?: string
|
|
modelID?: string
|
|
}
|
|
tools?: Record<string, boolean | "allow" | "deny" | "ask">
|
|
}
|
|
}
|
|
|
|
function createMockContext(messages: SessionMessage[]) {
|
|
return {
|
|
client: {
|
|
session: {
|
|
messages: async () => ({ data: messages }),
|
|
},
|
|
},
|
|
directory: "/tmp/test",
|
|
}
|
|
}
|
|
|
|
describe("session prompt config resolver", () => {
|
|
const sessionID = "ses_compaction_model_validation"
|
|
|
|
afterEach(() => {
|
|
_resetForTesting()
|
|
clearSessionModel(sessionID)
|
|
clearSessionTools()
|
|
})
|
|
|
|
it("prefers the latest non-compaction model over poisoned session state", async () => {
|
|
// given
|
|
setSessionModel(sessionID, {
|
|
providerID: "anthropic",
|
|
modelID: "claude-opus-4-1",
|
|
})
|
|
const ctx = createMockContext([
|
|
{
|
|
info: {
|
|
agent: "atlas",
|
|
model: { providerID: "openai", modelID: "gpt-5" },
|
|
tools: { bash: "allow" },
|
|
},
|
|
},
|
|
{
|
|
info: {
|
|
agent: "compaction",
|
|
model: { providerID: "anthropic", modelID: "claude-opus-4-1" },
|
|
},
|
|
},
|
|
])
|
|
|
|
// when
|
|
const promptConfig = await resolveSessionPromptConfig(ctx, sessionID)
|
|
|
|
// then
|
|
expect(promptConfig).toEqual({
|
|
agent: "atlas",
|
|
model: { providerID: "openai", modelID: "gpt-5" },
|
|
tools: { bash: true },
|
|
})
|
|
})
|
|
|
|
it("omits a compaction model from the latest prompt config", async () => {
|
|
// given
|
|
const ctx = createMockContext([
|
|
{
|
|
info: {
|
|
agent: "atlas",
|
|
model: { providerID: "openai", modelID: "gpt-5" },
|
|
},
|
|
},
|
|
{
|
|
info: {
|
|
agent: "compaction",
|
|
model: { providerID: "anthropic", modelID: "claude-opus-4-1" },
|
|
},
|
|
},
|
|
])
|
|
|
|
// when
|
|
const promptConfig = await resolveLatestSessionPromptConfig(ctx, sessionID)
|
|
|
|
// then
|
|
expect(promptConfig).toEqual({ agent: "compaction" })
|
|
})
|
|
})
|