mirror of
https://github.com/openclaw/openclaw.git
synced 2026-05-09 16:04:33 +00:00
chore: Fix types in tests 17/N.
This commit is contained in:
@@ -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 () => {
|
||||
|
||||
Reference in New Issue
Block a user