在UI自动化测试中,必然会遇到环境不稳定,网络慢的情况,这时如果你不做任何处理的话,代码会由于没有找到元素,而报错。这时我们就要用到wait(等待),而在Selenium中,我们可以用到一共三种等待,每一种等待都有自己的优点或缺点,我们需要选择最优的等待方式。下面我们来介绍三种等待方式。

第一种:固定等待(time.sleep())

  在开发自动化框架过程中,最忌讳使用Python自带模块的time的sleep方法进行等待,虽然可以自定义等待时间,但当网络条件良好时,依旧按照预设定的时间继续等待,导致整个项目执行自动化的时间太长。不建议使用(注:脚本调试过程时,还是可以使用的,方便快捷)。

第二种:隐式等待(implicitly_wait())

  隐式等待实际是设置了一个最长等待时间,如果在规定时间内网页加载完成,则执行下一步,否则一直等到时间结束,然后执行下一步。这样的隐式等待会有个坑。我们都知道js一般都是放在我们的body的最后进行加载,实际这是页面上的元素都已经加载完毕,但我们却还在等待全部页面加载结束。隐式等待对整个driver周期都起作用,在最开始设置一次就可以了。不要当做固定等待使用,到哪都来一下隐式等待。

第三种:显示等待(WebDriverWait())

WebDriverWait是selenium提供的显示等待模块引入路径



1 from selenium.webdriver.support.wait import WebDriverWait
2 WebDriverWait(driver,timeout,poll_frequency,ignored_exceptions).until(method,message)
3 WebDriverWait(driver,timeout,poll_frequency,ignored_exceptions).until_not(method,message)



WebDriverWait参数



1 driver: 传入WebDriver实例,即我们上例中的driver
2 timeout: 超时时间,等待的最长时间
3 poll_frequency: 调用until或until_not中的方法的间隔时间,默认是0.5秒
4 ignored_exceptions: 忽略的异常,如果在调用until或until_not的过程中抛出这个元组中的异常,则不中断代码,继续等待,如果抛出的是这个元组外的异常,则中断代码,抛出异常。默认只有NoSuchElementException。



WebDriverWait模块中,一共只有两种方法until与until_not



1 method: 在等待期间,每隔一段时间调用这个传入的方法,直到返回值不是False
2 message: 如果超时,抛出TimeoutException,将message传入异常



until

当某元素出现或什么条件成立则继续执行

until_not

当某元素消失或什么条件不成立则继续执行

两个方法的method,必须是含有__call__的可执行方法。所以我们引用selenium提供的一个模块



1 from selenium.webdriver.support import expected_conditions as EC



1 '''隐式等待和显示等待都存在时,超时时间取二者中较大的'''
 2 locator = (By.ID,'kw')
 3 driver.get(base_url)
 4  
 5 WebDriverWait(driver,10).until(EC.title_is("百度一下,你就知道"))
 6 '''判断title,返回布尔值'''
 7  
 8 WebDriverWait(driver,10).until(EC.title_contains("百度一下"))
 9 '''判断title,返回布尔值'''
