diff --git a/src/memory/manager.async-search.test.ts b/src/memory/manager.async-search.test.ts index 2f25db4b23..fdf5a97809 100644 --- a/src/memory/manager.async-search.test.ts +++ b/src/memory/manager.async-search.test.ts @@ -3,8 +3,9 @@ import os from "node:os"; import path from "node:path"; import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; import type { OpenClawConfig } from "../config/config.js"; -import { getMemorySearchManager, type MemoryIndexManager } from "./index.js"; +import type { MemoryIndexManager } from "./index.js"; import { createOpenAIEmbeddingProviderMock } from "./test-embeddings-mock.js"; +import { createMemoryManagerOrThrow } from "./test-manager.js"; const embedBatch = vi.fn(async () => []); const embedQuery = vi.fn(async () => [0.2, 0.2, 0.2]); @@ -56,12 +57,7 @@ describe("memory search async sync", () => { }, } as OpenClawConfig; - const result = await getMemorySearchManager({ cfg, agentId: "main" }); - expect(result.manager).not.toBeNull(); - if (!result.manager) { - throw new Error("manager missing"); - } - manager = result.manager as unknown as MemoryIndexManager; + manager = await createMemoryManagerOrThrow(cfg); const pending = new Promise(() => {}); const syncMock = vi.fn(async () => pending); diff --git a/src/memory/manager.vector-dedupe.test.ts b/src/memory/manager.vector-dedupe.test.ts index dcaf1dffaf..14f2e2f8f3 100644 --- a/src/memory/manager.vector-dedupe.test.ts +++ b/src/memory/manager.vector-dedupe.test.ts @@ -3,8 +3,9 @@ import os from "node:os"; import path from "node:path"; import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; import type { OpenClawConfig } from "../config/config.js"; -import { getMemorySearchManager, type MemoryIndexManager } from "./index.js"; +import type { MemoryIndexManager } from "./index.js"; import { buildFileEntry } from "./internal.js"; +import { createMemoryManagerOrThrow } from "./test-manager.js"; vi.mock("./embeddings.js", () => { return { @@ -57,12 +58,7 @@ describe("memory vector dedupe", () => { }, } as OpenClawConfig; - const result = await getMemorySearchManager({ cfg, agentId: "main" }); - expect(result.manager).not.toBeNull(); - if (!result.manager) { - throw new Error("manager missing"); - } - manager = result.manager as unknown as MemoryIndexManager; + manager = await createMemoryManagerOrThrow(cfg); const db = ( manager as unknown as { diff --git a/src/memory/test-manager.ts b/src/memory/test-manager.ts new file mode 100644 index 0000000000..67d317bd0e --- /dev/null +++ b/src/memory/test-manager.ts @@ -0,0 +1,13 @@ +import type { OpenClawConfig } from "../config/config.js"; +import { getMemorySearchManager, type MemoryIndexManager } from "./index.js"; + +export async function createMemoryManagerOrThrow( + cfg: OpenClawConfig, + agentId = "main", +): Promise { + const result = await getMemorySearchManager({ cfg, agentId }); + if (!result.manager) { + throw new Error("manager missing"); + } + return result.manager as unknown as MemoryIndexManager; +}