python selenium2示例 - 同步机制

前言

在使用python selenium2进行自动化测试实践的过程中，经常会遇到元素定位不到，弹出框定位不到等等各种定位不到的情况，在大多数的情况下，无非是以下两种情况：

1、有frame存在，定位前，未switch到对应的frame内

2、元素未加载完毕（从界面看已经显示），但DOM树还在load状态或在加载js

那对于这类情况，怎么解决呢？

通俗的讲法： 等待。

高大上点：解决自动化测试代码与浏览器加载渲染之间的同步问题。

下面我们分段讲述各种处理方式：

一 强制等待

这种方式简单粗暴直接有效，不足的地方就是不够灵活。下面看下代码片段：

#_*_ coding:utf-8 _*_

__author__ = '苦叶子'

from selenium import webdriver

from time import sleep   # 注意

if __name__ == '__main__':

    driver = webdrivrer.Firefox()

    driver.get('http://www.testingunion.com')

    sleep(3)    # 强制等待3s在执行下一步

    print u"当前url:  ", driver.current

    sleep(3)   # 强制等待3s在执行下一步

    driver.quit() 

注： 请注意加粗有删除线的代码行，用于实现强制等待

二、隐性等待

webdriver提供了基础的同步方法，隐性等待implicitly_wait(xx),该方式的意义是：不论业务代码运行在那一步，都需要等待webdriver  xx秒，如果在等待的xx秒内 webdriver完成了对应的动作，则业务代码和webdriver都正常继续执行；如果超过了xx秒，webdriver还未完成对应的动作，则业务代码继续执行，而webdriver则会抛出异常（例如timeout或元素未找到等等异常），请看代码实现片段：

#_*_ coding:utf-8 _*_

__author__ = '苦叶子'

from selenium import webdriver

if __name__ == '__main__':

    driver = webdrivrer.Firefox()

    driver.implicitly_wait(30)   # 隐性等待，最长等30s

    driver.get('http://www.testingunion.com')

    print u"当前url： ", driver.current_url
 
    driver.quit()

注：上述代码中加粗删除线的代码通过调用webdriver提供的标准隐性等待方式来实现一种同步机制。其设置的是一个最长等待时间，如果在规定的时间未完成，则进入下一步。

不足：在实践中，通常我们需要操作的元素已经显示出来，但因网络或其他因素，浏览器一直处于加载个别js或图片或其他资源时，隐性等待模式下，这时会依旧处于等待状态直至页面全部加载完毕才能进入下一步。那有没有更好的办法呢？当然是有的，请参见下一方式。

重要：隐性等待是全局性质的，只需在driver实例化后，设置一次即可。

在实践中，经常见到新手把隐性等待当做sleep来使用，在每个步骤后都用一次。

三、显性等待

更为强大的方式是显性等待来实现同步机制，需要WebDriverWait类，辅以until()或until_not()方法，根据判断条件进行灵活的同步，它的主要机制是：程序在规定的时间内每个xx秒看一下判断条件是否成立，如果成立则执行下一步，否则继续等待，直至超过设置的最长时间，然后抛出异常。请看具体的代码片段：

#_*_ coding:utf-8 _*_

__author__ = '苦叶子'

from selenium import webdriver

from selenium.webdriver.support.wait import WebDriverWait

from selenium.webdriver.support import expected_conditions as EC

from selenium.webdriver.common.by import By

if __name__ == '__main__':

   driver = webdrivrer.Firefox()

   # 隐性等待和显性等待可以同时用，要注意的是：最大等待时间取决两者之间的大值

   driver.implicitly_wait(10)  

   driver.get('http://www.testingunion.com')

   locator = (By.LINK_TEXT,  u'webdriver')

   try:

        # 在最长20s内，每个0.5秒去检查locator是否存在，如果存在则进入下一步

        WebDriverWait(driver, 20, 0.5).until(EC.presence_of_located(locator))

        # 提取该文本对应的url，并打印出来

        print driver.find_element_by_link_text(u'webdriver').get_attribute('href')

   finally:

       print u"异常了"

   driver.quit()

在本示例中，我们设置了隐性等待和显性等待，在其他的操作中隐性等待起决定性作用，在示例中的WebDriverWait设置了显性等待的地方，则显性等待起决定性作用，但要注意的是：最长等待时间取决于两者之间的大值，所以这里显性等待的最长时间为20s。

在这里我们主要用到了WebDriverWait类和expected_conditions模块，让我们近距离的接触下它们。

WebDriverWait类

定义实现在wait模块中，实现了webdriver的显性等待机制，先看下它有哪些参数和方法：

selenium.webdriver.support.wait.WebDriverWait（类）

__init__(self,   

driver,  # 传入实例化的webdriver对象

timeout,  # 超时时间，等待的最大时间（需考虑同时考虑隐性等待时间）

poll_frequency=POLL_FREQUENCY,  # 调用until或until_not方法的间隔时间，上例为0.5s

ignored_exceptions=None  #指定忽略的异常，如果指定了要忽略的异常类型，则在调用until或until_not过程中，捕获该类异常时不中断代码，继续等待。默认只有NoSuchElementException

)

---

until(self,
method,    # 在等待期间，每个一段时间调用这个传入的方法，直到返回值为false
message='' # 如果超时，则抛出TimeoutException，将message传入给异常
)
until_not 与until相反，until是当某个元素满足某种条件时（出现、存在等等）继续执行；until_not则是当某个元素不满足某种条件时继续执行，参数含义相同

特别注意：

很多时候大家在使用until或until_not时，会将一个WebElement对象传入给method，如下：

WebDriverWait(driver, 10).until(driver.find_element_by_id('kw'))     # 这是错误的用法 

这里的参数一定要是可调用的，这个对象一定要有__call__()方法，否则会抛出异常：

TypeError： 'xxx' object is not callable

在这里，也可以用selenium2提供的expected_conditions模块中提供的各种条件，也可用WebElement中的is_displayed()，is_selected()，is_enabled()方法或是自己封装的方法均可。下面我们再看看selenium2提供了哪些条件，如图所示：

结束语

      本文就python selenium2三种同步解决方式进行了较为详细的说明，这是使用selenium2进行自动化测试实践的必备技能，希望对大家有所帮助，有任何问题请关注公众号号，直接回复消息进行交流。