From 7cba2a7f51d648dad8d8e8e852ff3d44c4415131 Mon Sep 17 00:00:00 2001
From: Karan Hotchandani <33024356+karanh37@users.noreply.github.com>
Date: Wed, 13 Nov 2024 13:29:51 +0530
Subject: [PATCH] fix lineage tests (#18618)

---
 .../ui/playwright/e2e/Pages/Lineage.spec.ts      | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Pages/Lineage.spec.ts b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Pages/Lineage.spec.ts
index 1bb57c59097..b2c6698a7ed 100644
--- a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Pages/Lineage.spec.ts
+++ b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Pages/Lineage.spec.ts
@@ -357,7 +357,7 @@ test('Verify function data in edge drawer', async ({ browser }) => {
 
     await page.locator('.edge-info-drawer').isVisible();
 
-    await expect(await page.locator('[data-testid="Function"]')).toContainText(
+    await expect(page.locator('[data-testid="Function"]')).toContainText(
       'count'
     );
   } finally {
@@ -385,13 +385,6 @@ test('Verify global lineage config', async ({ browser }) => {
     await addPipelineBetweenNodes(page, topic, dashboard);
     await addPipelineBetweenNodes(page, dashboard, mlModel);
 
-    await topic.visitEntityPage(page);
-    await visitLineageTab(page);
-
-    await verifyNodePresent(page, table);
-    await verifyNodePresent(page, dashboard);
-    await verifyNodePresent(page, mlModel);
-
     await settingClick(page, GlobalSettingOptions.LINEAGE_CONFIG);
     await fillLineageConfigForm(page, {
       upstreamDepth: 1,
@@ -420,6 +413,13 @@ test('Verify global lineage config', async ({ browser }) => {
       downstreamDepth: 2,
       layer: 'Entity Lineage',
     });
+
+    await topic.visitEntityPage(page);
+    await visitLineageTab(page);
+
+    await verifyNodePresent(page, table);
+    await verifyNodePresent(page, dashboard);
+    await verifyNodePresent(page, mlModel);
   } finally {
     await table.delete(apiContext);
     await topic.delete(apiContext);