Function waitForTestComplete

testing/e2e/tests/tools-test/helpers.ts:168–195

(
  page: Page,
  timeout = 15000,
  expectedToolCount = 1,
)

 *   2. data-is-loading === "false" AND data-complete-tool-count >= expectedToolCount
 */
export async function waitForTestComplete(
  page: Page,
  timeout = 15000,
  expectedToolCount = 1,
): Promise<void> {
  await page.waitForFunction(
    ({ minTools }) => {
      const metadata = document.getElementById('test-metadata')
      const testComplete =
        metadata?.getAttribute('data-test-complete') === 'true'
      const isLoading = metadata?.getAttribute('data-is-loading') === 'true'
      const completeToolCount = parseInt(
        metadata?.getAttribute('data-complete-tool-count') || '0',
        10,
      )

      // Consider complete if:
      // 1. testComplete flag is true, OR
      // 2. Not loading and we have at least the expected number of complete tools
      return testComplete || (!isLoading && completeToolCount >= minTools)
    },
    { minTools: expectedToolCount },
    { timeout },
  )

  // Give a little extra time for final state updates
  await page.waitForTimeout(200)
}

/**
 * Wait for client-side execution events to be reflected in the event log.