Selenium 高阶应用之WebDriverWait 和 expected_conditions

Seleniium 是相当不错的一个第三方测试框架，可惜目前国内已经无法访问其官网（FQ可以）。

      不知道大家是否有认真查看过selenium 的api，我是有认真学习过的。selenium 的api中包含有WebDriverWait  和 expected_conditions这两个高级应用。

WebDriverWait 类位于selenium.webdriver.support.ui下面的例子很简单，

Example:
from selenium.webdriver.support.ui import WebDriverWait 

element = WebDriverWait(driver, 10).until(lambda x: x.find_element_by_id("someId")) 

is_disappeared = WebDriverWait(driver, 30, 1, (ElementNotVisibleException)). 

                 until_not(lambda x: x.find_element_by_id("someId").is_displayed())

WebDriverWait 里面主要有两个方法，一个是until和until_not

那么我们可以这样用：

WebDriverWait(driver, 10).until(lambda x: x.find_element_by_id("某个按钮")).click()

  意思就是在10秒内等待某个按钮被定位到，咱们再去点击。就是每隔0.5秒内调用一下until里面的表达式或者方法函数，要么10秒内表达式执行成功，要么10秒后抛出超时异常。

无非就是一些预期条件。

结合上面的WebDriverWait，我们可以这么用：

from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

'''
10秒钟等待浏览器弹出的对话框，如果出现，就点击确定按钮
'''
WebDriverWait(chromedriver,10).until(EC.alert_is_present()).accept()

我们的自动化脚本跑得快慢或者出现异常，很大程度上取决于我们设定的等待时间，如果你还是习惯于time.sleep(5)这种方式的话，这将会浪费很多时间。
根据expected_conditions.py文件的写法，我们也可以定义一些自己的期待类，例如：

from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC


class must_get_url(object):
    '''
    必须到达的URL
       参数：
       url    - 必须到达的地址
    '''
    def __init__(self, url):
        self.url = url

    def __call__(self, driver):
        driver.get(self.url)
        return self.url == driver.current_url

options = webdriver.ChromeOptions()
options.add_argument("--test-type")
chromedriver =     webdriver.Chrome(r"E:MyProjectWebDriverchromedriver.exe",chrome_options=options)
WebDriverWait(chromedriver,10).until(must_get_url("https://www.baidu.com"))

场景

Expected Conditions的使用场景有2种

• 直接在断言中使用
• 与WebDriverWait配合使用，动态等待页面上元素出现或者消失

方法注释

先翻译一下这些方法的用法

• title_is: 判断当前页面的title是否精确等于预期

• title_contains: 判断当前页面的title是否包含预期字符串

• presence_of_element_located: 判断某个元素是否被加到了dom树里，并不代表该元素一定可见

• visibility_of_element_located: 判断某个元素是否可见.可见代表元素非隐藏，并且元素的宽和高都不等于0

• visibility_of: 跟上面的方法做一样的事情，只是上面的方法要传入locator，这个方法直接传定位到的element就好了

• presence_of_all_elements_located: 判断是否至少有1个元素存在于dom树中。举个例子，如果页面上有n个元素的class都是'column-md-3'，那么只要有1个元素存在，这个方法就返回True

• text_to_be_present_in_element: 判断某个元素中的text是否包含了预期的字符串

• text_to_be_present_in_element_value: 判断某个元素中的value属性是否包含了预期的字符串

• frame_to_be_available_and_switch_to_it: 判断该frame是否可以switch进去，如果可以的话，返回True并且switch进去，否则返回False

• invisibility_of_element_located: 判断某个元素中是否不存在于dom树或不可见

• element_to_be_clickable: 判断某个元素中是否可见并且是enable的，这样的话才叫clickable

• staleness_of: 等某个元素从dom树中移除，注意，这个方法也是返回True或False

• element_to_be_selected: 判断某个元素是否被选中了,一般用在下拉列表

• element_selection_state_to_be: 判断某个元素的选中状态是否符合预期

• element_located_selection_state_to_be: 跟上面的方法作用一样，只是上面的方法传入定位到的element，而这个方法传入locator

• alert_is_present: 判断页面上是否存在alert，这是个老问题，很多同学会问到

具体的例子

例子的代码在这里，并且是可以运行通过的。（找不到）

下面的代码演示了一些常见疑问

• 如何等待页面上的某个元素出现，然后再对这个元素进行操作
• 如何在unittest框架中所有的用例都共用1个浏览器实例，然后在全部用例结束后关闭浏览器

expected_conditions.py

#encoding:utf-8
# example of how to use https://github.com/SeleniumHQ/selenium/blob/master/py/selenium/webdriver/support/expected_conditions.py

from selenium import webdriver
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By

import unittest

# dr = webdriver.PhantomJS('phantomjs')
dr = webdriver.Firefox()
# dr = webdriver.Chrome()
url = 'http://www.baidu.com'
search_text_field_id = 'kw'
dr.get(url)

class ECExample(unittest.TestCase):

  def test_title_is(self):
    ''' 判断title是否符合预期 '''
    title_is_baidu = EC.title_is(u'百度一下，你就知道')
    self.assertTrue(title_is_baidu(dr))

  def test_titile_contains(self):
    ''' 判断title是否包含预期字符 '''
    title_should_contains_baidu = EC.title_contains(u'百度')
    self.assertTrue(title_should_contains_baidu(dr))

  def test_presence_of_element_located(self):
    ''' 判断element是否出现在dom树 '''
    locator = (By.ID, search_text_field_id)
    search_text_field_should_present = EC.visibility_of_element_located(locator)

    ''' 动态等待10s，如果10s内element加载完成则继续执行下面的代码，否则抛出异常 '''
    WebDriverWait(dr, 10).until(EC.presence_of_element_located(locator))
    WebDriverWait(dr, 10).until(EC.visibility_of_element_located(locator))

    self.assertTrue(search_text_field_should_present(dr))

  def test_visibility_of(self):
    search_text_field = dr.find_element_by_id(search_text_field_id)
    search_text_field_should_visible = EC.visibility_of(search_text_field)
    self.assertTrue(search_text_field_should_visible('yes'))

  def test_text_to_be_present_in_element(self):
    text_should_present = EC.text_to_be_present_in_element((By.NAME, 'tj_trhao123'), 'hao123')
    self.assertTrue(text_should_present(dr))


  @classmethod
  def tearDownClass(kls):
    print 'after all test'
    dr.quit()
    print 'quit dr'

if __name__ == '__main__':
  unittest.main()

分析

以title_is为例

class title_is(object):
    """An expectation for checking the title of a page.
    title is the expected title, which must be an exact match
    returns True if the title matches, false otherwise."""
    def __init__(self, title):
        self.title = title

    def __call__(self, driver):
        return self.title == driver.title

可以看到title_is实际上是1个class，其__call__方法被定义成是返回1个bool值。因此，一般的用法就是

# 实例化
the_instance = title_is('expected')
# 直接在实例上调用__call__
the_instance(dr) #return True or False