Selenium自动化测试-设置元素等待

2019-11-14 17:05:11 浏览数 (1)

selenium中有三种时间等待：

  • 强制等待：sleep
  • 隐式等待：implicitly_wait
  • 显示等待：WebDriverWait

1.sleep

让程序暂停运行一定时间，等待时间到达后继续运行。

使用sleep，需先导入time模块，import time, 然后使用time.sleep()来让程序等待多久。

代码语言：javascript复制
from selenium import webdriver
import time
driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')
# 等待3s
time.sleep(3)
# 点击新闻链接
driver.find_element_by_link_text("新闻").click()

2. implicitly_wait

implicitly_wait()默认参数的单位为秒，默认值为0。在最大超时时间内找到元素了，会开始执行下一操作，如果在最大超时间内未找到元素，会抛出NoSuchElementException 异常,这样能节省定位时间。

代码语言：javascript复制
from selenium import webdriver
driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')
# 隐式等待3s
driver.implicitly_wait(3)
# 点击新闻链接
driver.find_element_by_link_text("新闻").click()

3. WebDriverWait

等待某个条件成立时继续执行，否则在达到最大时长时抛出超时异常TimeoutException。

WebDriverWait一般和until()和until_not()配合使用：

  • until() 当某元素出现或什么条件成立则继续执行
  • until_not 当某元素消失或什么条件不成立则继续执

WebDriverWait(driver, timeout, poll_frequency=0.5,

ignored_exceptions=None)

  • driver: 传入WebDriver实例；
  • timeout：指最大超时时间，默认单位为秒；
  • poll_frequency：调用until或until_not方法，每隔一定时间不断尝试是否能找到页面元素，默认间隔是0.5s,可自行调整间隔时间。
  • ignored_exceptions：超时后的异常信息，默认情况下NoSuchElementException 异常。

使用WebDriverWait，需要先导入WebDriverWait模块。

代码语言：javascript复制
from selenium.webdriver.support.ui import WebDriverWait

我们使用WebDriverWait方式来定位百度页面的新闻链接，

代码如下：

代码语言：javascript复制
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')
# 设置显式等待，超时时长最大为 5s，每隔0.5s查找元素一次
element = WebDriverWait(driver,5).until(
lambda x: x.find_element_by_link_text('新闻'))
element.click()

注意：until或until_not中的method参数一定要是可以调用的对象，即这个对象一定有 __call__方法，否则会抛出异常。

例如：

代码语言：javascript复制
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')
# 设置显式超时时长最大为5s，每隔0.5s查到元素一次
element = WebDriverWait(driver,5).until(
    driver.find_element_by_link_text('新闻'))
element.click()

运行的结果是报错的：

代码语言：javascript复制
Traceback (most recent call last):
File "C:/Users/96984/PycharmProjects/vivi_python/selenium_vivi/radio.py", line 7, in <module>
element = WebDriverWait(driver,5).until(driver.find_element_by_link_text('新闻'))
File "C:Users96984AppDataLocalProgramsPythonPython37-32libsite-packagesseleniumwebdriversupportwait.py", line 71, in until
value = method(self._driver)
TypeError: 'WebElement' object is not callable

所以我们可以用selenium提供的expected_conditions，提供一些场景的判断，或者用自己封装的方法都可以。

使用expected_conditions，需先导入。

代码语言：javascript复制
from selenium.webdriver.support import expected_conditions as EC

以下是常用的expected_conditions方法：

方法

说明

title_is

判断当前页面的 title 是否完全等于（==）预期字符串，返回布尔值

title_contains

判断当前页面的 title 是否包含预期字符串，返回布尔值

presence_of_element_located

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

visibility_of_element_located

判断元素是否可见（可见代表元素非隐藏，并且元素宽和高都不等于 0）

visibility_of

同上一方法，只是上一方法参数为locator，这个方法参数是 定位后的元素

presence_of_all_elements_located

判断是否至少有 1 个元素存在于 dom 树中。举例：如果页面上有 n 个元素的 class 都是’qw’，那么只要有 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

判断某个元素中是否可见并且可点击

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

我们以presence_of_element_located这个方法为例，看下WebDriverWait怎么和expected_conditions配合使用。

代码语言：javascript复制
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
driver = webdriver.Chrome()
driver.get('https://www.baidu.com/')
# 设置显式等待，超时时长最大为5s，每隔0.5s查找元素一次
element = WebDriverWait(driver,5).until(
    EC.presence_of_element_located(('id','kw')))
element.send_keys('vivi')

最后总结下三种元素等待的优缺点：

元素等待方式

优点

缺点

time

使用简单，在程序调试时使用

浪费不必要的等待时间，影响用例执行效率

implicitly_wait

一旦设置，这个隐式等待会在WebDriver对象实例的整个生命周期起作用

程序等待整个页面加载完成，才会下一步操作，不够灵活

WebDriverWait

条件成立，就会下一步操作，节省等待时间，提高用例执行效率

使用相对复杂，学习成本相对较大

编程算法 selenium

0 人点赞