clone/opencode
1
0
Fork 0
mirror of https://github.com/anomalyco/opencode.git synced 2026-05-16 17:42:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify compaction test helpers (#26742)

Browse Source
This commit is contained in:
Kit Langton
2026-05-10 20:03:11 -04:00
committed by GitHub
parent 5ef72e1101
commit 128d10d9e9
1 changed files with 8 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
packages/opencode/test/session/compaction.test.ts
View File

@@ -226,12 +226,6 @@ async function summaryAssistant(sessionID: SessionID, parentID: MessageID, root:
return msg
}
async function lastCompactionPart(sessionID: SessionID) {
return (await svc.messages({ sessionID }))
.at(-2)
?.parts.find((item): item is MessageV2.CompactionPart => item.type === "compaction")
}
function createCompactionMarker(sessionID: SessionID) {
return SessionNs.Service.use((ssn) =>
Effect.gen(function* () {
@@ -258,14 +252,6 @@ async function createCompactionMarkerAsync(sessionID: SessionID) {
return run(createCompactionMarker(sessionID))
}
function readLastCompactionPart(sessionID: SessionID) {
return SessionNs.Service.use((ssn) => ssn.messages({ sessionID })).pipe(
Effect.map((messages) =>
messages.at(-2)?.parts.find((item): item is MessageV2.CompactionPart => item.type === "compaction"),
),
)
}
function fake(
input: Parameters<SessionProcessorModule.SessionProcessor.Interface["create"]>[0],
result: "continue" | "compact",
@@ -379,6 +365,10 @@ function readCompactionPart(sessionID: SessionID) {
)
}
async function lastCompactionPart(sessionID: SessionID) {
return run(readCompactionPart(sessionID))
}
function llm() {
const queue: Array<
Stream.Stream<LLM.Event, unknown> | ((input: LLM.StreamInput) => Stream.Stream<LLM.Event, unknown>)
@@ -1057,7 +1047,7 @@ describe("session.compaction.process", () => {
yield* createUserMessage(session.id, "first")
const keep = yield* createUserMessage(session.id, "second")
yield* createUserMessage(session.id, "third")
yield* SessionCompaction.use.create({ sessionID: session.id, agent: "build", model: ref, auto: false })
yield* createSummaryCompaction(session.id)
const msgs = yield* ssn.messages({ sessionID: session.id })
const parent = msgs.at(-1)?.info.id
@@ -1069,7 +1059,7 @@ describe("session.compaction.process", () => {
auto: false,
})
const part = yield* readLastCompactionPart(session.id)
const part = yield* readCompactionPart(session.id)
expect(part?.type).toBe("compaction")
expect(part?.tail_start_id).toBe(keep.id)
}).pipe(Effect.provide(compactionProcessLayer({ config: cfg({ tail_turns: 2, preserve_recent_tokens: 10_000 }) }))),
@@ -1083,7 +1073,7 @@ describe("session.compaction.process", () => {
yield* createUserMessage(session.id, "first")
yield* createUserMessage(session.id, "x".repeat(2_000))
const keep = yield* createUserMessage(session.id, "tiny")
yield* SessionCompaction.use.create({ sessionID: session.id, agent: "build", model: ref, auto: false })
yield* createSummaryCompaction(session.id)
const msgs = yield* ssn.messages({ sessionID: session.id })
const parent = msgs.at(-1)?.info.id
@@ -1095,7 +1085,7 @@ describe("session.compaction.process", () => {
auto: false,
})
const part = yield* readLastCompactionPart(session.id)
const part = yield* readCompactionPart(session.id)
expect(part?.type).toBe("compaction")
expect(part?.tail_start_id).toBe(keep.id)
}).pipe(Effect.provide(compactionProcessLayer({ config: cfg({ tail_turns: 2, preserve_recent_tokens: 100 }) }))),