Fix compatibility issues with Safari

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -291,20 +291,19 @@ def click(
         pre_action_url = self.driver.current_url
         pre_window_count = len(self.driver.window_handles)
         try:
-            if self.browser == "ie" and by == By.LINK_TEXT:
-                # An issue with clicking Link Text on IE means using jquery
+            if (
+                by == By.LINK_TEXT
+                and (
+                    self.browser == "ie" or self.browser == "safari"
+                )
+            ):
                 self.__jquery_click(selector, by=by)
-            elif self.browser == "safari":
-                if by == By.LINK_TEXT:
-                    self.__jquery_click(selector, by=by)
-                else:
-                    self.__js_click(selector, by=by)
             else:
                 href = None
                 new_tab = False
                 onclick = None
                 try:
-                    if self.headless and element.tag_name == "a":
+                    if self.headless and element.tag_name.lower() == "a":
                         # Handle a special case of opening a new tab (headless)
                         href = element.get_attribute("href").strip()
                         onclick = element.get_attribute("onclick")
@@ -343,11 +342,8 @@ def click(
                 self.__scroll_to_element(element, selector, by)
             except Exception:
                 pass
-            if self.browser == "safari":
-                if by == By.LINK_TEXT:
-                    self.__jquery_click(selector, by=by)
-                else:
-                    self.__js_click(selector, by=by)
+            if self.browser == "safari" and by == By.LINK_TEXT:
+                self.__jquery_click(selector, by=by)
             else:
                 element.click()
         except ENI_Exception:
@@ -364,7 +360,7 @@ def click(
             new_tab = False
             onclick = None
             try:
-                if element.tag_name == "a":
+                if element.tag_name.lower() == "a":
                     # Handle a special case of opening a new tab (non-headless)
                     href = element.get_attribute("href").strip()
                     onclick = element.get_attribute("onclick")
@@ -1449,10 +1445,11 @@ def get_text(self, selector, by="css selector", timeout=None):
         try:
             element_text = element.text
             if self.browser == "safari":
-                element_text = element.get_attribute("innerText")
-            if element.tag_name == "input":
-                element_text = element.get_property("value")
-            if element.tag_name == "textarea":
+                if element.tag_name.lower() in ["input", "textarea"]:
+                    element_text = element.get_attribute("value")
+                else:
+                    element_text = element.get_attribute("innerText")
+            elif element.tag_name.lower() in ["input", "textarea"]:
                 element_text = element.get_property("value")
         except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
@@ -1462,8 +1459,11 @@ def get_text(self, selector, by="css selector", timeout=None):
             )
             element_text = element.text
             if self.browser == "safari":
-                element_text = element.get_attribute("innerText")
-            if element.tag_name == "input":
+                if element.tag_name.lower() in ["input", "textarea"]:
+                    element_text = element.get_attribute("value")
+                else:
+                    element_text = element.get_attribute("innerText")
+            elif element.tag_name.lower() in ["input", "textarea"]:
                 element_text = element.get_property("value")
         return element_text
 
@@ -2219,9 +2219,6 @@ def hover_and_click(
             if self.mobile_emulator:
                 # On mobile, click to hover the element
                 dropdown_element.click()
-            elif self.browser == "safari":
-                # Use the workaround for hover-clicking on Safari
-                raise Exception("This Exception will be caught.")
             else:
                 page_actions.hover_element(self.driver, dropdown_element)
         except Exception:
@@ -2264,6 +2261,12 @@ def hover_and_click(
             )
         ):
             self.__switch_to_newest_window_if_not_blank()
+        elif self.browser == "safari":
+            # Release the hover by hovering elsewhere
+            try:
+                page_actions.hover_on_element(self.driver, "body")
+            except Exception:
+                pass
         if self.demo_mode:
             if self.driver.current_url != pre_action_url:
                 self.__demo_mode_pause_if_active()
@@ -2625,10 +2628,11 @@ def get_select_options(
             raise Exception("The attribute must be in %s" % allowed_attributes)
         selector, by = self.__recalculate_selector(selector, by)
         element = self.wait_for_element(selector, by=by, timeout=timeout)
-        if element.tag_name != "select":
+        if element.tag_name.lower() != "select":
             raise Exception(
                 'Element tag_name for get_select_options(selector) must be a '
-                '"select"! Actual tag_name found was: "%s"' % element.tag_name
+                '"select"! Actual tag_name found was: "%s"'
+                % element.tag_name.lower()
             )
         if by != "css selector":
             selector = self.convert_to_css_selector(selector, by=by)
@@ -2712,9 +2716,13 @@ def load_html_string(self, html_string, new_page=True):
         html_string = html_string.replace("\\ ", " ")
 
         if new_page:
-            self.open("data:text/html,")
+            self.open("data:text/html,<head></head><body></body>")
         inner_head = """document.getElementsByTagName("head")[0].innerHTML"""
         inner_body = """document.getElementsByTagName("body")[0].innerHTML"""
+        try:
+            self.wait_for_element_present("body", timeout=1)
+        except Exception:
+            pass
         if not found_body:
             self.execute_script('''%s = \"%s\"''' % (inner_body, html_string))
         elif found_body and not found_head:
@@ -5097,13 +5105,16 @@ def slow_scroll_to(self, selector, by="css selector", timeout=None):
             original_selector, by=original_by, timeout=timeout
         )
         try:
-            scroll_distance = js_utils.get_scroll_distance_to_element(
-                self.driver, element
-            )
-            if abs(scroll_distance) > constants.Values.SSMD:
-                self.__jquery_slow_scroll_to(selector, by)
+            if self.browser != "safari":
+                scroll_distance = js_utils.get_scroll_distance_to_element(
+                    self.driver, element
+                )
+                if abs(scroll_distance) > constants.Values.SSMD:
+                    self.__jquery_slow_scroll_to(selector, by)
+                else:
+                    self.__slow_scroll_to_element(element)
             else:
-                self.__slow_scroll_to_element(element)
+                self.__jquery_slow_scroll_to(selector, by)
         except Exception:
             self.wait_for_ready_state_complete()
             time.sleep(0.12)
@@ -6716,7 +6727,7 @@ def set_text(self, selector, text, by="css selector", timeout=None):
         element = page_actions.wait_for_element_present(
             self.driver, selector, by, timeout
         )
-        if element.tag_name == "input" or element.tag_name == "textarea":
+        if element.tag_name.lower() in ["input", "textarea"]:
             self.js_update_text(selector, text, by=by, timeout=timeout)
         else:
             self.set_text_content(selector, text, by=by, timeout=timeout)
@@ -6736,7 +6747,7 @@ def set_text_content(
         element = page_actions.wait_for_element_present(
             self.driver, selector, by, timeout
         )
-        if element.tag_name == "input" or element.tag_name == "textarea":
+        if element.tag_name.lower() in ["input", "textarea"]:
             self.js_update_text(selector, text, by=by, timeout=timeout)
             return
         orginal_selector = selector
@@ -12250,13 +12261,16 @@ def __demo_mode_highlight_if_active(self, selector, by):
                 selector, by=by, timeout=settings.SMALL_TIMEOUT
             )
             try:
-                scroll_distance = js_utils.get_scroll_distance_to_element(
-                    self.driver, element
-                )
-                if abs(scroll_distance) > constants.Values.SSMD:
-                    self.__jquery_slow_scroll_to(selector, by)
+                if self.browser != "safari":
+                    scroll_distance = js_utils.get_scroll_distance_to_element(
+                        self.driver, element
+                    )
+                    if abs(scroll_distance) > constants.Values.SSMD:
+                        self.__jquery_slow_scroll_to(selector, by)
+                    else:
+                        self.__slow_scroll_to_element(element)
                 else:
-                    self.__slow_scroll_to_element(element)
+                    self.__jquery_slow_scroll_to(selector, by)
             except (StaleElementReferenceException, ENI_Exception):
                 self.wait_for_ready_state_complete()
                 time.sleep(0.12)
@@ -12290,13 +12304,16 @@ def __highlight_with_assert_success(
             selector, by=by, timeout=settings.SMALL_TIMEOUT
         )
         try:
-            scroll_distance = js_utils.get_scroll_distance_to_element(
-                self.driver, element
-            )
-            if abs(scroll_distance) > constants.Values.SSMD:
-                self.__jquery_slow_scroll_to(selector, by)
+            if self.browser != "safari":
+                scroll_distance = js_utils.get_scroll_distance_to_element(
+                    self.driver, element
+                )
+                if abs(scroll_distance) > constants.Values.SSMD:
+                    self.__jquery_slow_scroll_to(selector, by)
+                else:
+                    self.__slow_scroll_to_element(element)
             else:
-                self.__slow_scroll_to_element(element)
+                self.__jquery_slow_scroll_to(selector, by)
         except Exception:
             self.wait_for_ready_state_complete()
             time.sleep(0.12)

seleniumbase/fixtures/page_actions.py

@@ -127,10 +127,13 @@ def is_text_visible(driver, text, selector, by="css selector", browser=None):
     try:
         element = driver.find_element(by=by, value=selector)
         element_text = element.text
-        if element.tag_name == "input" or element.tag_name == "textarea":
+        if browser == "safari":
+            if element.tag_name.lower() in ["input", "textarea"]:
+                element_text = element.get_attribute("value")
+            else:
+                element_text = element.get_attribute("innerText")
+        elif element.tag_name.lower() in ["input", "textarea"]:
             element_text = element.get_property("value")
-        elif browser == "safari":
-            element_text = element.get_attribute("innerText")
         return element.is_displayed() and text in element_text
     except Exception:
         return False
@@ -477,7 +480,10 @@ def wait_for_text_visible(
         try:
             element = driver.find_element(by=by, value=selector)
             is_present = True
-            if element.tag_name == "input" or element.tag_name == "textarea":
+            if (
+                element.tag_name.lower() in ["input", "textarea"]
+                and browser != "safari"
+            ):
                 if (
                     element.is_displayed()
                     and text in element.get_property("value")
@@ -489,14 +495,17 @@ def wait_for_text_visible(
                     element = None
                     raise Exception()
             elif browser == "safari":
+                text_attr = "innerText"
+                if element.tag_name.lower() in ["input", "textarea"]:
+                    text_attr = "value"
                 if (
                     element.is_displayed()
-                    and text in element.get_attribute("innerText")
+                    and text in element.get_attribute(text_attr)
                 ):
                     return element
                 else:
                     if element.is_displayed():
-                        full_text = element.get_attribute("innerText")
+                        full_text = element.get_attribute(text_attr)
                         full_text = full_text.strip()
                     element = None
                     raise Exception()
@@ -582,7 +591,7 @@ def wait_for_exact_text_visible(
         try:
             element = driver.find_element(by=by, value=selector)
             is_present = True
-            if element.tag_name == "input" or element.tag_name == "textarea":
+            if element.tag_name.lower() in ["input", "textarea"]:
                 if (
                     element.is_displayed()
                     and text.strip() == element.get_property("value").strip()
@@ -594,13 +603,16 @@ def wait_for_exact_text_visible(
                     element = None
                     raise Exception()
             elif browser == "safari":
+                text_attr = "innerText"
+                if element.tag_name.lower() in ["input", "textarea"]:
+                    text_attr = "value"
                 if element.is_displayed() and (
-                    text.strip() == element.get_attribute("innerText").strip()
+                    text.strip() == element.get_attribute(text_attr).strip()
                 ):
                     return element
                 else:
                     if element.is_displayed():
-                        actual_text = element.get_attribute("innerText")
+                        actual_text = element.get_attribute(text_attr)
                         actual_text = actual_text.strip()
                     element = None
                     raise Exception()