From 183bc736ec4d436015eab257984eda175d899c44 Mon Sep 17 00:00:00 2001
From: Mohamed Boudra <boudra.moha@gmail.com>
Date: Wed, 11 Nov 2026 04:00:11 +0100
Subject: [PATCH] test(server): await provider child reads in the replay and pi
 child tests

These tests came in with the main merge. The child store reads are asynchronous now, so the tests compared a promise with the expected child.
---
 .../providers/claude/subagents/workflow-replay-source.test.ts | 4 ++--
 .../pi/extensions/gotgenes-pi-subagents/index.test.ts         | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

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" },
+      ]);
     },
   );