Spaces:
Sleeping
Sleeping
File size: 3,507 Bytes
fb4d8fe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { afterEach, beforeEach, describe, expect, it } from "vitest";
import {
appendAssistantMessageToSessionTranscript,
resolveMirroredTranscriptText,
} from "./transcript.js";
describe("resolveMirroredTranscriptText", () => {
it("prefers media filenames over text", () => {
const result = resolveMirroredTranscriptText({
text: "caption here",
mediaUrls: ["https://example.com/files/report.pdf?sig=123"],
});
expect(result).toBe("report.pdf");
});
it("returns trimmed text when no media", () => {
const result = resolveMirroredTranscriptText({ text: " hello " });
expect(result).toBe("hello");
});
});
describe("appendAssistantMessageToSessionTranscript", () => {
let tempDir: string;
let storePath: string;
let sessionsDir: string;
beforeEach(() => {
tempDir = fs.mkdtempSync(path.join(os.tmpdir(), "transcript-test-"));
sessionsDir = path.join(tempDir, "agents", "main", "sessions");
fs.mkdirSync(sessionsDir, { recursive: true });
storePath = path.join(sessionsDir, "sessions.json");
});
afterEach(() => {
fs.rmSync(tempDir, { recursive: true, force: true });
});
it("returns error for missing sessionKey", async () => {
const result = await appendAssistantMessageToSessionTranscript({
sessionKey: "",
text: "test",
storePath,
});
expect(result.ok).toBe(false);
if (!result.ok) {
expect(result.reason).toBe("missing sessionKey");
}
});
it("returns error for empty text", async () => {
const result = await appendAssistantMessageToSessionTranscript({
sessionKey: "test-session",
text: " ",
storePath,
});
expect(result.ok).toBe(false);
if (!result.ok) {
expect(result.reason).toBe("empty text");
}
});
it("returns error for unknown sessionKey", async () => {
fs.writeFileSync(storePath, JSON.stringify({}), "utf-8");
const result = await appendAssistantMessageToSessionTranscript({
sessionKey: "nonexistent",
text: "test message",
storePath,
});
expect(result.ok).toBe(false);
if (!result.ok) {
expect(result.reason).toContain("unknown sessionKey");
}
});
it("creates transcript file and appends message for valid session", async () => {
const sessionId = "test-session-id";
const sessionKey = "test-session";
const store = {
[sessionKey]: {
sessionId,
chatType: "direct",
channel: "discord",
},
};
fs.writeFileSync(storePath, JSON.stringify(store), "utf-8");
const result = await appendAssistantMessageToSessionTranscript({
sessionKey,
text: "Hello from delivery mirror!",
storePath,
});
expect(result.ok).toBe(true);
if (result.ok) {
expect(fs.existsSync(result.sessionFile)).toBe(true);
const lines = fs.readFileSync(result.sessionFile, "utf-8").trim().split("\n");
expect(lines.length).toBe(2); // header + message
const header = JSON.parse(lines[0]);
expect(header.type).toBe("session");
expect(header.id).toBe(sessionId);
const messageLine = JSON.parse(lines[1]);
expect(messageLine.type).toBe("message");
expect(messageLine.message.role).toBe("assistant");
expect(messageLine.message.content[0].type).toBe("text");
expect(messageLine.message.content[0].text).toBe("Hello from delivery mirror!");
}
});
});
|