test: speed up memory provider suites

This commit is contained in:
Peter Steinberger
2026-03-24 21:44:44 +00:00
parent d648aebf4d
commit 3622569853
2 changed files with 12 additions and 7 deletions

View File

@@ -1,13 +1,15 @@
import { afterEach, beforeEach, describe, it, expect, vi } from "vitest";
import { afterEach, beforeAll, beforeEach, describe, it, expect, vi } from "vitest";
import type { OpenClawConfig } from "../config/config.js";
let createOllamaEmbeddingProvider: typeof import("./embeddings-ollama.js").createOllamaEmbeddingProvider;
beforeEach(async () => {
beforeAll(async () => {
({ createOllamaEmbeddingProvider } = await import("./embeddings-ollama.js"));
});
beforeEach(() => {
vi.useRealTimers();
vi.doUnmock("undici");
vi.resetModules();
({ createOllamaEmbeddingProvider } = await import("./embeddings-ollama.js"));
});
afterEach(() => {

View File

@@ -2,7 +2,7 @@ import fs from "node:fs/promises";
import os from "node:os";
import path from "node:path";
import { setTimeout as sleep } from "node:timers/promises";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
import type { OpenClawConfig } from "../config/config.js";
import "./test-runtime-mocks.js";
import type { MemoryIndexManager } from "./index.js";
@@ -42,11 +42,14 @@ let RawMemoryIndexManager: ManagerModule["MemoryIndexManager"];
describe("memory manager cache hydration", () => {
let workspaceDir = "";
beforeEach(async () => {
vi.resetModules();
beforeAll(async () => {
({ getMemorySearchManager, closeAllMemorySearchManagers } = await import("./index.js"));
({ closeAllMemoryIndexManagers, MemoryIndexManager: RawMemoryIndexManager } =
await import("./manager.js"));
});
beforeEach(async () => {
await closeAllMemoryIndexManagers();
vi.clearAllMocks();
workspaceDir = await fs.mkdtemp(path.join(os.tmpdir(), "openclaw-mem-concurrent-"));
await fs.mkdir(path.join(workspaceDir, "memory"), { recursive: true });