Maintain compatibility with Selenium 2 if still being used

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -45,9 +45,15 @@ class MyTestClass(BaseCase):
 from seleniumbase.fixtures import page_utils
 from seleniumbase.fixtures import xpath_to_css
 from selenium.common.exceptions import (StaleElementReferenceException,
-                                        ElementNotInteractableException,
                                         TimeoutException,
                                         WebDriverException)
+from selenium.common import exceptions as selenium_exceptions
+try:
+    # Selenium 3
+    ENI_Exception = selenium_exceptions.ElementNotInteractableException
+except Exception:
+    # Selenium 2 (Keep compatibility with seleneium 2.53.6 if still being used)
+    ENI_Exception = selenium_exceptions.ElementNotSelectableException
 from selenium.webdriver.remote.webdriver import WebDriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.keys import Keys
@@ -97,8 +103,7 @@ def click(self, selector, by=By.CSS_SELECTOR,
         pre_action_url = self.driver.current_url
         try:
             element.click()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.05)
             element = page_actions.wait_for_element_visible(
@@ -129,8 +134,7 @@ def double_click(self, selector, by=By.CSS_SELECTOR,
             actions.move_to_element(element)
             actions.double_click(element)
             actions.perform()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.05)
             element = page_actions.wait_for_element_visible(
@@ -194,8 +198,7 @@ def click_link_text(self, link_text, timeout=settings.SMALL_TIMEOUT):
         pre_action_url = self.driver.current_url
         try:
             element.click()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.05)
             element = self.wait_for_link_text_visible(
@@ -251,8 +254,7 @@ def click_partial_link_text(self, partial_link_text,
         pre_action_url = self.driver.current_url
         try:
             element.click()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.05)
             element = self.wait_for_partial_link_text(
@@ -278,8 +280,7 @@ def get_text(self, selector, by=By.CSS_SELECTOR,
             self.driver, selector, by, timeout)
         try:
             element_text = element.text
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.06)
             element = page_actions.wait_for_element_visible(
@@ -299,8 +300,7 @@ def get_attribute(self, selector, attribute, by=By.CSS_SELECTOR,
             self.driver, selector, by, timeout)
         try:
             attribute_value = element.get_attribute(attribute)
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.06)
             element = page_actions.wait_for_element_present(
@@ -361,8 +361,7 @@ def add_text(self, selector, new_value, by=By.CSS_SELECTOR,
                 element.send_keys(Keys.RETURN)
                 if settings.WAIT_FOR_RSC_ON_PAGE_LOADS:
                     self.wait_for_ready_state_complete()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.06)
             element = self.wait_for_element_visible(
@@ -411,8 +410,7 @@ def update_text_value(self, selector, new_value, by=By.CSS_SELECTOR,
             self._scroll_to_element(element)
         try:
             element.clear()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.06)
             element = self.wait_for_element_visible(
@@ -429,8 +427,7 @@ def update_text_value(self, selector, new_value, by=By.CSS_SELECTOR,
                 element.send_keys(Keys.RETURN)
                 if settings.WAIT_FOR_RSC_ON_PAGE_LOADS:
                     self.wait_for_ready_state_complete()
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.06)
             element = self.wait_for_element_visible(
@@ -631,8 +628,7 @@ def scroll_to(self, selector, by=By.CSS_SELECTOR,
             selector, by=by, timeout=timeout)
         try:
             self._scroll_to_element(element)
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.05)
             element = self.wait_for_element_visible(
@@ -1274,8 +1270,7 @@ def _pick_select_option(self, dropdown_selector, option,
                 Select(element).select_by_value(option)
             else:
                 Select(element).select_by_visible_text(option)
-        except (StaleElementReferenceException,
-                ElementNotInteractableException):
+        except (StaleElementReferenceException, ENI_Exception):
             self.wait_for_ready_state_complete()
             time.sleep(0.05)
             element = self.find_element(