Function waitUntilCompleted

apps/vscode-e2e/src/suite/utils.ts:59–75

({
	api,
	taskId: passedTaskId,
	start,
	...options
}: WaitUntilCompletedOptions)

}

export const waitUntilCompleted = async ({
	api,
	taskId: passedTaskId,
	start,
	...options
}: WaitUntilCompletedOptions): Promise<string> => {
	const completed = new Set<string>()
	const handler = (id: string) => completed.add(id)
	api.on(RooCodeEventName.TaskCompleted, handler)
	try {
		const taskId = passedTaskId ?? (await start!())
		await waitFor(() => completed.has(taskId), options)
		return taskId
	} finally {
		api.off(RooCodeEventName.TaskCompleted, handler)
	}
}

export const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms))