78 lines
3.6 KiB
JavaScript
78 lines
3.6 KiB
JavaScript
|
import { afterEach, describe, expect, it, vi } from 'vitest';
|
||
|
|
|
||
|
|
import { createContextObligatoryRuntime } from './runtime.js';
|
||
|
|
|
||
|
|
const json = (body) => new Response(JSON.stringify(body), {
|
||
|
|
status: 200,
|
||
|
|
headers: { 'Content-Type': 'application/json' },
|
||
|
|
});
|
||
|
|
|
||
|
|
describe('context obligatory runtime', () => {
|
||
|
|
afterEach(() => {
|
||
|
|
vi.unstubAllGlobals();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('injects pinned text in chronological order after compaction and records the summary cursor', async () => {
|
||
|
|
const requests = [];
|
||
|
|
let sessionReads = 0;
|
||
|
|
vi.stubGlobal('fetch', vi.fn(async (input, init = {}) => {
|
||
|
|
const url = new URL(typeof input === 'string' ? input : input.url);
|
||
|
|
requests.push({ path: url.pathname, method: init.method ?? 'GET', body: init.body });
|
||
|
|
if (url.pathname === '/session/ses_1' && init.method === 'PATCH') return json({});
|
||
|
|
if (url.pathname === '/session/ses_1') {
|
||
|
|
sessionReads += 1;
|
||
|
|
return json({
|
||
|
|
id: 'ses_1',
|
||
|
|
metadata: { openchamber: { context_obligatory_messages: [
|
||
|
|
{ id: 'msg_2', createdAt: 20, role: 'assistant' },
|
||
|
|
{ id: 'msg_1', createdAt: 10, role: 'user' },
|
||
|
|
] } },
|
||
|
|
});
|
||
|
|
}
|
||
|
|
if (url.pathname === '/session/ses_1/message') return json([
|
||
|
|
{ info: { id: 'msg_agent', role: 'assistant', providerID: 'provider', modelID: 'model', agent: 'build' } },
|
||
|
|
{ info: { id: 'msg_summary', role: 'assistant', summary: true, time: { completed: 30 } } },
|
||
|
|
]);
|
||
|
|
if (url.pathname === '/session/ses_1/message/msg_1') return json({ parts: [{ type: 'text', text: 'First' }] });
|
||
|
|
if (url.pathname === '/session/ses_1/message/msg_2') return json({ parts: [{ type: 'text', text: 'Second' }] });
|
||
|
|
if (url.pathname === '/session/ses_1/prompt_async') return json({});
|
||
|
|
throw new Error(`Unexpected ${url.pathname}`);
|
||
|
|
}));
|
||
|
|
const runtime = createContextObligatoryRuntime({
|
||
|
|
buildOpenCodeUrl: (path) => `http://opencode.test${path}`,
|
||
|
|
getOpenCodeAuthHeaders: () => ({}),
|
||
|
|
});
|
||
|
|
|
||
|
|
await runtime.processPayload({ type: 'session.compacted', properties: { sessionID: 'ses_1' } });
|
||
|
|
|
||
|
|
const prompt = requests.find((request) => request.path.endsWith('/prompt_async'));
|
||
|
|
const payload = JSON.parse(prompt.body);
|
||
|
|
expect(payload).toMatchObject({
|
||
|
|
model: { providerID: 'provider', modelID: 'model' },
|
||
|
|
agent: 'build',
|
||
|
|
parts: [{ type: 'text', synthetic: true }],
|
||
|
|
});
|
||
|
|
expect(payload.parts[0].text.indexOf('First')).toBeLessThan(payload.parts[0].text.indexOf('Second'));
|
||
|
|
expect(payload.parts[0].text).toContain('continuing the pre-compaction work');
|
||
|
|
expect(payload.parts[0].text).toContain('use it silently as background context');
|
||
|
|
expect(payload.parts[0].text).toContain('Only if no tasks or next steps remain');
|
||
|
|
expect(payload.parts[0].text).toContain('no more than one short paragraph');
|
||
|
|
const patch = requests.find((request) => request.method === 'PATCH');
|
||
|
|
expect(JSON.parse(patch.body).metadata.openchamber.context_obligatory_last_compaction_message_id).toBe('msg_summary');
|
||
|
|
expect(sessionReads).toBe(2);
|
||
|
|
runtime.stop();
|
||
|
|
});
|
||
|
|
|
||
|
|
it('ignores ordinary idle events without making requests', async () => {
|
||
|
|
const fetchImpl = vi.fn();
|
||
|
|
vi.stubGlobal('fetch', fetchImpl);
|
||
|
|
const runtime = createContextObligatoryRuntime({
|
||
|
|
buildOpenCodeUrl: (path) => `http://opencode.test${path}`,
|
||
|
|
getOpenCodeAuthHeaders: () => ({}),
|
||
|
|
});
|
||
|
|
await runtime.processPayload({ type: 'session.status', properties: { sessionID: 'ses_1', status: { type: 'idle' } } });
|
||
|
|
expect(fetchImpl).not.toHaveBeenCalled();
|
||
|
|
runtime.stop();
|
||
|
|
});
|
||
|
|
});
|