test: use sqlite locators in gateway chat history

This commit is contained in:
Peter Steinberger
2026-05-09 05:34:04 +01:00
parent e5d9b0ba66
commit c02b0be46e
2 changed files with 10 additions and 10 deletions

View File

@@ -4,7 +4,7 @@ import path from "node:path";
import { afterAll, beforeAll, describe, expect, test, vi } from "vitest";
import type { GetReplyOptions } from "../auto-reply/get-reply-options.types.js";
import { clearConfigCache } from "../config/config.js";
import { getSessionEntry } from "../config/sessions.js";
import { createSqliteSessionTranscriptLocator, getSessionEntry } from "../config/sessions.js";
import { replaceSqliteSessionTranscriptEvents } from "../config/sessions/transcript-store.sqlite.js";
import { GATEWAY_CLIENT_MODES, GATEWAY_CLIENT_NAMES } from "../utils/message-channel.js";
import { __setMaxChatHistoryMessagesBytesForTest } from "./server-constants.js";
@@ -65,8 +65,8 @@ function createDeferred<T>() {
return { promise, resolve, reject };
}
function resolveMainTranscriptPath(root: string): string {
return path.join(root, "agents", "main", "sessions", "sess-main.jsonl");
function resolveMainTranscriptPath(): string {
return createSqliteSessionTranscriptLocator({ agentId: "main", sessionId: "sess-main" });
}
async function withGatewayChatHarness(
@@ -112,11 +112,11 @@ async function writeGatewayConfig(config: Record<string, unknown>) {
clearConfigCache();
}
async function writeMainSessionTranscript(sessionDir: string, lines: string[]) {
async function writeMainSessionTranscript(_sessionDir: string, lines: string[]) {
replaceSqliteSessionTranscriptEvents({
agentId: "main",
sessionId: "sess-main",
transcriptPath: resolveMainTranscriptPath(sessionDir),
transcriptPath: resolveMainTranscriptPath(),
events: lines.map((line) => JSON.parse(line) as unknown),
});
}

View File

@@ -3,7 +3,7 @@ import os from "node:os";
import path from "node:path";
import { beforeEach, describe, expect, test, vi } from "vitest";
import { WebSocket } from "ws";
import { getSessionEntry } from "../config/sessions.js";
import { createSqliteSessionTranscriptLocator, getSessionEntry } from "../config/sessions.js";
import { replaceSqliteSessionTranscriptEvents } from "../config/sessions/transcript-store.sqlite.js";
import { emitAgentEvent, registerAgentRunContext } from "../infra/agent-events.js";
import { extractFirstTextBlock } from "../shared/chat-message-content.js";
@@ -30,8 +30,8 @@ const CHAT_RESPONSE_TIMEOUT_MS = 10_000;
let ws: WebSocket;
let port: number;
function resolveMainTranscriptPath(root: string): string {
return path.join(root, "agents", "main", "sessions", "sess-main.jsonl");
function resolveMainTranscriptPath(): string {
return createSqliteSessionTranscriptLocator({ agentId: "main", sessionId: "sess-main" });
}
installConnectedControlUiServerSuite((started) => {
@@ -130,11 +130,11 @@ describe("gateway server chat", () => {
}
};
const writeMainSessionTranscript = (dir: string, events: unknown[]): void => {
const writeMainSessionTranscript = (_dir: string, events: unknown[]): void => {
replaceSqliteSessionTranscriptEvents({
agentId: "main",
sessionId: "sess-main",
transcriptPath: resolveMainTranscriptPath(dir),
transcriptPath: resolveMainTranscriptPath(),
events,
});
};