From 7f5d129a37ea03e469e446ffa7b29292c67a786c Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 8 Apr 2026 16:53:23 +0100
Subject: [PATCH] fix: keep runtime task test harness behind task seams

---
 src/plugins/runtime/runtime-task-test-harness.ts | 4 ++--
 src/tasks/runtime-internal.ts                    | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/plugins/runtime/runtime-task-test-harness.ts b/src/plugins/runtime/runtime-task-test-harness.ts
index 12e4732cadb..b91351b5701 100644
--- a/src/plugins/runtime/runtime-task-test-harness.ts
+++ b/src/plugins/runtime/runtime-task-test-harness.ts
@@ -1,10 +1,10 @@
 import { vi } from "vitest";
-import { resetTaskFlowRegistryForTests } from "../../tasks/task-flow-registry.js";
 import {
   resetTaskRegistryDeliveryRuntimeForTests,
   resetTaskRegistryForTests,
   setTaskRegistryDeliveryRuntimeForTests,
-} from "../../tasks/task-registry.js";
+} from "../../tasks/runtime-internal.js";
+import { resetTaskFlowRegistryForTests } from "../../tasks/task-flow-runtime-internal.js";
 
 const runtimeTaskMocks = vi.hoisted(() => ({
   sendMessageMock: vi.fn(),
diff --git a/src/tasks/runtime-internal.ts b/src/tasks/runtime-internal.ts
index 8a9f41d7764..d1c7ea55641 100644
--- a/src/tasks/runtime-internal.ts
+++ b/src/tasks/runtime-internal.ts
@@ -21,9 +21,11 @@ export {
   markTaskTerminalByRunId,
   maybeDeliverTaskTerminalUpdate,
   recordTaskProgressByRunId,
+  resetTaskRegistryDeliveryRuntimeForTests,
   resolveTaskForLookupToken,
   resetTaskRegistryForTests,
   isParentFlowLinkError,
+  setTaskRegistryDeliveryRuntimeForTests,
   setTaskCleanupAfterById,
   setTaskProgressById,
   setTaskRunDeliveryStatusByRunId,