From 414964e52df183d3b692fa18ba52f3432d5c7a3f 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)

(cherry picked from commit 7cba2a7f51d648dad8d8e8e852ff3d44c4415131)
---
 .../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 77e655769a7..90cb17e2910 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
@@ -353,7 +353,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 {
@@ -381,13 +381,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,
@@ -416,6 +409,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);