Fixed Entity spec (#26767)

Author: Rohit0301

openmetadata-ui/src/main/resources/ui/playwright/e2e/Pages/Entity.spec.ts

@@ -2078,12 +2078,15 @@ Object.entries(entities).forEach(([key, EntityClass]) => {
         const columnFqn =
           (entity as TableClass).entityResponseData.columns[0]
             .fullyQualifiedName ?? '';
+        const tableFqn =
+          (entity as TableClass).entityResponseData.fullyQualifiedName ?? '';
 
         for (const type of properties) {
           await test.step(`Set ${type} custom property on column and verify in UI`, async () => {
             await verifyTableColumnCustomPropertyPersistence({
               page,
               columnFqn,
+              tableFqn,
               propertyName: customPropertyValue[type].property.name,
               propertyType: type,
               users,

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

@@ -1137,21 +1137,36 @@ export const validateColumnCustomProperty = async (
 export const verifyTableColumnCustomPropertyPersistence = async ({
   page,
   columnFqn,
+  tableFqn,
   propertyName,
   propertyType,
   users,
 }: {
   page: Page;
   columnFqn: string;
+  tableFqn: string;
   propertyName: string;
   propertyType: string;
   users: Record<string, string>;
 }) => {
   const testValue = getPropertyValues(propertyType, users).value;
+  const columnsProfileResponse = () =>
+    page.waitForResponse(
+      (response) =>
+        response
+          .url()
+          .includes(
+            `/api/v1/tables/name/${encodeURIComponent(tableFqn)}/columns`
+          ) &&
+        response.url().includes('profile') &&
+        response.request().method() === 'GET',
+      { timeout: 90_000 }
+    );
 
   // 1. Navigate and Open Column Detail Panel
-  await page.goto(`/table/${columnFqn}`);
-  await page.waitForLoadState('networkidle');
+  const initialColumnsResponse = columnsProfileResponse();
+  await page.goto(`/table/${columnFqn}`, { waitUntil: 'domcontentloaded' });
+  await initialColumnsResponse;
   await waitForAllLoadersToDisappear(page);
   const sidePanel = page.locator('.column-detail-panel-container');
   await expect(sidePanel).toBeVisible();
@@ -1197,23 +1212,15 @@ export const verifyTableColumnCustomPropertyPersistence = async ({
     propertyName
   );
 
-  // 5. Reload Page
-  const getColumnDetails = page.waitForResponse(
-    '/api/v1/tables/name/*/columns?*fields=*extension*'
-  );
-  const getTableColumnTypes = page.waitForResponse(
-    '/api/v1/metadata/types/name/tableColumn*'
-  );
-  await page.reload();
-  await getTableColumnTypes;
-  await getColumnDetails;
-
-  await page.waitForSelector(
-    '.column-detail-panel-container [data-testid="custom-properties-tab"]',
-    {
-      state: 'visible',
-    }
-  );
+  const reloadColumnsResponse = columnsProfileResponse();
+  await page.reload({ waitUntil: 'domcontentloaded' });
+  await reloadColumnsResponse;
+  await waitForAllLoadersToDisappear(page);
+  await expect(
+    page.locator(
+      '.column-detail-panel-container [data-testid="custom-properties-tab"]'
+    )
+  ).toBeVisible();
   await customPropertiesTab.click();
   await expect(searchbar).toBeVisible();
   await searchbar.fill(propertyName);

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

@@ -1020,6 +1020,17 @@ export const openColumnDetailPanel = async ({
       )
     : null;
 
+  const columnsProfileResponsePromise =
+    entityType === 'table'
+      ? page.waitForResponse(
+          (response) =>
+            response.url().includes('/api/v1/tables/name/') &&
+            response.url().includes('/columns') &&
+            response.url().includes('profile') &&
+            response.request().method() === 'GET',
+          { timeout: 90_000 }
+        )
+      : null;
   if (entityType === 'MlModel') {
     const columnName = page
       .locator(`[${rowSelector}="${columnId}"]`)
@@ -1050,6 +1061,10 @@ export const openColumnDetailPanel = async ({
     expect(apiResponse.status()).toBe(200);
   }
 
+  if (columnsProfileResponsePromise) {
+    await columnsProfileResponsePromise;
+  }
+
   const panelContainer = page.locator('.column-detail-panel');
 
   // Wait for the panel content to be loaded