test: speed up memory manager hotspot tests

extensions/memory-core/src/memory/manager.atomic-reindex.test.ts

@@ -23,16 +23,16 @@ describe("memory manager atomic reindex", () => {
   let closeAllMemorySearchManagers: MemoryIndexModule["closeAllMemorySearchManagers"];
 
   beforeAll(async () => {
-    fixtureRoot = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-mem-atomic-"));
-  });
-
-  beforeEach(async () => {
     vi.resetModules();
     const embeddingMocks = await import("./embedding.test-mocks.js");
     embedBatch = embeddingMocks.getEmbedBatchMock();
     resetEmbeddingMocks = embeddingMocks.resetEmbeddingMocks;
     ({ getRequiredMemoryIndexManager } = await import("./test-manager-helpers.js"));
     ({ closeAllMemorySearchManagers } = await import("./index.js"));
+    fixtureRoot = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-mem-atomic-"));
+  });
+
+  beforeEach(async () => {
     vi.stubEnv("OPENCLAW_TEST_MEMORY_UNSAFE_REINDEX", "0");
     resetEmbeddingMocks();
     shouldFail = false;
@@ -60,9 +60,11 @@ describe("memory manager atomic reindex", () => {
 
   afterAll(async () => {
     if (!fixtureRoot) {
+      vi.resetModules();
       return;
     }
     await fs.rm(fixtureRoot, { recursive: true, force: true });
+    vi.resetModules();
   });
 
   it("keeps the prior index when a full reindex fails", async () => {

extensions/memory-core/src/memory/manager.fts-only-reindex.test.ts

@@ -28,12 +28,12 @@ describe("memory manager FTS-only reindex", () => {
   let closeAllMemorySearchManagers: MemoryIndexModule["closeAllMemorySearchManagers"];
 
   beforeAll(async () => {
+    vi.resetModules();
+    ({ getMemorySearchManager, closeAllMemorySearchManagers } = await import("./index.js"));
     fixtureRoot = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-mem-fts-only-"));
   });
 
   beforeEach(async () => {
-    vi.resetModules();
-    ({ getMemorySearchManager, closeAllMemorySearchManagers } = await import("./index.js"));
     workspaceDir = path.join(fixtureRoot, `case-${caseId++}`);
     await fs.mkdir(path.join(workspaceDir, "memory"), { recursive: true });
     await fs.writeFile(path.join(workspaceDir, "MEMORY.md"), "Alpha topic\n\nKeep this note.");
@@ -50,9 +50,11 @@ describe("memory manager FTS-only reindex", () => {
 
   afterAll(async () => {
     if (!fixtureRoot) {
+      vi.resetModules();
       return;
     }
     await fs.rm(fixtureRoot, { recursive: true, force: true });
+    vi.resetModules();
   });
 
   async function createManager(): Promise<MemoryIndexManager> {

extensions/memory-core/src/memory/manager.mistral-provider.test.ts

@@ -2,7 +2,7 @@ import fs from "node:fs/promises";
 import os from "node:os";
 import path from "node:path";
 import type { OpenClawConfig } from "openclaw/plugin-sdk/memory-core-host-engine-foundation";
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
 import { DEFAULT_OLLAMA_EMBEDDING_MODEL } from "./embeddings.js";
 import type {
   EmbeddingProvider,
@@ -77,9 +77,12 @@ describe("memory manager mistral provider wiring", () => {
   let indexPath = "";
   let manager: MemoryIndexManager | null = null;
 
-  beforeEach(async () => {
+  beforeAll(async () => {
     vi.resetModules();
     ({ getMemorySearchManager, closeAllMemorySearchManagers } = await import("./index.js"));
+  });
+
+  beforeEach(async () => {
     vi.clearAllMocks();
     createEmbeddingProviderMock.mockReset();
     workspaceDir = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-memory-mistral-"));
@@ -101,6 +104,10 @@ describe("memory manager mistral provider wiring", () => {
     }
   });
 
+  afterAll(() => {
+    vi.resetModules();
+  });
+
   it("stores mistral client when mistral provider is selected", async () => {
     const mistralRuntime: EmbeddingProviderRuntime = {
       id: "mistral",

extensions/memory-core/src/memory/manager.watcher-config.test.ts

@@ -5,7 +5,7 @@ import type {
   MemorySearchConfig,
   OpenClawConfig,
 } from "openclaw/plugin-sdk/memory-core-host-engine-foundation";
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
 import type { MemoryIndexManager } from "./index.js";
 import { registerBuiltInMemoryEmbeddingProviders } from "./provider-adapters.js";
 
@@ -51,13 +51,16 @@ describe("memory watcher config", () => {
   let workspaceDir = "";
   let extraDir = "";
 
-  beforeEach(async () => {
+  beforeAll(async () => {
     vi.resetModules();
     ({ getMemorySearchManager, closeAllMemorySearchManagers } = await import("./index.js"));
     ({
       clearMemoryEmbeddingProviders: clearRegistry,
       registerMemoryEmbeddingProvider: registerAdapter,
     } = await import("../../../../src/plugins/memory-embedding-providers.js"));
+  });
+
+  beforeEach(async () => {
     vi.clearAllMocks();
     clearRegistry();
     registerBuiltInMemoryEmbeddingProviders({ registerMemoryEmbeddingProvider: registerAdapter });
@@ -78,6 +81,10 @@ describe("memory watcher config", () => {
     }
   });
 
+  afterAll(() => {
+    vi.resetModules();
+  });
+
   async function setupWatcherWorkspace(seedFile: { name: string; contents: string }) {
     workspaceDir = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-memory-watch-"));
     extraDir = path.join(workspaceDir, "extra");