vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-18 15:14:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: guard memory lancedb mock calls

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-11 23:11:02 +01:00
parent b9920447ca
commit dbf0458d1b
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/memory-lancedb/index.test.ts
View File

@@ -71,7 +71,11 @@ function createRuntimeLoader(
type MockCallSource = { mock: { calls: Array<Array<unknown>> } };
function firstMockArg(source: MockCallSource, label: string, argIndex = 0) {
const arg = source.mock.calls[0]?.[argIndex];
const [call] = source.mock.calls;
if (!call) {
throw new Error(`expected ${label} call`);
}
const arg = call[argIndex];
if (arg === undefined) {
throw new Error(`expected ${label} arg`);
}
@@ -109,7 +113,9 @@ function expectToolExecute(tool: unknown, name?: string) {
}
function firstAddedMemory(add: ReturnType<typeof vi.fn>) {
const batch = add.mock.calls[0]?.[0] as Array<Record<string, unknown>> | undefined;
const batch = firstMockArg(add as MockCallSource, "memory add") as
| Array<Record<string, unknown>>
| undefined;
const memory = batch?.[0];
if (!memory) {
throw new Error("expected first added memory");