diff options
Diffstat (limited to 'tests/functional/features/steps/common.py')
-rw-r--r-- | tests/functional/features/steps/common.py | 138 |
1 files changed, 138 insertions, 0 deletions
diff --git a/tests/functional/features/steps/common.py b/tests/functional/features/steps/common.py new file mode 100644 index 00000000..91858f85 --- /dev/null +++ b/tests/functional/features/steps/common.py @@ -0,0 +1,138 @@ +import time + +from selenium.common.exceptions import ( + StaleElementReferenceException, + TimeoutException) +from selenium.webdriver.common.by import By +from selenium.webdriver.support import expected_conditions as EC +from selenium.webdriver.support.wait import WebDriverWait + +TIMEOUT_IN_S = 10 + +DEFAULT_IMPLICIT_WAIT_TIMEOUT_IN_S = 10 + + +def wait_until_element_is_invisible_by_locator(context, locator_tuple, + timeout=TIMEOUT_IN_S): + wait = WebDriverWait(context.browser, timeout) + wait.until(EC.invisibility_of_element_located(locator_tuple)) + + +def wait_for_loading_to_finish(context, timeout=TIMEOUT_IN_S): + wait_until_element_is_invisible_by_locator( + context, (By.ID, 'loading'), timeout) + + +def wait_for_user_alert_to_disapear(context, timeout=TIMEOUT_IN_S): + wait_until_element_is_invisible_by_locator( + context, (By.ID, 'user-alerts'), timeout) + + +def _wait_until_elements_are_visible_by_locator(context, locator_tuple, + timeout=TIMEOUT_IN_S): + wait = WebDriverWait(context.browser, timeout) + wait.until(EC.presence_of_all_elements_located(locator_tuple)) + return context.browser.find_elements(locator_tuple[0], locator_tuple[1]) + + +def _wait_until_element_is_visible_by_locator(context, locator_tuple, + timeout=TIMEOUT_IN_S): + wait = WebDriverWait(context.browser, timeout) + wait.until(EC.visibility_of_element_located(locator_tuple)) + return context.browser.find_element(locator_tuple[0], locator_tuple[1]) + + +def wait_for_condition(context, predicate_func, timeout=TIMEOUT_IN_S, + poll_frequency=0.1): + wait = WebDriverWait(context.browser, timeout, + poll_frequency=poll_frequency) + wait.until(predicate_func) + + +def fill_by_xpath(context, xpath, text): + field = context.browser.find_element_by_xpath(xpath) + field.send_keys(text) + + +def fill_by_css_selector(context, css_selector, text, timeout=TIMEOUT_IN_S): + field = find_element_by_css_selector(context, css_selector, + timeout=timeout) + field.send_keys(text) + + +def take_screenshot(context, filename): + context.browser.save_screenshot(filename) + + +def page_has_css(context, css): + try: + find_element_by_css_selector(context, css) + return True + except TimeoutException: + return False + + +def find_element_by_xpath(context, xpath): + return _wait_until_element_is_visible_by_locator( + context, (By.XPATH, xpath)) + + +def find_element_by_id(context, id): + return _wait_until_element_is_visible_by_locator(context, (By.ID, id)) + + +def find_element_by_css_selector(context, css_selector, timeout=TIMEOUT_IN_S): + return _wait_until_element_is_visible_by_locator( + context, (By.CSS_SELECTOR, css_selector), timeout=timeout) + + +def find_element_by_class_name(context, class_name): + return _wait_until_element_is_visible_by_locator( + context, (By.CLASS_NAME, class_name)) + + +def find_elements_by_css_selector(context, css_selector, timeout=TIMEOUT_IN_S): + return _wait_until_elements_are_visible_by_locator( + context, (By.CSS_SELECTOR, css_selector), timeout=timeout) + + +def find_elements_by_xpath(context, xpath, timeout=TIMEOUT_IN_S): + return _wait_until_elements_are_visible_by_locator( + context, (By.XPATH, xpath), timeout=timeout) + + +def find_element_containing_text(context, text, element_type='*'): + return find_element_by_xpath( + context, "//%s[contains(.,'%s')]" % (element_type, text)) + + +def element_should_have_content(context, css_selector, content): + e = find_element_by_css_selector(context, css_selector) + assert e.text == content + + +def wait_until_button_is_visible(context, title, timeout=TIMEOUT_IN_S): + wait = WebDriverWait(context.browser, timeout) + locator_tuple = (By.XPATH, ("//%s[contains(.,'%s')]" % ('button', title))) + wait.until(EC.visibility_of_element_located(locator_tuple)) + + +def execute_ignoring_staleness(func, timeout=TIMEOUT_IN_S): + end_time = time.time() + timeout + while time.time() <= end_time: + try: + return func() + except StaleElementReferenceException: + pass + raise TimeoutException( + 'did not solve stale state until timeout %f' % timeout) + + +def click_button(context, title, element='button'): + button = find_element_containing_text(context, title, element_type=element) + button.click() + + +def reply_subject(context): + e = find_element_by_css_selector(context, '#reply-subject') + return e.text |