From 250c756fb4cafc81e5e8853c8a3c2eb3df88ed94 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 20 Apr 2026 22:51:01 +0100
Subject: [PATCH] test: share directive reply mock payloads

---
 ....directive.directive-behavior.e2e-mocks.ts | 22 ++++++-------------
 1 file changed, 7 insertions(+), 15 deletions(-)

diff --git a/src/auto-reply/reply.directive.directive-behavior.e2e-mocks.ts b/src/auto-reply/reply.directive.directive-behavior.e2e-mocks.ts
index 107c2b7aa7f..4146a466bd4 100644
--- a/src/auto-reply/reply.directive.directive-behavior.e2e-mocks.ts
+++ b/src/auto-reply/reply.directive.directive-behavior.e2e-mocks.ts
@@ -37,8 +37,7 @@ function normalizeReplyAgentPayload(payload: Record<string, unknown>, params: un
   };
 }
 
-export async function runDirectiveBehaviorReplyAgent(params: unknown) {
-  const runParams = objectRecord(objectRecord(params)?.followupRun)?.run ?? {};
+async function runMockedReplyAgent(runParams: unknown, params: unknown) {
   const result = await runEmbeddedPiAgentMock(runParams);
   const payloadsRaw = objectRecord(result)?.payloads;
   const payloads = Array.isArray(payloadsRaw)
@@ -54,6 +53,11 @@ export async function runDirectiveBehaviorReplyAgent(params: unknown) {
   return normalized.length === 1 ? normalized[0] : normalized;
 }
 
+export async function runDirectiveBehaviorReplyAgent(params: unknown) {
+  const runParams = objectRecord(objectRecord(params)?.followupRun)?.run ?? {};
+  return await runMockedReplyAgent(runParams, params);
+}
+
 export const runReplyAgentMock: Mock = vi.fn(runDirectiveBehaviorReplyAgent);
 
 export async function runDirectiveBehaviorPreparedReply(params: unknown) {
@@ -71,19 +75,7 @@ export async function runDirectiveBehaviorPreparedReply(params: unknown) {
       fullAccessAvailable: true,
     },
   };
-  const result = await runEmbeddedPiAgentMock(runParams);
-  const payloadsRaw = objectRecord(result)?.payloads;
-  const payloads = Array.isArray(payloadsRaw)
-    ? payloadsRaw.flatMap((payload) => {
-        const record = objectRecord(payload);
-        return record ? [record] : [];
-      })
-    : [];
-  const normalized = payloads.map((payload) => normalizeReplyAgentPayload(payload, params));
-  if (normalized.length === 0) {
-    return undefined;
-  }
-  return normalized.length === 1 ? normalized[0] : normalized;
+  return await runMockedReplyAgent(runParams, params);
 }
 
 export const runPreparedReplyMock: Mock = vi.fn(runDirectiveBehaviorPreparedReply);