YeonGyu-Kim
57 lines
1.5 KiB
TypeScript
57 lines
1.5 KiB
TypeScript
const { describe, expect, mock, test, afterAll } = require("bun:test")
|
|
|
|
afterAll(() => { mock.restore() })
|
|
|
|
async function importFreshSessionLastAgentModule() {
|
|
mock.module("../../shared/opencode-message-dir", () => ({
|
|
getMessageDir: () => null,
|
|
}))
|
|
|
|
mock.module("../../shared/opencode-storage-detection", () => ({
|
|
isSqliteBackend: () => true,
|
|
}))
|
|
|
|
const module = await import(`./session-last-agent?test=${Date.now()}-${Math.random()}`)
|
|
mock.restore()
|
|
return module
|
|
}
|
|
|
|
const { getLastAgentFromSession } = await importFreshSessionLastAgentModule()
|
|
|
|
function createMockClient(messages: Array<{ info?: { agent?: string } }>) {
|
|
return {
|
|
session: {
|
|
messages: async () => ({ data: messages }),
|
|
},
|
|
}
|
|
}
|
|
|
|
describe("getLastAgentFromSession sqlite branch", () => {
|
|
test("should skip compaction and return the previous real agent from sqlite messages", async () => {
|
|
// given
|
|
const client = createMockClient([
|
|
{ info: { agent: "atlas" } },
|
|
{ info: { agent: "compaction" } },
|
|
])
|
|
|
|
// when
|
|
const result = await getLastAgentFromSession("ses_sqlite_compaction", client)
|
|
|
|
// then
|
|
expect(result).toBe("atlas")
|
|
})
|
|
|
|
test("should return null when sqlite history contains only compaction", async () => {
|
|
// given
|
|
const client = createMockClient([{ info: { agent: "compaction" } }])
|
|
|
|
// when
|
|
const result = await getLastAgentFromSession("ses_sqlite_only_compaction", client)
|
|
|
|
// then
|
|
expect(result).toBeNull()
|
|
})
|
|
})
|
|
|
|
export {}
|