Fix end-to-end tests

treiher · Oct 31, 2023 · 8df6155 · 8df6155
1 parent 4065fab
commit 8df6155
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/tests/e2e/page.py b/tests/e2e/page.py
@@ -250,13 +250,11 @@ def title(self) -> str:
     def url(self) -> str:
         raise NotImplementedError
 
-    def load(self, *, accept_unsaved_changes: bool = False) -> None:
-        self._driver.get(f"http://{HOST}:{PORT}/#{self.url}")
-
-        if accept_unsaved_changes:
-            alert = self.wait_for_alert()
-            alert.accept()
+    def refresh(self) -> None:
+        self._driver.refresh()
 
+    def load(self) -> None:
+        self._driver.get(f"http://{HOST}:{PORT}/#{self.url}")
         wait(self._driver).until(
             EC.presence_of_element_located((By.XPATH, "/html/body/div[@id='app']/nav/div/div"))
         )

diff --git a/tests/e2e/web_test.py b/tests/e2e/web_test.py
@@ -515,7 +515,9 @@ def test_training_session_change_entries(driver: webdriver.Chrome) -> None:
 
     assert page.get_sets() == [new_values, *sets[1:]]
 
-    page.load(accept_unsaved_changes=True)
+    page.refresh()
+    alert = page.wait_for_alert()
+    alert.accept()
 
     page.wait_for_title(str(workout.date))
     assert page.get_sets() == sets
@@ -561,7 +563,9 @@ def test_training_session_change_notes(driver: webdriver.Chrome) -> None:
     assert page.get_sets() == sets
     assert page.get_notes() == new_notes
 
-    page.load(accept_unsaved_changes=True)
+    page.refresh()
+    alert = page.wait_for_alert()
+    alert.accept()
 
     page.wait_for_title(str(workout.date))
     assert page.get_sets() == sets