can1357
779ca4872b
- Removed Prettier configuration files (.prettierignore and .prettierrc) and migrated formatting to Biome. - Updated Biome configuration from version 2.3.11 to 2.3.12 and changed arrowParentheses rule from 'always' to 'asNeeded'. - Pinned @biomejs/biome dependency to exact version 2.3.12 in package.json and bun.lock. - Applied consistent arrow function formatting across 489 files by removing unnecessary parentheses around single parameters. - Removed blank lines after comment blocks and reorganized imports for consistency across the codebase.
56 lines
1.7 KiB
TypeScript
56 lines
1.7 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import { type CustomEntry, SessionManager } from "@oh-my-pi/pi-coding-agent/session/session-manager";
|
|
|
|
describe("SessionManager.saveCustomEntry", () => {
|
|
it("saves custom entries and includes them in tree traversal", () => {
|
|
const session = SessionManager.inMemory();
|
|
|
|
// Save a message
|
|
const msgId = session.appendMessage({ role: "user", content: "hello", timestamp: 1 });
|
|
|
|
// Save a custom entry
|
|
const customId = session.appendCustomEntry("my_hook", { foo: "bar" });
|
|
|
|
// Save another message
|
|
const msg2Id = session.appendMessage({
|
|
role: "assistant",
|
|
content: [{ type: "text", text: "hi" }],
|
|
api: "anthropic-messages",
|
|
provider: "anthropic",
|
|
model: "test",
|
|
usage: {
|
|
input: 1,
|
|
output: 1,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 2,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
},
|
|
stopReason: "stop",
|
|
timestamp: 2,
|
|
});
|
|
|
|
// Custom entry should be in entries
|
|
const entries = session.getEntries();
|
|
expect(entries).toHaveLength(3);
|
|
|
|
const customEntry = entries.find(e => e.type === "custom") as CustomEntry;
|
|
expect(customEntry).toBeDefined();
|
|
expect(customEntry.customType).toBe("my_hook");
|
|
expect(customEntry.data).toEqual({ foo: "bar" });
|
|
expect(customEntry.id).toBe(customId);
|
|
expect(customEntry.parentId).toBe(msgId);
|
|
|
|
// Tree structure should be correct
|
|
const path = session.getBranch();
|
|
expect(path).toHaveLength(3);
|
|
expect(path[0].id).toBe(msgId);
|
|
expect(path[1].id).toBe(customId);
|
|
expect(path[2].id).toBe(msg2Id);
|
|
|
|
// buildSessionContext should work (custom entries skipped in messages)
|
|
const ctx = session.buildSessionContext();
|
|
expect(ctx.messages).toHaveLength(2); // only message entries
|
|
});
|
|
});
|