From aee5dc32334fbc7f4fa3f586ab5d391098efadee Mon Sep 17 00:00:00 2001
From: lforst <8118419+lforst@users.noreply.github.com>
Date: Wed, 26 Aug 2026 13:45:42 +0000
Subject: [PATCH] chore: Pass on attachments in e2e test prod forwarding

---
 e2e/helpers/mock-braintrust-server.ts | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/e2e/helpers/mock-braintrust-server.ts b/e2e/helpers/mock-braintrust-server.ts
index d6f053011..7f59a57ed 100644
--- a/e2e/helpers/mock-braintrust-server.ts
+++ b/e2e/helpers/mock-braintrust-server.ts
@@ -682,6 +682,16 @@ export async function startMockBraintrustServer(
           capturedRequest.method === "POST" &&
           capturedRequest.path === "/attachment"
         ) {
+          if (prodForwarding) {
+            const forwardedResponse = await forwardProdRequest(capturedRequest);
+            respondJson(
+              res,
+              200,
+              (await forwardedResponse.json()) as JsonValue,
+            );
+            return;
+          }
+
           const key =
             isRecord(capturedRequest.jsonBody) &&
             typeof capturedRequest.jsonBody.key === "string"
@@ -706,6 +716,11 @@ export async function startMockBraintrustServer(
           capturedRequest.method === "POST" &&
           capturedRequest.path === "/attachment/status"
         ) {
+          if (prodForwarding) {
+            await forwardProdRequest(capturedRequest, {
+              drainResponseBody: true,
+            });
+          }
           respondJson(res, 200, { ok: true });
           return;
         }