Add missing page wait. (#2305)

mozilla · Feb 25, 2020 · be9803d · be9803d
1 parent 2296fd0
commit be9803d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/integration/pages/experiment_design.py b/tests/integration/pages/experiment_design.py
@@ -23,6 +23,13 @@ class DesignPage(Base):
     )
     _page_wait_locator = (By.CSS_SELECTOR, "body.page-edit-variants")
 
+    def wait_for_page_to_load(self):
+        self.wait.until(
+            lambda _: self.find_element(
+                *self._firefox_pref_name_locator).is_displayed()
+            )
+        return self
+
     def create_new_branch(self):
         """Creates a new branch."""
         num_of_branches = len(self.current_branches) + 1