66 lines
2 KiB
TypeScript
66 lines
2 KiB
TypeScript
import { existsSync, mkdirSync, rmSync } from "node:fs";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { getModel } from "@jaeswift/jae-ai";
|
|
import { afterEach, beforeEach, describe, expect, it } from "vitest";
|
|
import { createAgentSession } from "../src/core/sdk.js";
|
|
import { SessionManager } from "../src/core/session-manager.js";
|
|
|
|
describe("createAgentSession session manager defaults", () => {
|
|
let tempDir: string;
|
|
let cwd: string;
|
|
let agentDir: string;
|
|
|
|
beforeEach(() => {
|
|
tempDir = join(tmpdir(), `pi-sdk-session-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);
|
|
cwd = join(tempDir, "project");
|
|
agentDir = join(tempDir, "agent");
|
|
mkdirSync(cwd, { recursive: true });
|
|
mkdirSync(agentDir, { recursive: true });
|
|
});
|
|
|
|
afterEach(() => {
|
|
if (tempDir && existsSync(tempDir)) {
|
|
rmSync(tempDir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
it("uses agentDir for the default persisted session path", async () => {
|
|
const model = getModel("anthropic", "claude-sonnet-4-5");
|
|
expect(model).toBeTruthy();
|
|
|
|
const { session } = await createAgentSession({
|
|
cwd,
|
|
agentDir,
|
|
model: model!,
|
|
});
|
|
|
|
const safePath = `--${cwd.replace(/^[/\\]/, "").replace(/[/\\:]/g, "-")}--`;
|
|
const expectedSessionDir = join(agentDir, "sessions", safePath);
|
|
const sessionDir = session.sessionManager.getSessionDir();
|
|
const sessionFile = session.sessionManager.getSessionFile();
|
|
|
|
expect(sessionDir).toBe(expectedSessionDir);
|
|
expect(sessionFile?.startsWith(`${expectedSessionDir}/`)).toBe(true);
|
|
|
|
session.dispose();
|
|
});
|
|
|
|
it("keeps an explicit sessionManager override", async () => {
|
|
const model = getModel("anthropic", "claude-sonnet-4-5");
|
|
expect(model).toBeTruthy();
|
|
|
|
const sessionManager = SessionManager.inMemory(cwd);
|
|
const { session } = await createAgentSession({
|
|
cwd,
|
|
agentDir,
|
|
model: model!,
|
|
sessionManager,
|
|
});
|
|
|
|
expect(session.sessionManager).toBe(sessionManager);
|
|
expect(session.sessionManager.isPersisted()).toBe(false);
|
|
|
|
session.dispose();
|
|
});
|
|
});
|