From be9803de25b74db01f6de1fb6d8f2f63ca48f0d8 Mon Sep 17 00:00:00 2001 From: Benjamin Forehand Jr Date: Tue, 25 Feb 2020 11:18:53 -0600 Subject: [PATCH] Add missing page wait. (#2305) --- tests/integration/pages/experiment_design.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/integration/pages/experiment_design.py b/tests/integration/pages/experiment_design.py index db6e4c3dcd..efd1e6c66d 100644 --- 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