fixed flaky bulk import

Author: dhruvjsx

openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/BulkImport.spec.ts

@@ -158,7 +158,15 @@ test.describe('Bulk Import Export', () => {
       'should import and edit with two additional database',
       async () => {
         await dbService.visitEntityPage(page);
-        await page.click('[data-testid="manage-button"] > .anticon');
+        await page.waitForSelector('[data-testid="loader"]', {
+          state: 'detached',
+        });
+
+        await page.getByTestId('manage-button').click();
+        await page.waitForSelector(
+          '[data-testid="manage-dropdown-list-container"]',
+          { state: 'visible' }
+        );
         await page.click('[data-testid="import-button-title"]');
         const fileInput = page.getByTestId('upload-file-widget');
         await fileInput?.setInputFiles([
@@ -409,15 +417,26 @@ test.describe('Bulk Import Export', () => {
       'should import and edit with two additional database schema',
       async () => {
         await dbEntity.visitEntityPage(page);
-        await page.click('[data-testid="manage-button"] > .anticon');
+
+        await page.waitForSelector('[data-testid="loader"]', {
+          state: 'detached',
+        });
+
+        await page.getByTestId('manage-button').click();
+        await page.waitForSelector(
+          '[data-testid="manage-dropdown-list-container"]',
+          { state: 'visible' }
+        );
         await page.click('[data-testid="import-button-title"]');
         const fileInput = await page.$('[type="file"]');
         await fileInput?.setInputFiles([
           'downloads/' + dbEntity.entity.name + '.csv',
         ]);
 
-        // Adding manual wait for the file to load
-        await page.waitForTimeout(500);
+        // Wait for upload widget to be hidden indicating file is loaded
+        await page.waitForSelector('[data-testid="upload-file-widget"]', {
+          state: 'hidden',
+        });
 
         // Adding some assertion to make sure that CSV loaded correctly
         await expect(page.locator('.rdg-header-row')).toBeVisible();

openmetadata-ui/src/main/resources/ui/playwright/support/entity/DatabaseSchemaClass.ts

@@ -152,13 +152,18 @@ export class DatabaseSchemaClass extends EntityClass {
     await page.getByTestId(this.database.name).click();
     await databaseResponse;
 
-    // Wait for database schema to be visible
-    await page.getByTestId(this.entity.name).waitFor({ state: 'visible' });
+    // Wait for page to fully load after navigation
+    await page.waitForLoadState('networkidle');
+
+    // Target schema specifically within the table container to avoid clicking breadcrumbs or other elements
+    const schemaLocator = page.getByTestId(this.entity.name);
+
+    await schemaLocator.waitFor({ state: 'visible' });
 
     const databaseSchemaResponse = page.waitForResponse(
       `/api/v1/databaseSchemas/name/*${this.entity.name}?*`
     );
-    await page.getByTestId(this.entity.name).click();
+    await schemaLocator.click();
     await databaseSchemaResponse;
   }
 

openmetadata-ui/src/main/resources/ui/playwright/utils/importUtils.ts

@@ -103,6 +103,8 @@ export const fillOwnerDetails = async (page: Page, owners: string[]) => {
   const userListResponse = page.waitForResponse(
     '/api/v1/search/query?q=&index=user_search_index&*'
   );
+   //wait for small duration to stablize the page
+   page.waitForTimeout(500)
   await page.getByRole('tab', { name: 'Users' }).click();
   await userListResponse;