First Release of Claw3D (#11)
Co-authored-by: iamlukethedev <iamlukethedev@users.noreply.github.com>
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
import fs from "node:fs";
|
||||
import os from "node:os";
|
||||
import path from "node:path";
|
||||
|
||||
import { afterEach, describe, expect, it } from "vitest";
|
||||
|
||||
import { restoreAgentStateLocally, trashAgentStateLocally } from "@/lib/agent-state/local";
|
||||
|
||||
const mkTmpStateDir = () => fs.mkdtempSync(path.join(os.tmpdir(), "claw3d-test-"));
|
||||
|
||||
describe("agent state local", () => {
|
||||
const originalStateDir = process.env.OPENCLAW_STATE_DIR;
|
||||
|
||||
afterEach(() => {
|
||||
if (originalStateDir === undefined) delete process.env.OPENCLAW_STATE_DIR;
|
||||
else process.env.OPENCLAW_STATE_DIR = originalStateDir;
|
||||
});
|
||||
|
||||
it("trashes and restores agent workspace + state", () => {
|
||||
const stateDir = mkTmpStateDir();
|
||||
process.env.OPENCLAW_STATE_DIR = stateDir;
|
||||
|
||||
const agentId = "test-agent";
|
||||
const workspace = path.join(stateDir, `workspace-${agentId}`);
|
||||
const agentDir = path.join(stateDir, "agents", agentId);
|
||||
fs.mkdirSync(workspace, { recursive: true });
|
||||
fs.mkdirSync(agentDir, { recursive: true });
|
||||
fs.writeFileSync(path.join(workspace, "hello.txt"), "hi", "utf8");
|
||||
fs.writeFileSync(path.join(agentDir, "state.json"), "{}", "utf8");
|
||||
|
||||
const trashed = trashAgentStateLocally({ agentId });
|
||||
expect(fs.existsSync(workspace)).toBe(false);
|
||||
expect(fs.existsSync(agentDir)).toBe(false);
|
||||
expect(fs.existsSync(trashed.trashDir)).toBe(true);
|
||||
|
||||
const restored = restoreAgentStateLocally({ agentId, trashDir: trashed.trashDir });
|
||||
expect(restored.restored.length).toBeGreaterThan(0);
|
||||
expect(fs.existsSync(workspace)).toBe(true);
|
||||
expect(fs.existsSync(agentDir)).toBe(true);
|
||||
expect(fs.readFileSync(path.join(workspace, "hello.txt"), "utf8")).toBe("hi");
|
||||
});
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user