diff --git a/packages/server/src/server/agent/providers/claude/subagents/workflow-replay-source.test.ts b/packages/server/src/server/agent/providers/claude/subagents/workflow-replay-source.test.ts index dc1418a5f1..a995b2c04c 100644 --- a/packages/server/src/server/agent/providers/claude/subagents/workflow-replay-source.test.ts +++ b/packages/server/src/server/agent/providers/claude/subagents/workflow-replay-source.test.ts @@ -88,7 +88,7 @@ describe("Claude workflow replay", () => { await store.apply("parent", "claude", event); } - expect(store.get("parent", TOOL_CALL_ID)).toMatchObject({ + expect(await store.get("parent", TOOL_CALL_ID)).toMatchObject({ id: TOOL_CALL_ID, title: "Workflow", description: "Runs one deterministic child and returns its structured result", @@ -113,7 +113,7 @@ describe("Claude workflow replay", () => { await store.apply("parent", "claude", event); } - expect(store.get("parent", TOOL_CALL_ID)).toMatchObject({ + expect(await store.get("parent", TOOL_CALL_ID)).toMatchObject({ id: TOOL_CALL_ID, title: "Workflow", description: "Structured result workflow", diff --git a/packages/server/src/server/agent/providers/pi/extensions/gotgenes-pi-subagents/index.test.ts b/packages/server/src/server/agent/providers/pi/extensions/gotgenes-pi-subagents/index.test.ts index 3e7648d8a4..8174b5066e 100644 --- a/packages/server/src/server/agent/providers/pi/extensions/gotgenes-pi-subagents/index.test.ts +++ b/packages/server/src/server/agent/providers/pi/extensions/gotgenes-pi-subagents/index.test.ts @@ -52,7 +52,9 @@ describe("@gotgenes/pi-subagents adapter", () => { }); for (const event of events) await store.apply("parent", "pi", event.event); expect(events.map(({ event }) => event.type)).toEqual(["upsert", "remove"]); - expect(store.list("parent")).toMatchObject([{ id: "previous-child", status: "completed" }]); + expect(await store.list("parent")).toMatchObject([ + { id: "previous-child", status: "completed" }, + ]); }, );