Fix explore quick filters test (#25955)

* Fix the test ExploreQuickFilters failing in AUTs

* Add test setup

* modify the variable name

Author: aniketkatkar97

openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ExploreQuickFilters.spec.ts

@@ -11,9 +11,11 @@
  *  limitations under the License.
  */
 import test, { expect } from '@playwright/test';
+import { isUndefined } from 'lodash';
 import { SidebarItem } from '../../constant/sidebar';
 import { Domain } from '../../support/domain/Domain';
 import { TableClass } from '../../support/entity/TableClass';
+import { TagClass } from '../../support/tag/TagClass';
 import {
   clickOutside,
   createNewPage,
@@ -28,13 +30,17 @@ test.use({ storageState: 'playwright/.auth/admin.json' });
 
 const domain = new Domain();
 const table = new TableClass();
+const tier = new TagClass({
+  classification: 'Tier',
+});
 
 test.beforeAll('Setup pre-requests', async ({ browser }) => {
   test.slow();
 
   const { apiContext, afterAction } = await createNewPage(browser);
   await table.create(apiContext);
   await domain.create(apiContext);
+  await tier.create(apiContext);
 
   await table.patch({
     apiContext,
@@ -49,7 +55,7 @@ test.beforeAll('Setup pre-requests', async ({ browser }) => {
       {
         op: 'add',
         value: {
-          tagFQN: 'Tier.Tier5',
+          tagFQN: tier.responseData.fullyQualifiedName,
         },
         path: '/tags/1',
       },
@@ -122,27 +128,37 @@ test('should show correct count for initial options', async ({ page }) => {
     const aggregateAPI = page.waitForResponse(
       '/api/v1/search/aggregate?index=dataAsset&field=tier.tagFQN*'
     );
+    const tierFetchAPI = page.waitForResponse(
+      '/api/v1/tags?parent=Tier&limit=50'
+    );
     await page.click(`[data-testid="search-dropdown-${filter.label}"]`);
 
     const res = await aggregateAPI;
+    const tierRes = await tierFetchAPI;
     const data = await res.json();
+    const tierList = (await tierRes.json()).data;
     const buckets = data.aggregations['sterms#tier.tagFQN'].buckets;
 
     await waitForAllLoadersToDisappear(page);
 
-    for (const bucket of buckets) {
-      const normalizedKey = bucket.key
-        .split('.')
-        .map((seg: string) =>
-          seg ? seg.charAt(0).toUpperCase() + seg.slice(1) : seg
-        )
-        .join('.');
-
-      await expect(
-        page
-          .locator(`[data-menu-id$="-${normalizedKey}"]`)
-          .getByTestId('filter-count')
-      ).toHaveText(bucket.doc_count.toString());
+    // The following logic is required due to special case for tier filter
+    // where we are fetching the tier options from tag API and
+    // the count from aggregation API.
+    // So we need to match the bucket count with the corresponding tier option.
+    for (const tierItem of tierList) {
+      // Find the corresponding bucket for the tier
+      const bucket = buckets.find(
+        (item: { key: string }) =>
+          item.key.toLowerCase() === tierItem.fullyQualifiedName?.toLowerCase()
+      );
+      // Check if the tier in the dropdown has a corresponding bucket in elastic search response
+      if (!isUndefined(bucket)) {
+        await expect(
+          page
+            .locator(`[data-menu-id$="-${tierItem.fullyQualifiedName}"]`)
+            .getByTestId('filter-count')
+        ).toHaveText(bucket.doc_count.toString());
+      }
     }
 
     await clickOutside(page);