Improve tests

openradx · Dec 25, 2024 · efcfef4 · efcfef4
1 parent 7055fb6
commit efcfef4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/adit/batch_query/tests/integration/test_batch_query.py b/adit/batch_query/tests/integration/test_batch_query.py
@@ -4,7 +4,6 @@
     add_permission,
     add_user_to_group,
     create_and_login_example_user,
-    poll_locator,
     run_worker_once,
 )
 from playwright.sync_api import Page, expect
@@ -49,9 +48,10 @@ def test_urgent_batch_query_with_dimse_server(page: Page, live_server: LiveServe
     page.locator('input:has-text("Create job")').click()
 
     run_worker_once()
+    page.reload()
 
     # Assert
-    expect(poll_locator(page.locator('dl:has-text("Success")'))).to_be_visible()
+    expect(page.locator('dl:has-text("Success")')).to_be_visible()
 
 
 @pytest.mark.integration
@@ -83,6 +83,7 @@ def test_urgent_batch_query_with_dicomweb_server(page: Page, live_server: LiveSe
     page.locator('input:has-text("Create job")').click()
 
     run_worker_once()
+    page.reload()
 
     # Assert
-    expect(poll_locator(page.locator('dl:has-text("Success")'))).to_be_visible()
+    expect(page.locator('dl:has-text("Success")')).to_be_visible()
diff --git a/adit/batch_transfer/tests/integration/test_batch_transfer.py b/adit/batch_transfer/tests/integration/test_batch_transfer.py
@@ -4,7 +4,6 @@
     add_permission,
     add_user_to_group,
     create_and_login_example_user,
-    poll_locator,
     run_worker_once,
 )
 from playwright.sync_api import Page, expect
@@ -54,9 +53,10 @@ def test_unpseudonymized_urgent_batch_transfer_with_dimse_server(
     page.locator('input:has-text("Create job")').click()
 
     run_worker_once()
+    page.reload()
 
     # Assert
-    expect(poll_locator(page.locator('dl:has-text("Success")'))).to_be_visible()
+    expect(page.locator('dl:has-text("Success")')).to_be_visible()
 
 
 @pytest.mark.integration
@@ -93,6 +93,7 @@ def test_unpseudonymized_urgent_batch_transfer_with_dicomweb_server(
     page.locator('input:has-text("Create job")').click()
 
     run_worker_once()
+    page.reload()
 
     # Assert
-    expect(poll_locator(page.locator('dl:has-text("Success")'))).to_be_visible()
+    expect(page.locator('dl:has-text("Success")')).to_be_visible()