File size: 2,788 Bytes
f778c12 | 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 | import fs from "node:fs/promises";
import path from "node:path";
import { expectDefined } from "@openclaw/normalization-core";
import { afterEach, beforeEach, describe, expect, it } from "vitest";
import { useAutoCleanupTempDirTracker } from "../../../../test/helpers/temp-dir.js";
import type { OpenClawConfig } from "../../../config/config.js";
import { replaceTranscriptEvents } from "../../../config/sessions/session-accessor.js";
import { createInternalHookEvent } from "../../internal-hooks.js";
import handler, { flushSessionMemoryWritesForTest } from "./handler.js";
const tempDirs = useAutoCleanupTempDirTracker(afterEach);
describe("session-memory automatic reset", () => {
let tempDir = "";
beforeEach(() => {
tempDir = tempDirs.make("openclaw-session-memory-auto-");
});
afterEach(async () => {
await flushSessionMemoryWritesForTest();
});
it.each(["daily", "idle"] as const)(
"creates memory from the ended session on %s reset",
async (reason) => {
const sessionKey = "agent:main:main";
const sessionId = `${reason}-session`;
const storePath = path.join(tempDir, "sessions.json");
const cfg = {
agents: { defaults: { workspace: tempDir } },
session: { store: storePath },
} satisfies OpenClawConfig;
await replaceTranscriptEvents({ agentId: "main", sessionId, sessionKey, storePath }, [
{
type: "message",
id: `${reason}-user`,
parentId: null,
message: {
role: "user",
content: `Remember the ${reason} rollover`,
__openclaw: { senderIsOwner: true },
},
},
{
type: "message",
id: `${reason}-assistant`,
parentId: `${reason}-user`,
message: { role: "assistant", content: "Captured automatically" },
},
]);
const event = createInternalHookEvent("session", "auto-reset", sessionKey, {
cfg,
agentId: "main",
workspaceDir: tempDir,
storePath,
sessionEntry: { sessionId },
reason,
});
const completed = handler(event);
expect(completed).toBeInstanceOf(Promise);
await completed;
const memoryDir = path.join(tempDir, "memory");
const files = await fs.readdir(memoryDir);
const memoryContent = await fs.readFile(
path.join(memoryDir, expectDefined(files[0], "files[0] test invariant")),
"utf8",
);
expect(files).toHaveLength(1);
expect(memoryContent).toContain(`- **Reason**: ${reason}`);
expect(memoryContent).toContain(`user: ${JSON.stringify(`Remember the ${reason} rollover`)}`);
expect(memoryContent).toContain(`assistant: ${JSON.stringify("Captured automatically")}`);
},
);
});
|