122 lines
3.7 KiB
TypeScript
122 lines
3.7 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
||
|
||
const { decodeCodexTtyCapture20260311 } = require("./codexCaptureFixture.js");
|
||
const {
|
||
consumeTerminalSyncUpdateFrames,
|
||
createTerminalSyncUpdateState
|
||
} = require("./vtParser.js");
|
||
const {
|
||
getActiveTerminalBuffer,
|
||
rebuildTerminalBufferStateFromReplayText
|
||
} = require("./terminalBufferState.js");
|
||
const { buildTerminalViewportState } = require("./terminalViewportModel.js");
|
||
|
||
function splitTextIntoChunks(text: string, chunkSize: number) {
|
||
const chunks = [];
|
||
for (let index = 0; index < text.length; index += chunkSize) {
|
||
chunks.push(text.slice(index, index + chunkSize));
|
||
}
|
||
return chunks;
|
||
}
|
||
|
||
function serializeViewportLines(replayText: string) {
|
||
const state = rebuildTerminalBufferStateFromReplayText(replayText, {
|
||
bufferCols: 80,
|
||
bufferRows: 24
|
||
});
|
||
const active = getActiveTerminalBuffer(state);
|
||
const viewport = buildTerminalViewportState({
|
||
bufferRows: active.cells,
|
||
cursorRow: active.cursorRow,
|
||
activeBufferName: state.activeBufferName,
|
||
visibleRows: 24,
|
||
lineHeight: 20
|
||
});
|
||
return viewport.renderRows
|
||
.map((line) =>
|
||
(Array.isArray(line) ? line : [])
|
||
.filter((cell) => cell && !cell.continuation)
|
||
.map((cell) => cell.text || " ")
|
||
.join("")
|
||
.replace(/\s+$/, "")
|
||
)
|
||
.filter(Boolean);
|
||
}
|
||
|
||
describe("codexCaptureReplay", () => {
|
||
it("真实 Codex 抓包回放时，会在交互进行中同时保留 Working 行和底部 footer", () => {
|
||
const sample = decodeCodexTtyCapture20260311();
|
||
|
||
expect(sample).toContain("\u001b[?2026h");
|
||
expect(sample).toContain("Working");
|
||
expect(sample).toContain("gpt-5.4 xhigh");
|
||
|
||
let syncState = createTerminalSyncUpdateState();
|
||
let replayText = "";
|
||
let matchedLines: string[] | null = null;
|
||
|
||
splitTextIntoChunks(sample, 97).forEach((chunk) => {
|
||
if (matchedLines) {
|
||
return;
|
||
}
|
||
const result = consumeTerminalSyncUpdateFrames(chunk, syncState);
|
||
syncState = result.state;
|
||
if (!result.text) {
|
||
return;
|
||
}
|
||
replayText += result.text;
|
||
const lines = serializeViewportLines(replayText);
|
||
const hasWorking = lines.some((line) => line.includes("Working"));
|
||
const hasFooter = lines.some(
|
||
(line) => line.includes("gpt-5.4 xhigh") && line.includes("~/remoteconn")
|
||
);
|
||
const hasConversation = lines.some(
|
||
(line) =>
|
||
line.includes("Reply with the single word OK and then stop.") ||
|
||
line.includes("Summarize recent commits")
|
||
);
|
||
if (hasWorking && hasFooter && hasConversation) {
|
||
matchedLines = lines;
|
||
}
|
||
});
|
||
|
||
expect(matchedLines).not.toBeNull();
|
||
expect(matchedLines).toEqual(
|
||
expect.arrayContaining([
|
||
expect.stringContaining("Working"),
|
||
expect.stringContaining("gpt-5.4 xhigh"),
|
||
expect.stringContaining("Summarize recent commits")
|
||
])
|
||
);
|
||
});
|
||
|
||
it("真实 Codex 抓包完整回放后，normal buffer viewport 仍会保留底部 footer", () => {
|
||
let syncState = createTerminalSyncUpdateState();
|
||
let replayText = "";
|
||
|
||
splitTextIntoChunks(decodeCodexTtyCapture20260311(), 97).forEach((chunk) => {
|
||
const result = consumeTerminalSyncUpdateFrames(chunk, syncState);
|
||
syncState = result.state;
|
||
replayText += result.text;
|
||
});
|
||
|
||
expect(syncState).toEqual({
|
||
depth: 0,
|
||
carryText: "",
|
||
bufferedText: ""
|
||
});
|
||
|
||
const lines = serializeViewportLines(replayText);
|
||
|
||
expect(
|
||
lines.some((line) => line.includes("gpt-5.4 xhigh") && line.includes("~/remoteconn"))
|
||
).toBe(true);
|
||
expect(lines).toEqual(
|
||
expect.arrayContaining([
|
||
expect.stringContaining("Conversation interrupted"),
|
||
expect.stringContaining("Summarize recent commits")
|
||
])
|
||
);
|
||
});
|
||
});
|