e2e: add utility to wait for URL

2020-06-26 16:21:59 +02:00
parent cc7e4ad0e2
commit c1b2093cf7
2 changed files with 5 additions and 0 deletions
--- a/e2e/utils.py
+++ b/e2e/utils.py
@ -66,6 +66,10 @@ class SeleniumTestCase(StaticLiveServerTestCase):
        self.driver.quit()
        super().tearDown()

+    def wait_for_url(self, desired_url):
+        """Wait until URL is `desired_url`."""
+        self.wait.until(lambda driver: driver.current_url == desired_url)
+
    def url(self, view, **kwargs) -> str:
        """reverse `view` with `**kwargs` into full URL using live_server_url"""
        return self.live_server_url + reverse(view, kwargs=kwargs)