diff --git a/test/helpers/memory-tool-manager-mock.ts b/extensions/memory-core/src/memory-tool-manager-mock.ts similarity index 84% rename from test/helpers/memory-tool-manager-mock.ts rename to extensions/memory-core/src/memory-tool-manager-mock.ts index 12a5e8aeebb..4397a4d817a 100644 --- a/test/helpers/memory-tool-manager-mock.ts +++ b/extensions/memory-core/src/memory-tool-manager-mock.ts @@ -39,20 +39,7 @@ const readAgentMemoryFileMock = vi.fn( async (params: MemoryReadParams) => await readFileImpl(params), ); -const { memoryIndexModuleId, memoryToolsRuntimeModuleId } = vi.hoisted(() => ({ - memoryIndexModuleId: "../../extensions/memory-core/src/memory/index.js", - memoryToolsRuntimeModuleId: "../../extensions/memory-core/src/tools.runtime.js", -})); - -vi.mock(memoryIndexModuleId, () => ({ - getMemorySearchManager: getMemorySearchManagerMock, -})); - -vi.mock("../../src/memory-host-sdk/host/read-file.js", () => ({ - readAgentMemoryFile: readAgentMemoryFileMock, -})); - -vi.mock(memoryToolsRuntimeModuleId, () => ({ +vi.mock("./tools.runtime.js", () => ({ resolveMemoryBackendConfig: ({ cfg, }: { diff --git a/extensions/memory-core/src/tools.citations.test.ts b/extensions/memory-core/src/tools.citations.test.ts index c2bce446936..cc5461d711d 100644 --- a/extensions/memory-core/src/tools.citations.test.ts +++ b/extensions/memory-core/src/tools.citations.test.ts @@ -14,7 +14,7 @@ import { setMemorySearchImpl, setMemoryWorkspaceDir, type MemoryReadParams, -} from "../../../test/helpers/memory-tool-manager-mock.js"; +} from "./memory-tool-manager-mock.js"; import { createMemoryCoreTestHarness } from "./test-helpers.js"; import { asOpenClawConfig, diff --git a/extensions/memory-core/src/tools.recall-tracking.test.ts b/extensions/memory-core/src/tools.recall-tracking.test.ts index 6c25c1e36c8..7e6485a4607 100644 --- a/extensions/memory-core/src/tools.recall-tracking.test.ts +++ b/extensions/memory-core/src/tools.recall-tracking.test.ts @@ -1,11 +1,11 @@ import type { MemorySearchResult } from "openclaw/plugin-sdk/memory-core-host-runtime-files"; import { beforeEach, describe, expect, it, vi } from "vitest"; +import type { OpenClawConfig } from "../api.js"; import { resetMemoryToolMockState, setMemoryBackend, setMemorySearchImpl, -} from "../../../test/helpers/memory-tool-manager-mock.js"; -import type { OpenClawConfig } from "../api.js"; +} from "./memory-tool-manager-mock.js"; import { createMemorySearchTool } from "./tools.js"; type RecordShortTermRecallsFn = (params: { diff --git a/extensions/memory-core/src/tools.test.ts b/extensions/memory-core/src/tools.test.ts index 158fb845bca..276cd754d83 100644 --- a/extensions/memory-core/src/tools.test.ts +++ b/extensions/memory-core/src/tools.test.ts @@ -1,8 +1,5 @@ import { beforeEach, describe, it } from "vitest"; -import { - resetMemoryToolMockState, - setMemorySearchImpl, -} from "../../../test/helpers/memory-tool-manager-mock.js"; +import { resetMemoryToolMockState, setMemorySearchImpl } from "./memory-tool-manager-mock.js"; import { createMemorySearchToolOrThrow, expectUnavailableMemorySearchDetails,