70 lines
2.9 KiB
JavaScript
70 lines
2.9 KiB
JavaScript
|
import test from 'node:test';
|
||
|
|
import assert from 'node:assert/strict';
|
||
|
|
import { DatabaseSync } from 'node:sqlite';
|
||
|
|
import { mkdtempSync, readFileSync, writeFileSync } from 'node:fs';
|
||
|
|
import { tmpdir } from 'node:os';
|
||
|
|
import { join } from 'node:path';
|
||
|
|
|
||
|
|
import { assembleSessionDetail } from '../app/src/shared/session-detail-assembly.mjs';
|
||
|
|
import { persist } from '../packages/core/src/persist.ts';
|
||
|
|
import { parse } from '../packages/core/src/providers/codex.ts';
|
||
|
|
|
||
|
|
const SCHEMA = readFileSync(new URL('../packages/core/src/schema.sql', import.meta.url), 'utf8');
|
||
|
|
const PARENT_ID = '019ed000-0000-7000-8000-000000000101';
|
||
|
|
const REPLAY_ID = '019ed000-0000-7000-8000-000000000102';
|
||
|
|
|
||
|
|
function writeRollout(path, meta, source) {
|
||
|
|
writeFileSync(path, [
|
||
|
|
{ timestamp: '2026-06-15T10:00:00Z', type: 'session_meta', payload: meta },
|
||
|
|
{
|
||
|
|
timestamp: '2026-06-15T10:00:01Z',
|
||
|
|
type: 'response_item',
|
||
|
|
payload: { type: 'custom_tool_call', call_id: 'call_shared', name: 'exec', input: source },
|
||
|
|
},
|
||
|
|
{
|
||
|
|
timestamp: '2026-06-15T10:00:02Z',
|
||
|
|
type: 'response_item',
|
||
|
|
payload: { type: 'custom_tool_call_output', call_id: 'call_shared', output: 'done' },
|
||
|
|
},
|
||
|
|
].map(line => JSON.stringify(line)).join('\n') + '\n');
|
||
|
|
}
|
||
|
|
|
||
|
|
test('a replayed Codex call cannot steal the visible message tool association', () => {
|
||
|
|
const dir = mkdtempSync(join(tmpdir(), 'obelisk-codex-replay-'));
|
||
|
|
const parentPath = join(dir, 'parent.jsonl');
|
||
|
|
const replayPath = join(dir, 'replay.jsonl');
|
||
|
|
writeRollout(parentPath, {
|
||
|
|
id: PARENT_ID,
|
||
|
|
timestamp: '2026-06-15T10:00:00Z',
|
||
|
|
cwd: '/proj',
|
||
|
|
}, 'text("parent")');
|
||
|
|
writeRollout(replayPath, {
|
||
|
|
id: REPLAY_ID,
|
||
|
|
forked_from_id: PARENT_ID,
|
||
|
|
timestamp: '2026-06-15T10:00:00Z',
|
||
|
|
cwd: '/proj',
|
||
|
|
}, 'text("replay")');
|
||
|
|
|
||
|
|
const db = new DatabaseSync(':memory:');
|
||
|
|
db.exec(SCHEMA);
|
||
|
|
for (const path of [parentPath, replayPath]) {
|
||
|
|
const unit = { key: path, sessionId: '', meta: { source: 'codex' } };
|
||
|
|
persist(db, unit, parse(unit, null));
|
||
|
|
}
|
||
|
|
|
||
|
|
const sessionId = `codex:${PARENT_ID}`;
|
||
|
|
const messages = db.prepare(
|
||
|
|
'SELECT * FROM messages WHERE session_id=? AND agent_id IS NULL ORDER BY timestamp, uuid',
|
||
|
|
).all(sessionId);
|
||
|
|
const toolCalls = db.prepare('SELECT * FROM tool_calls WHERE session_id=?').all(sessionId);
|
||
|
|
const toolResults = db.prepare('SELECT * FROM tool_results WHERE session_id=?').all(sessionId);
|
||
|
|
const assembled = assembleSessionDetail({ messages, toolCalls, toolResults, subagents: [], workflows: [] }).messages;
|
||
|
|
|
||
|
|
assert.equal(messages.length, 1, 'the replay remains outside the visible session timeline');
|
||
|
|
assert.equal(toolCalls.length, 2, 'each rollout owns an independently addressable tool call');
|
||
|
|
assert.equal(toolResults.length, 2, 'each rollout owns an independently addressable tool result');
|
||
|
|
assert.deepEqual(assembled[0].tool_calls?.map(call => call.input_json), ['"text(\\"parent\\")"']);
|
||
|
|
assert.equal(assembled[0].tool_calls?.[0].result?.content, 'done');
|
||
|
|
db.close();
|
||
|
|
});
|