From ec2bef8154922722b9eff737c0be489bd4c8f671 Mon Sep 17 00:00:00 2001 From: Shakker Date: Mon, 11 May 2026 10:56:32 +0100 Subject: [PATCH] test: tighten memory citation assertions --- .../memory-core/src/tools.citations.test.ts | 22 +++++++++++-------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/extensions/memory-core/src/tools.citations.test.ts b/extensions/memory-core/src/tools.citations.test.ts index 2629a0f60db..0f54cd29908 100644 --- a/extensions/memory-core/src/tools.citations.test.ts +++ b/extensions/memory-core/src/tools.citations.test.ts @@ -257,19 +257,17 @@ describe("memory tools", () => { }; const entries = Object.values(store.entries ?? {}); expect(entries).toHaveLength(1); - expect(entries[0]).toMatchObject({ - path: "memory/2026-04-03.md", - recallCount: 1, - }); + const entry = entries[0]; + expect(entry?.path).toBe("memory/2026-04-03.md"); + expect(entry?.recallCount).toBe(1); const events = await waitFor(async () => { const memoryEvents = await readMemoryHostEvents({ workspaceDir }); expect(memoryEvents).toHaveLength(1); return memoryEvents; }); - expect(events[0]).toMatchObject({ - type: "memory.recall.recorded", - query: "glacier backup", - }); + const event = events[0]; + expect(event?.type).toBe("memory.recall.recorded"); + expect(event?.query).toBe("glacier backup"); } finally { await fs.rm(workspaceDir, { recursive: true, force: true }); } @@ -293,7 +291,7 @@ describe("memory tools", () => { const tool = createMemorySearchToolOrThrow(); const result = await tool.execute("call_wiki_only", { query: "alpha", corpus: "wiki" }); - expect(result.details).toMatchObject({ + expect(result.details).toStrictEqual({ results: [ { corpus: "wiki", @@ -304,6 +302,12 @@ describe("memory tools", () => { snippet: "Alpha wiki entry", }, ], + citations: "auto", + debug: undefined, + fallback: undefined, + mode: undefined, + model: undefined, + provider: undefined, }); expect(getMemorySearchManagerMockCalls()).toBe(0); });