fix(delegate-task): route sync prompts by directory

2026-05-12 18:14:17 +09:00
parent 6035a551ad
commit c740ed8aca
3 changed files with 98 additions and 2 deletions
@@ -0,0 +1,92 @@
+import { describe, expect, mock, test } from "bun:test"
+
+import { unsafeTestValue } from "../../../test-support/unsafe-test-value"
+import type { OpencodeClient } from "./types"
+import { sendSyncPrompt } from "./sync-prompt-sender"
+import {
+  promptSyncWithModelSuggestionRetry,
+  promptWithModelSuggestionRetry,
+} from "../../shared/model-suggestion-retry"
+
+type PromptRetryClient = Parameters<typeof promptWithModelSuggestionRetry>[0]
+type PromptRetryArgs = Parameters<typeof promptWithModelSuggestionRetry>[1]
+type PromptSyncRetryClient = Parameters<typeof promptSyncWithModelSuggestionRetry>[0]
+type PromptSyncRetryArgs = Parameters<typeof promptSyncWithModelSuggestionRetry>[1]
+
+describe("sendSyncPrompt session routing", () => {
+  test("#given a sync child session directory #when sending the prompt #then promptAsync uses that OpenCode directory route", async () => {
+    // given
+    const promptCalls: PromptRetryArgs[] = []
+    const promptWithRetry = mock(async (_client: PromptRetryClient, input: PromptRetryArgs) => {
+      promptCalls.push(input)
+    })
+
+    // when
+    await sendSyncPrompt(
+      unsafeTestValue<OpencodeClient>({ session: {} }),
+      {
+        sessionID: "ses_child",
+        agentToUse: "sisyphus-junior",
+        args: {
+          description: "test task",
+          prompt: "test prompt",
+          run_in_background: false,
+          load_skills: [],
+        },
+        systemContent: undefined,
+        categoryModel: undefined,
+        directory: "/parent/project",
+        toastManager: null,
+        taskId: undefined,
+      },
+      {
+        promptWithModelSuggestionRetry: promptWithRetry,
+        promptSyncWithModelSuggestionRetry: mock(async () => {}),
+      },
+    )
+
+    // then
+    expect(promptCalls).toHaveLength(1)
+    expect(promptCalls[0]?.query).toEqual({ directory: "/parent/project" })
+  })
+
+  test("#given oracle falls back to promptSync #when async prompt returns unexpected EOF #then the sync retry keeps the same directory route", async () => {
+    // given
+    const promptSyncCalls: PromptSyncRetryArgs[] = []
+    const promptWithRetry = mock(async () => {
+      throw new Error("JSON Parse error: Unexpected EOF")
+    })
+    const promptSyncWithRetry = mock(async (_client: PromptSyncRetryClient, input: PromptSyncRetryArgs) => {
+      promptSyncCalls.push(input)
+    })
+
+    // when
+    const result = await sendSyncPrompt(
+      unsafeTestValue<OpencodeClient>({ session: {} }),
+      {
+        sessionID: "ses_child",
+        agentToUse: "oracle",
+        args: {
+          description: "test task",
+          prompt: "test prompt",
+          run_in_background: false,
+          load_skills: [],
+        },
+        systemContent: undefined,
+        categoryModel: undefined,
+        directory: "/parent/project",
+        toastManager: null,
+        taskId: undefined,
+      },
+      {
+        promptWithModelSuggestionRetry: promptWithRetry,
+        promptSyncWithModelSuggestionRetry: promptSyncWithRetry,
+      },
+    )
+
+    // then
+    expect(result).toBeNull()
+    expect(promptSyncCalls).toHaveLength(1)
+    expect(promptSyncCalls[0]?.query).toEqual({ directory: "/parent/project" })
+  })
+})