Merge pull request #188 from GoogleCloudDataproc/ojarjur/playwright-t…

…imeout Fix the Kokoro integration tests
GoogleCloudDataproc · Oct 23, 2024 · db8fff0 · db8fff0
2 parents 62d0b4c + dadb7d4
commit db8fff0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/ui-tests/playwright.config.js b/ui-tests/playwright.config.js
@@ -22,6 +22,7 @@ const baseConfig = require('@jupyterlab/galata/lib/playwright-config');
 
 module.exports = {
   ...baseConfig,
+  timeout: 120000,
   webServer: {
     command: 'jlpm start --allow-root',
     url: 'http://localhost:8888/lab',

diff --git a/ui-tests/tests/bigquery_dataset_explorer.spec.ts b/ui-tests/tests/bigquery_dataset_explorer.spec.ts
@@ -35,13 +35,20 @@ test('bigquery-dataset-explorer', async ({ page, request }) => {
       .getByRole('tab', { name: 'Dataset Explorer - BigQuery' })
       .click();
 
+    // Wait for the Dataset projects to populate.
+    await page.waitForSelector('div[role="treeitem"][aria-level="1"]');
     await page.waitForSelector('div[role="treeitem"].caret-icon.down');
 
-    await page.locator('div[role="treeitem"].caret-icon.down').nth(1).click();
-    await page.locator('div[role="treeitem"][aria-level="1"]').first().click();
-    await page.locator('div[role="treeitem"].caret-icon.down').nth(1).click();
+    // Expand the first dataset project. This should always be the `bigquery-public-data` one.
+    await page.locator('div[role="treeitem"].caret-icon.down').nth(0).click();
 
-    await page.locator('div[role="treeitem"][aria-level="2"]').first().click();
+    // Wait for the first dataset to be displayed, and then expand it.
+    await page.waitForSelector('div[role="treeitem"][aria-level="2"]');
+    await page.waitForSelector('div[role="treeitem"].caret-icon.down');
+    await page.locator('div[role="treeitem"].caret-icon.down').nth(0).click();
+
+    // Click on the first table displayed.
+    await page.locator('div[role="treeitem"][aria-level="3"]').first().click();
     await page.getByText('Schema', { exact: true }).click();
     await page.getByText('Preview', { exact: true }).click();
   } else {