chore: Fix types in tests 17/N.

This commit is contained in:
cpojer
2026-02-17 11:59:41 +09:00
parent 7bc783cb03
commit 43c97d18aa
6 changed files with 66 additions and 39 deletions

View File

@@ -198,16 +198,19 @@ describe("memory index", () => {
it("reindexes when the embedding model changes", async () => {
const indexModelPath = path.join(workspaceDir, `index-model-change-${Date.now()}.sqlite`);
const base = createCfg({ storePath: indexModelPath });
const baseAgents = base.agents!;
const baseDefaults = baseAgents.defaults!;
const baseMemorySearch = baseDefaults.memorySearch!;
const first = await getMemorySearchManager({
cfg: {
...base,
agents: {
...base.agents,
...baseAgents,
defaults: {
...base.agents.defaults,
...baseDefaults,
memorySearch: {
...base.agents.defaults.memorySearch,
...baseMemorySearch,
model: "mock-embed-v1",
},
},
@@ -219,19 +222,19 @@ describe("memory index", () => {
if (!first.manager) {
throw new Error("manager missing");
}
await first.manager.sync({ reason: "test" });
await first.manager.sync?.({ reason: "test" });
const callsAfterFirstSync = embedBatchCalls;
await first.manager.close();
await first.manager.close?.();
const second = await getMemorySearchManager({
cfg: {
...base,
agents: {
...base.agents,
...baseAgents,
defaults: {
...base.agents.defaults,
...baseDefaults,
memorySearch: {
...base.agents.defaults.memorySearch,
...baseMemorySearch,
model: "mock-embed-v2",
},
},
@@ -243,11 +246,11 @@ describe("memory index", () => {
if (!second.manager) {
throw new Error("manager missing");
}
await second.manager.sync({ reason: "test" });
await second.manager.sync?.({ reason: "test" });
expect(embedBatchCalls).toBeGreaterThan(callsAfterFirstSync);
const status = second.manager.status();
expect(status.files).toBeGreaterThan(0);
await second.manager.close();
await second.manager.close?.();
});
it("reuses cached embeddings on forced reindex", async () => {

View File

@@ -1,4 +1,5 @@
import { beforeEach, describe, expect, it, vi } from "vitest";
import type { OpenClawConfig } from "../config/config.js";
const mockPrimary = {
search: vi.fn(async () => []),
@@ -69,15 +70,16 @@ vi.mock("./manager.js", () => ({
import { QmdMemoryManager } from "./qmd-manager.js";
import { getMemorySearchManager } from "./search-manager.js";
const createQmdManagerMock = vi.mocked(QmdMemoryManager.create.bind(QmdMemoryManager));
type SearchManagerResult = Awaited<ReturnType<typeof getMemorySearchManager>>;
type SearchManager = NonNullable<SearchManagerResult["manager"]>;
function createQmdCfg(agentId: string) {
function createQmdCfg(agentId: string): OpenClawConfig {
return {
memory: { backend: "qmd", qmd: {} },
agents: { list: [{ id: agentId, default: true, workspace: "/tmp/workspace" }] },
} as const;
};
}
function requireManager(result: SearchManagerResult): SearchManager {
@@ -112,7 +114,7 @@ beforeEach(() => {
fallbackManager.close.mockClear();
mockMemoryIndexGet.mockReset();
mockMemoryIndexGet.mockResolvedValue(fallbackManager);
QmdMemoryManager.create.mockClear();
createQmdManagerMock.mockClear();
});
describe("getMemorySearchManager caching", () => {
@@ -124,7 +126,7 @@ describe("getMemorySearchManager caching", () => {
expect(first.manager).toBe(second.manager);
// eslint-disable-next-line @typescript-eslint/unbound-method
expect(QmdMemoryManager.create).toHaveBeenCalledTimes(1);
expect(createQmdManagerMock).toHaveBeenCalledTimes(1);
});
it("evicts failed qmd wrapper so next call retries qmd", async () => {
@@ -146,7 +148,7 @@ describe("getMemorySearchManager caching", () => {
requireManager(second);
expect(second.manager).not.toBe(first.manager);
// eslint-disable-next-line @typescript-eslint/unbound-method
expect(QmdMemoryManager.create).toHaveBeenCalledTimes(2);
expect(createQmdManagerMock).toHaveBeenCalledTimes(2);
});
it("does not cache status-only qmd managers", async () => {
@@ -159,14 +161,14 @@ describe("getMemorySearchManager caching", () => {
requireManager(first);
requireManager(second);
// eslint-disable-next-line @typescript-eslint/unbound-method
expect(QmdMemoryManager.create).toHaveBeenCalledTimes(2);
expect(createQmdManagerMock).toHaveBeenCalledTimes(2);
// eslint-disable-next-line @typescript-eslint/unbound-method
expect(QmdMemoryManager.create).toHaveBeenNthCalledWith(
expect(createQmdManagerMock).toHaveBeenNthCalledWith(
1,
expect.objectContaining({ agentId, mode: "status" }),
);
// eslint-disable-next-line @typescript-eslint/unbound-method
expect(QmdMemoryManager.create).toHaveBeenNthCalledWith(
expect(createQmdManagerMock).toHaveBeenNthCalledWith(
2,
expect.objectContaining({ agentId, mode: "status" }),
);
@@ -193,7 +195,7 @@ describe("getMemorySearchManager caching", () => {
const third = await getMemorySearchManager({ cfg, agentId: retryAgentId });
expect(third.manager).toBe(secondManager);
// eslint-disable-next-line @typescript-eslint/unbound-method
expect(QmdMemoryManager.create).toHaveBeenCalledTimes(2);
expect(createQmdManagerMock).toHaveBeenCalledTimes(2);
});
it("falls back to builtin search when qmd fails with sqlite busy", async () => {