add wait_for_selector to the cells property and to_command_mode

jupyter · Mar 22, 2018 · 1889e56 · 1889e56
1 parent 78254f0
commit 1889e56
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/notebook/tests/selenium/utils.py b/notebook/tests/selenium/utils.py
@@ -32,6 +32,10 @@ def body(self):
 
     @property
     def cells(self):
+        """Gets all cells once they are visible.
+        
+        """
+        wait_for_selector(self.browser, ".cell")
         return self.browser.find_elements_by_class_name("cell")
 
 
@@ -48,7 +52,9 @@ def remove_safety_check(self):
 
     def to_command_mode(self):
         """Changes us into command mode on currently focused cell
+        
         """
+        wait_for_selector(self.browser, ".cell")
         self.browser.switch_to.active_element.send_keys(Keys.ESCAPE)
         self.browser.execute_script("return Jupyter.notebook.handle_command_mode("
                                        "Jupyter.notebook.get_cell("