Function mockChatScript

src/browser/stores/WorkspaceStore.test.ts:413–428

(steps: ChatStep[], options: { keepOpen?: boolean } = {})

}

function mockChatScript(steps: ChatStep[], options: { keepOpen?: boolean } = {}): void {
  mockOnChat.mockImplementation(async function* (
    _input?: { workspaceId: string; mode?: unknown },
    signalOptions?: { signal?: AbortSignal }
  ): AsyncGenerator<WorkspaceChatMessage, void, unknown> {
    for (const step of steps) {
      const event = await runChatStep(step);
      if (event) {
        yield event;
      }
    }
    if (options.keepOpen ?? false) {
      await waitForAbortSignal(signalOptions?.signal);
    }
  });
}

function mockChatReconnectScript(
  getSteps: (subscriptionCount: number, signal: AbortSignal | undefined) => ChatStep[]