Improve tests

Author: medihack

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()

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()