Replaced DOM retrieval function with get_element().

New function waits for the element to appear before selecting it.

Author: freskoulix

codebender_testing/utils.py

@@ -861,7 +861,7 @@ def execute_script(self, script, *deps):
 
     def create_sketch(self, privacy, name, description):
         """Creates a sketch with a given name"""
-        createSketchBtn = self.driver.find_element_by_id('create_sketch_btn')
+        createSketchBtn = self.get_element(By.ID, 'create_sketch_btn')
         createSketchBtn.click()
         WebDriverWait(self.driver, VERIFY_TIMEOUT).until(
                 expected_conditions.visibility_of_element_located(
@@ -903,8 +903,7 @@ def change_name(self, name):
         nameField.send_keys(Keys.ENTER)
 
     def change_name_editor(self, name):
-        print "inside change name"
-        sketchHeading = self.driver.find_element_by_id('editor_heading_project_name')
+        sketchHeading = self.get_element(By.ID, 'editor_heading_project_name')
         sketchHeading.click()
         renameInput = '#editor_heading_project_name input'
         headingInput = self.get_element(By.CSS_SELECTOR, renameInput)
@@ -955,7 +954,6 @@ def change_short_description_editor(self, description):
             )
         )
 
-
     def check_iframe(self):
         """Returns the contents of an iframe [project_name, user_name, sketch_contents]"""
         self.driver.switch_to_frame(self.driver.find_element_by_tag_name('iframe'))