From aa30cdf7ed3bed3d64f4ff05b01b2ff83d2dfbdd Mon Sep 17 00:00:00 2001
From: Shrushti Polekar <shrushtipolekar@gmail.com>
Date: Wed, 14 May 2025 14:35:15 +0530
Subject: [PATCH] fix flaky activity feed test (#21159)

---
 .../resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts
index 71772b35456..b1847fdad30 100644
--- a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts
+++ b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts
@@ -714,7 +714,7 @@ base.describe('Activity feed with Data Consumer User', () => {
       await page2.waitForLoadState('networkidle');
       // Count for task should be 1 both open and closed
 
-      checkTaskCountInActivityFeed(page2, 1, 1);
+      await checkTaskCountInActivityFeed(page2, 1, 1);
 
       // Should not see the close button
       expect(page2.locator('[data-testid="close-button"]')).not.toBeVisible();
@@ -730,7 +730,7 @@ base.describe('Activity feed with Data Consumer User', () => {
       await toastNotification(page2, /Task resolved successfully/);
 
       await page2.waitForLoadState('networkidle');
-      checkTaskCountInActivityFeed(page2, 0, 2);
+      await checkTaskCountInActivityFeed(page2, 0, 2);
 
       await afterActionUser2();
     });