From 1981622b92b207fbd03e98f2670ec0e70f466122 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Fri, 24 Apr 2026 18:14:24 +0100
Subject: [PATCH] test: update ci shard and reply mocks

---
 src/auto-reply/reply/get-reply-run.media-only.test.ts | 1 +
 test/scripts/ci-node-test-plan.test.ts                | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/src/auto-reply/reply/get-reply-run.media-only.test.ts b/src/auto-reply/reply/get-reply-run.media-only.test.ts
index 43707590104..d5896b102c4 100644
--- a/src/auto-reply/reply/get-reply-run.media-only.test.ts
+++ b/src/auto-reply/reply/get-reply-run.media-only.test.ts
@@ -76,6 +76,7 @@ vi.mock("./body.js", () => ({
 vi.mock("./groups.js", () => ({
   buildGroupIntro: vi.fn().mockReturnValue(""),
   buildGroupChatContext: vi.fn().mockReturnValue(""),
+  buildDirectChatContext: vi.fn().mockReturnValue(""),
 }));
 
 vi.mock("./inbound-meta.js", () => ({
diff --git a/test/scripts/ci-node-test-plan.test.ts b/test/scripts/ci-node-test-plan.test.ts
index ece9e37e310..9bd9a6c1753 100644
--- a/test/scripts/ci-node-test-plan.test.ts
+++ b/test/scripts/ci-node-test-plan.test.ts
@@ -96,6 +96,7 @@ describe("scripts/lib/ci-node-test-plan.mjs", () => {
       .map((shard) => ({
         configs: shard.configs,
         requiresDist: shard.requiresDist,
+        runner: shard.runner,
         shardName: shard.shardName,
       }));
 
@@ -110,6 +111,7 @@ describe("scripts/lib/ci-node-test-plan.mjs", () => {
           "test/vitest/vitest.runtime-config.config.ts",
         ],
         requiresDist: false,
+        runner: "blacksmith-4vcpu-ubuntu-2404",
         shardName: "core-runtime-infra",
       },
       {
@@ -121,6 +123,7 @@ describe("scripts/lib/ci-node-test-plan.mjs", () => {
           "test/vitest/vitest.wizard.config.ts",
         ],
         requiresDist: false,
+        runner: undefined,
         shardName: "core-runtime-media-ui",
       },
       {
@@ -132,6 +135,7 @@ describe("scripts/lib/ci-node-test-plan.mjs", () => {
           "test/vitest/vitest.utils.config.ts",
         ],
         requiresDist: false,
+        runner: undefined,
         shardName: "core-runtime-shared",
       },
     ]);
@@ -149,6 +153,7 @@ describe("scripts/lib/ci-node-test-plan.mjs", () => {
       checkName: "checks-node-agentic-control-plane",
       shardName: "agentic-control-plane",
       configs: ["test/vitest/vitest.gateway-server.config.ts"],
+      runner: "blacksmith-4vcpu-ubuntu-2404",
       requiresDist: false,
     });
     expect(commandsShard).toEqual({