add wait_for_selector to the cells property and to_command_mode

mpacer · mpacer · commit 1889e560a80a · 2018-03-22T15:23:47.000-07:00
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("
