YeonGyu-Kim
93 lines
3.1 KiB
TypeScript
93 lines
3.1 KiB
TypeScript
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" })
|
|
})
|
|
})
|