Skip to content

Commit 94b4312

Browse files
committed
test: tighten memory wiki query assertions
1 parent 1741a7f commit 94b4312

2 files changed

Lines changed: 23 additions & 31 deletions

File tree

extensions/memory-wiki/src/query.test.ts

Lines changed: 11 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -352,15 +352,11 @@ describe("searchMemoryWiki", () => {
352352
query: "bgroux",
353353
mode: "find-person",
354354
});
355-
expect(personResults[0]).toEqual(
356-
expect.objectContaining({
357-
path: "entities/brad.md",
358-
canonicalId: "maintainer.brad-groux",
359-
aliases: ["bgroux"],
360-
privacyTier: "local-private",
361-
searchMode: "find-person",
362-
}),
363-
);
355+
expect(personResults[0]?.path).toBe("entities/brad.md");
356+
expect(personResults[0]?.canonicalId).toBe("maintainer.brad-groux");
357+
expect(personResults[0]?.aliases).toEqual(["bgroux"]);
358+
expect(personResults[0]?.privacyTier).toBe("local-private");
359+
expect(personResults[0]?.searchMode).toBe("find-person");
364360

365361
const routeResults = await searchMemoryWiki({
366362
config,
@@ -374,15 +370,11 @@ describe("searchMemoryWiki", () => {
374370
query: "strong route Teams",
375371
mode: "raw-claim",
376372
});
377-
expect(claimResults[0]).toEqual(
378-
expect.objectContaining({
379-
path: "entities/brad.md",
380-
matchedClaimId: "claim.brad.teams",
381-
matchedClaimConfidence: 0.88,
382-
evidenceKinds: ["maintainer-whois"],
383-
evidenceSourceIds: ["source.maintainers"],
384-
}),
385-
);
373+
expect(claimResults[0]?.path).toBe("entities/brad.md");
374+
expect(claimResults[0]?.matchedClaimId).toBe("claim.brad.teams");
375+
expect(claimResults[0]?.matchedClaimConfidence).toBe(0.88);
376+
expect(claimResults[0]?.evidenceKinds).toEqual(["maintainer-whois"]);
377+
expect(claimResults[0]?.evidenceSourceIds).toEqual(["source.maintainers"]);
386378

387379
const evidenceResults = await searchMemoryWiki({
388380
config,
@@ -648,9 +640,7 @@ describe("searchMemoryWiki", () => {
648640
});
649641

650642
expect(results).toHaveLength(2);
651-
expect(results.map((result) => result.corpus)).toEqual(
652-
expect.arrayContaining(["wiki", "memory"]),
653-
);
643+
expect(results.map((result) => result.corpus).sort()).toEqual(["memory", "wiki"]);
654644
expect(manager.search).toHaveBeenCalledWith("alpha", { maxResults: 5 });
655645
expect(getActiveMemorySearchManagerMock).toHaveBeenCalledWith({
656646
cfg: createAppConfig(),
@@ -763,9 +753,6 @@ describe("searchMemoryWiki", () => {
763753
"sessions/child-session.jsonl",
764754
"MEMORY.md",
765755
]);
766-
expect(results.map((result) => result.path)).not.toEqual(
767-
expect.arrayContaining([expect.stringContaining("sibling-session")]),
768-
);
769756
});
770757

771758
it("filters session memory hits for session-bound non-sandboxed callers", async () => {
@@ -819,9 +806,6 @@ describe("searchMemoryWiki", () => {
819806
"sessions/child-session.jsonl",
820807
"MEMORY.md",
821808
]);
822-
expect(results.map((result) => result.path)).not.toEqual(
823-
expect.arrayContaining([expect.stringContaining("sibling-session")]),
824-
);
825809
});
826810

827811
it("requires appConfig for session-bound shared memory searches", async () => {

extensions/memory-wiki/src/tool.test.ts

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,17 @@ describe("memory-wiki tools", () => {
2323
const evidenceArraySchema = asSchemaObject(evidenceSchema.items);
2424
const evidenceProperties = asSchemaObject(evidenceArraySchema.properties);
2525

26-
expect(Object.keys(evidenceProperties)).toEqual(
27-
expect.arrayContaining(["kind", "confidence", "privacyTier"]),
28-
);
29-
expect(evidenceProperties.confidence).toMatchObject({ minimum: 0, maximum: 1 });
26+
expect(Object.keys(evidenceProperties).sort()).toEqual([
27+
"confidence",
28+
"kind",
29+
"lines",
30+
"note",
31+
"path",
32+
"privacyTier",
33+
"sourceId",
34+
"updatedAt",
35+
"weight",
36+
]);
37+
expect(evidenceProperties.confidence).toEqual({ type: "number", minimum: 0, maximum: 1 });
3038
});
3139
});

0 commit comments

Comments
 (0)