Fix LineageFilter playwright

Author: Rohit0301

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

@@ -40,6 +40,7 @@ import {
   openImpactAnalysisTab,
   performZoomOut,
   rearrangeNodes,
+  setLineageDepthAndVerify,
   visitLineageTab,
 } from '../../../utils/lineage';
 import { test } from '../../fixtures/pages';
@@ -133,6 +134,8 @@ test.describe('Lineage Filters', () => {
     await lineageEntity.visitEntityPage(page);
     await visitLineageTab(page);
     await waitForAllLoadersToDisappear(page);
+    await setLineageDepthAndVerify(page, 2, 2);
+    await waitForAllLoadersToDisappear(page);
     await rearrangeNodes(page);
     await performZoomOut(page);
     await expect(
@@ -279,6 +282,8 @@ test.describe('Lineage Filters', () => {
         await test.step('Verify filters working for Lineage tab', async () => {
           await page.reload();
           await waitForAllLoadersToDisappear(page);
+          await setLineageDepthAndVerify(page, 2, 2);
+          await waitForAllLoadersToDisappear(page);
 
           await page.getByRole('button', { name: 'Filters' }).click();
           await page.getByTestId(`search-dropdown-${filterTestId}`).click();
@@ -962,7 +967,9 @@ test.describe('Lineage Filters', () => {
       for (const entity of depth2ndEntities) {
         await entity.visitEntityPage(page);
         await visitLineageTab(page);
-        await openImpactAnalysisTab(page);
+        await waitForAllLoadersToDisappear(page);
+        await setLineageDepthAndVerify(page, 2, 2);
+        await page.getByRole('tab', { name: 'Impact Analysis' }).click();
         await waitForAllLoadersToDisappear(page);
 
         await expect(
@@ -1000,7 +1007,9 @@ test.describe('Lineage Filters', () => {
       for (const entity of depth2ndEntities) {
         await entity.visitEntityPage(page);
         await visitLineageTab(page);
-        await openImpactAnalysisTab(page);
+        await waitForAllLoadersToDisappear(page);
+        await setLineageDepthAndVerify(page, 2, 2);
+        await page.getByRole('tab', { name: 'Impact Analysis' }).click();
 
         // Verify Dashboard is visible in Impact Analysis for Upstream
         await page.getByRole('button', { name: 'Upstream' }).click();

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

@@ -869,9 +869,7 @@ export const verifyColumnLineageInCSV = async (
 export const verifyLineageConfig = async (page: Page) => {
   await page.getByTestId('lineage-config').click();
 
-  await page.locator('.ant-modal-content').first().waitFor({
-    state: 'visible',
-  });
+  await page.getByTestId('field-upstream').waitFor({ state: 'visible' });
 
   await page.getByTestId('field-upstream').fill('-1');
   await page.getByTestId('field-downstream').fill('-1');
@@ -941,9 +939,7 @@ export const updateLineageConfigFromModal = async (
 ) => {
   await page.getByTestId('lineage-config').click();
 
-  await page.locator('.ant-modal-content').first().waitFor({
-    state: 'visible',
-  });
+  await page.getByTestId('field-upstream').waitFor({ state: 'visible' });
 
   await page
     .getByTestId('field-upstream')
@@ -956,6 +952,33 @@ export const updateLineageConfigFromModal = async (
   await page.getByRole('dialog').waitFor({ state: 'hidden' });
 };
 
+export const setLineageDepthAndVerify = async (
+  page: Page,
+  upstreamDepth: number,
+  downstreamDepth: number
+) => {
+  await page.getByTestId('lineage-config').click();
+
+  await page.getByTestId('field-upstream').waitFor({ state: 'visible' });
+
+  await page.getByTestId('field-upstream').fill(upstreamDepth.toString());
+  await page.getByTestId('field-downstream').fill(downstreamDepth.toString());
+
+  const lineageRes = page.waitForResponse((response) => {
+    const url = response.url();
+
+    return (
+      url.includes('/api/v1/lineage/getLineage') &&
+      url.includes(`upstreamDepth=${upstreamDepth}`) &&
+      url.includes(`downstreamDepth=${downstreamDepth}`)
+    );
+  });
+
+  await page.getByText('OK').click();
+  await page.getByRole('dialog').waitFor({ state: 'hidden' });
+  await lineageRes;
+};
+
 export const verifyPlatformLineageForEntity = async (
   page: Page,
   fromFqn: string,