【Selenium学习】三种元素等待

在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