Fix flaky e2e tests

Author: mayagbarnes

e2e_playwright/e2e_utils.py

@@ -183,10 +183,13 @@ def open_settings_panel(page: Page, container: Locator) -> Locator:
     # If the panel is animating out, wait for it to fully disappear first
     if panel.count():
         try:
-            panel.wait_for(state="hidden", timeout=500)
+            panel.wait_for(state="hidden", timeout=1500)
         except Exception:
-            expect(panel).to_be_visible(timeout=5000)
-            return panel
+            try:
+                if panel.is_visible():
+                    return panel
+            except Exception:
+                pass
 
     button = container.get_by_test_id("toolbar-settings")
     button.scroll_into_view_if_needed()

e2e_playwright/pivot_table_interactions_test.py

@@ -535,21 +535,29 @@ def test_mixed_row_dimension_collapse_preserves_temporal_state(page_at_app: Page
         1, timeout=10000
     )
 
-    us_total_row = container.locator("tr").filter(has_text="US Total")
+    us_total_row = container.locator(
+        'tr[data-testid="pivot-subtotal-row"]:visible'
+    ).filter(has_text="US Total")
     expect(us_total_row).to_have_count(1, timeout=5000)
     us_group_toggle = us_total_row.get_by_test_id("pivot-group-toggle-US")
-    us_group_toggle.dispatch_event("click")
+    us_group_toggle.evaluate(
+        "el => { el.scrollIntoView({ block: 'center', inline: 'nearest' }); el.click(); }"
+    )
     expect(us_group_toggle).to_have_attribute("aria-expanded", "false", timeout=5000)
     expect(container.get_by_text("US Total")).to_be_visible(timeout=5000)
     expect(container.get_by_test_id("pivot-temporal-parent-row")).to_have_count(
         0, timeout=5000
     )
 
-    us_total_row = container.locator("tr").filter(has_text="US Total")
+    us_total_row = container.locator(
+        'tr[data-testid="pivot-subtotal-row"]:visible'
+    ).filter(has_text="US Total")
     expect(us_total_row).to_have_count(1, timeout=5000)
     us_group_toggle = us_total_row.get_by_test_id("pivot-group-toggle-US")
     expect(us_group_toggle).to_have_attribute("aria-expanded", "false", timeout=5000)
-    us_group_toggle.dispatch_event("click")
+    us_group_toggle.evaluate(
+        "el => { el.scrollIntoView({ block: 'center', inline: 'nearest' }); el.click(); }"
+    )
     expect(us_group_toggle).to_have_attribute("aria-expanded", "true", timeout=5000)
     expect(container.get_by_test_id("pivot-temporal-parent-row")).to_have_count(
         1, timeout=10000
@@ -1184,6 +1192,15 @@ def _drilldown_sort_button(panel, column_name: str):
     )
 
 
+def _drilldown_sort_header(panel, column_name: str):
+    """Return the sortable header cell for a drilldown column."""
+    return (
+        panel.get_by_test_id("drilldown-table")
+        .locator("th")
+        .filter(has_text=column_name)
+    )
+
+
 def test_drilldown_client_pagination_shows_controls(page_at_app: Page):
     """Client-only: clicking a cell with >500 records shows pagination controls."""
     page = page_at_app
@@ -1215,8 +1232,11 @@ def test_drilldown_client_sort_orders_full_result_before_pagination(page_at_app:
     page = page_at_app
     _, panel = _open_drilldown_with_pagination(page, "test_pivot_drilldown_pagination")
 
+    revenue_header = _drilldown_sort_header(panel, "Revenue")
     _drilldown_sort_button(panel, "Revenue").click()
+    expect(revenue_header).to_have_attribute("aria-sort", "ascending", timeout=5000)
     _drilldown_sort_button(panel, "Revenue").click()
+    expect(revenue_header).to_have_attribute("aria-sort", "descending", timeout=5000)
 
     # Fixture values for Alpha/2023 are 10..709 inclusive, so descending page 1
     # starts at 709 and descending page 2 starts at 209 after the first 500 rows.