diff --git a/packages/opencode/test/session/prompt-effect.test.ts b/packages/opencode/test/session/prompt-effect.test.ts index 458041123..cb73683dd 100644 --- a/packages/opencode/test/session/prompt-effect.test.ts +++ b/packages/opencode/test/session/prompt-effect.test.ts @@ -79,6 +79,8 @@ function defer() { return { promise, resolve } } +const cancelRaceCheckpointTimeout = "5 seconds" + type Mutable = { -readonly [K in keyof T]: T[K] } function withSh(fx: () => Effect.Effect) { @@ -1801,11 +1803,14 @@ it.live( yield* Effect.addFinalizer(() => Effect.sync(() => void (mutableSessions.updateMessage = originalUpdateMessage))) const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild) - const startedExit = yield* Effect.promise(() => started.promise).pipe(Effect.timeout("1 second"), Effect.exit) + const startedExit = yield* Effect.promise(() => started.promise).pipe( + Effect.timeout(cancelRaceCheckpointTimeout), + Effect.exit, + ) expect(Exit.isSuccess(startedExit)).toBe(true) - const cancelExit = yield* prompt.cancel(chat.id).pipe(Effect.timeout("1 second"), Effect.exit) + const cancelExit = yield* prompt.cancel(chat.id).pipe(Effect.timeout(cancelRaceCheckpointTimeout), Effect.exit) expect(Exit.isSuccess(cancelExit)).toBe(true) - const exit = yield* Fiber.await(fiber).pipe(Effect.timeout("1 second")) + const exit = yield* Fiber.await(fiber).pipe(Effect.timeout(cancelRaceCheckpointTimeout)) expect(Exit.isSuccess(exit)).toBe(true) const messages = yield* sessions.messages({ sessionID: chat.id }) @@ -1855,11 +1860,14 @@ it.live( yield* Effect.addFinalizer(() => Effect.sync(() => void (mutablePlugin.trigger = originalTrigger))) const fiber = yield* prompt.loop({ sessionID: chat.id }).pipe(Effect.forkChild) - const startedExit = yield* Effect.promise(() => started.promise).pipe(Effect.timeout("1 second"), Effect.exit) + const startedExit = yield* Effect.promise(() => started.promise).pipe( + Effect.timeout(cancelRaceCheckpointTimeout), + Effect.exit, + ) expect(Exit.isSuccess(startedExit)).toBe(true) - const cancelExit = yield* prompt.cancel(chat.id).pipe(Effect.timeout("1 second"), Effect.exit) + const cancelExit = yield* prompt.cancel(chat.id).pipe(Effect.timeout(cancelRaceCheckpointTimeout), Effect.exit) expect(Exit.isSuccess(cancelExit)).toBe(true) - const exit = yield* Fiber.await(fiber).pipe(Effect.timeout("1 second")) + const exit = yield* Fiber.await(fiber).pipe(Effect.timeout(cancelRaceCheckpointTimeout)) expect(Exit.isSuccess(exit)).toBe(true) const messages = yield* sessions.messages({ sessionID: chat.id })