10  
11 WebDriverWait(driver,10).until(EC.presence_of_element_located((By.ID,'kw')))
12 '''判断某个元素是否被加到了dom树里,并不代表该元素一定可见,如果定位到就返回WebElement'''
13  
14 WebDriverWait(driver,10).until(EC.visibility_of_element_located((By.ID,'su')))
15 '''判断某个元素是否被添加到了dom里并且可见,可见代表元素可显示且宽和高都大于0'''
16  
17 WebDriverWait(driver,10).until(EC.visibility_of(driver.find_element(by=By.ID,value='kw')))
18 '''判断元素是否可见,如果可见就返回这个元素'''
19  
20 WebDriverWait(driver,10).until(EC.presence_of_all_elements_located((By.CSS_SELECTOR,'.mnav')))
21 '''判断是否至少有1个元素存在于dom树中,如果定位到就返回列表'''
22  
23 WebDriverWait(driver,10).until(EC.visibility_of_any_elements_located((By.CSS_SELECTOR,'.mnav')))
24 '''判断是否至少有一个元素在页面中可见,如果定位到就返回列表'''
25  
26 WebDriverWait(driver,10).until(EC.text_to_be_present_in_element((By.XPATH,"//*[@id='u1']/a[8]"),u'设置'))
27 '''判断指定的元素中是否包含了预期的字符串,返回布尔值'''
28  
29 WebDriverWait(driver,10).until(EC.text_to_be_present_in_element_value((By.CSS_SELECTOR,'#su'),u'百度一下'))
30 '''判断指定元素的属性值中是否包含了预期的字符串,返回布尔值'''
31  
32 #WebDriverWait(driver,10).until(EC.frame_to_be_available_and_switch_to_it(locator))
33 '''判断该frame是否可以switch进去,如果可以的话,返回True并且switch进去,否则返回False'''
34 #注意这里并没有一个frame可以切换进去
35  
36 WebDriverWait(driver,10).until(EC.invisibility_of_element_located((By.CSS_SELECTOR,'#swfEveryCookieWrap')))
37 '''判断某个元素在是否存在于dom或不可见,如果可见返回False,不可见返回这个元素'''
38 #注意#swfEveryCookieWrap在此页面中是一个隐藏的元素
39  
40 WebDriverWait(driver,10).until(EC.element_to_be_clickable((By.XPATH,"//*[@id='u1']/a[8]"))).click()
41 '''判断某个元素中是否可见并且是enable的,代表可点击'''
42 driver.find_element_by_xpath("//*[@id='wrapper']/div[6]/a[1]").click()
43 #WebDriverWait(driver,10).until(EC.element_to_be_clickable((By.XPATH,"//*[@id='wrapper']/div[6]/a[1]"))).click()
44  
45 #WebDriverWait(driver,10).until(EC.staleness_of(driver.find_element(By.ID,'su')))
46 '''等待某个元素从dom树中移除'''
47 #这里没有找到合适的例子
48  
49 WebDriverWait(driver,10).until(EC.element_to_be_selected(driver.find_element(By.XPATH,"//*[@id='nr']/option[1]")))
50 '''判断某个元素是否被选中了,一般用在下拉列表'''
51  
52 WebDriverWait(driver,10).until(EC.element_selection_state_to_be(driver.find_element(By.XPATH,"//*[@id='nr']/option[1]"),True))
53 '''判断某个元素的选中状态是否符合预期'''
54  
55 WebDriverWait(driver,10).until(EC.element_located_selection_state_to_be((By.XPATH,"//*[@id='nr']/option[1]"),True))
56 '''判断某个元素的选中状态是否符合预期'''
57 driver.find_element_by_xpath(".//*[@id='gxszButton']/a[1]").click()
58  
59 instance = WebDriverWait(driver,10).until(EC.alert_is_present())
60 '''判断页面上是否存在alert,如果有就切换到alert并返回alert的内容'''
61 print instance.text
62 instance.accept()
63  
64 driver.close()



WebDriverWait举例:



1 from selenium.webdriver.support.wait import WebDriverWait
 2 from selenium.webdriver.support import expected_conditions as EC
 3 from selenium.webdriver.common.by import By
 4 from selenium import webdriver
 5 driver = webdriver.Chrome()  # 实例化浏览器,启动浏览器
 6 driver.get('http://ui.imdsx.cn/uitester/')  # 以http协议发送请求,打开这个网址
 7 import time
 8 time.sleep(1)  # 等待1秒
 9 driver.execute_script('window.scrollTo(0,0);')  # 跳转到网页的最上面
10 time.sleep(1)  # 等待1秒
11 # 每隔1秒扫描一次dom,最多扫描10次,判断id为i1的元素是否被加到了dom里,如果定位到了就返回True,没定位到就等待下一次扫描,直到超过10秒则报超时错误
12 ele = WebDriverWait(driver,10,1).until(EC.presence_of_element_located((By.ID,'i1')))
13 ele.send_keys('xxxxx')



自定义expected_conditions



1 class current_url(object):
2     def __init__(self,current_url):
3         self.current_url = current_url
4     def __call__(self, driver):
5         return self.current_url == driver.current_url