repo-mirrors/dlt
1
0
Fork 0
mirror of https://github.com/dlt-hub/dlt.git synced 2025-12-17 19:31:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

wait in marimo UI in test_e2e.py to for schema selection (#3453)

Browse Source 
Should fix flaky test
This commit is contained in:
Anton Burnashev
2025-12-10 14:32:18 +01:00
committed by GitHub
parent be9aa1bf03
commit 1c49c2081c
1 changed files with 35 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
tests/e2e/helpers/dashboard/test_e2e.py
View File

@@ -119,6 +119,20 @@ def test_multi_schema_selection(page: Page, multi_schema_pipeline: Any):
expect(page.get_by_text("name: fruitshop_customers").nth(1)).to_be_attached()
def _select_schema_and_verify(
schema_selector: Any,
schema_name: str,
expected: str,
not_expected: set[str],
):
schema_selector.select_option(schema_name)
expect(schema_selector).to_have_value(schema_name)
# allow marimo reactivity to process
page.wait_for_timeout(500)
expect(page.get_by_text(expected, exact=True).nth(0)).to_be_visible(timeout=15000)
for table in not_expected:
expect(page.get_by_text(table, exact=True)).to_have_count(0, timeout=10000)
# select each schema and see if the right tables are shown
# do this both for schema and data section
for section in ["schema", "data"]:
@@ -126,27 +140,30 @@ def test_multi_schema_selection(page: Page, multi_schema_pipeline: Any):
# NOTE: this is using unspecific selector and may select other dropdowns id present (?)
schema_selector = page.get_by_test_id("marimo-plugin-dropdown")
schema_selector.select_option("fruitshop_customers")
expect(schema_selector).to_have_value("fruitshop_customers")
all_tables = {"customers", "inventory", "purchases"}
_select_schema_and_verify(
schema_selector,
"fruitshop_customers",
expected="customers",
not_expected=all_tables - {"customers"},
)
schema_selector.scroll_into_view_if_needed()
expect(page.get_by_text("customers", exact=True).nth(0)).to_be_visible()
expect(page.get_by_text("inventory", exact=True)).to_have_count(0)
expect(page.get_by_text("purchases", exact=True)).to_have_count(0)
_select_schema_and_verify(
schema_selector,
"fruitshop_inventory",
expected="inventory",
not_expected=all_tables - {"inventory"},
)
schema_selector.select_option("fruitshop_inventory")
expect(schema_selector).to_have_value("fruitshop_inventory")
expect(page.get_by_text("inventory", exact=True).nth(0)).to_be_visible()
expect(page.get_by_text("customers", exact=True)).to_have_count(0)
expect(page.get_by_text("purchases", exact=True)).to_have_count(0)
schema_selector.select_option("fruitshop_purchases")
expect(schema_selector).to_have_value("fruitshop_purchases")
expect(page.get_by_text("purchases", exact=True).nth(0)).to_be_visible()
expect(page.get_by_text("inventory", exact=True)).to_have_count(0)
expect(page.get_by_text("customers", exact=True)).to_have_count(0)
_select_schema_and_verify(
schema_selector,
"fruitshop_purchases",
expected="purchases",
not_expected=all_tables - {"purchases"},
)
_close_sections(page)
# make sure schema selector removed from page