vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-03-26 09:21:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ci): reset deep test runtime state

Browse Source
This commit is contained in:
Vincent Koc
2026-03-19 14:23:17 -07:00
parent ae02f40144
commit 83a267e2f3
6 changed files with 39 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/memory/index.test.ts
View File

@@ -3,13 +3,14 @@ import { mkdirSync, rmSync } from "node:fs";
import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { afterAll, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
import "./test-runtime-mocks.js";
import type { MemoryIndexManager } from "./index.js";
type MemoryIndexModule = typeof import("./index.js");
let getMemorySearchManager: MemoryIndexModule["getMemorySearchManager"];
let closeAllMemorySearchManagers: MemoryIndexModule["closeAllMemorySearchManagers"];
let embedBatchCalls = 0;
let embedBatchInputCalls = 0;
@@ -125,14 +126,12 @@ describe("memory index", () => {
}),
].join("\n");
// Perf: keep managers open across tests, but only reset the one a test uses.
const managersByCacheKey = new Map<string, MemoryIndexManager>();
const managersForCleanup = new Set<MemoryIndexManager>();
beforeAll(async () => {
vi.resetModules();
await import("./test-runtime-mocks.js");
({ getMemorySearchManager } = await import("./index.js"));
({ getMemorySearchManager, closeAllMemorySearchManagers } = await import("./index.js"));
fixtureRoot = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-mem-fixtures-"));
workspaceDir = path.join(fixtureRoot, "workspace");
memoryDir = path.join(workspaceDir, "memory");
@@ -158,6 +157,11 @@ describe("memory index", () => {
await fs.rm(fixtureRoot, { recursive: true, force: true });
});
afterEach(async () => {
await closeAllMemorySearchManagers();
managersForCleanup.clear();
});
beforeEach(async () => {
// Perf: most suites don't need atomic swap behavior for full reindexes.
// Keep atomic reindex tests on the safe path.
@@ -166,7 +170,6 @@ describe("memory index", () => {
embedBatchInputCalls = 0;
providerCalls = [];
// Keep the workspace stable to allow manager reuse across tests.
mkdirSync(memoryDir, { recursive: true });
// Clean additional paths that may have been created by earlier cases.
@@ -243,30 +246,9 @@ describe("memory index", () => {
return result.manager as MemoryIndexManager;
}
function getManagerCacheKey(cfg: TestCfg): string {
const memorySearch = cfg.agents?.defaults?.memorySearch;
const storePath = memorySearch?.store?.path;
if (!storePath) {
throw new Error("store path missing");
}
return JSON.stringify({
workspaceDir,
storePath,
memorySearch,
});
}
async function getPersistentManager(cfg: TestCfg): Promise<MemoryIndexManager> {
const cacheKey = getManagerCacheKey(cfg);
const cached = managersByCacheKey.get(cacheKey);
if (cached) {
resetManagerForTest(cached);
return cached;
}
const result = await getMemorySearchManager({ cfg, agentId: "main" });
const manager = requireManager(result);
managersByCacheKey.set(cacheKey, manager);
managersForCleanup.add(manager);
resetManagerForTest(manager);
return manager;