Selenium常用的断言包括
- 页面属性断言：断言标题、url或页面源码中是否包含或不包含特定字符
- 元素存在断言：断言指定元素存在
- 图片及链接断言：断言图片正常显示、链接可以正常打开
页面属性断言
这是最常用的断言方式，可以用来断言页面是否正常打开，是否在指定的窗口及页面上，示例如下。
from selenium import webdriver
dr = webdriver.Chrome()
dr.get('https://www.baidu.com')
dr.find_element('link text', '新闻').click()
dr.switch_to.window(dr.window_handles[-1]) # 切换到最后一个窗口
assert '百度新闻' in dr.title # 断言标题包含
assert 'http://news.baidu.com/' == dr.current_url # 断言为指定url
assert 'not found' not in dr.page_source # 断言页面源码不包含not found（一般Nginx找不到页面时返回404页面，显示Not Found）
dr.quit()
页面元素断言
有时我们需要判断页面是否存在指定元素来继续操作，或通过是否存在该元素进行断言是否在指定页面上。
断言元素是否存在可以使用try ... except捕获并抑制元素定位不到的异常，或者使用find_elements+判断返回的元素列表是否为空判断是否能定位到元素。
对于需要等待一定时间才出现的元素，我们可以私用智能等待或主动等待来轮询检查元素，直到超时。示例如下。
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support.wait import WebDriverWait, TimeoutException
dr = webdriver.Chrome()
dr.get('https://www.baidu.com')
dr.maximize_window()
try:
settings = dr.find_element('id', 's-usersetting-top')
except NoSuchElementException:
print('未定位到"设置"')
# raise AssertionError('设置链接不存在') # 测试时,需抛出标准的断言异常, 对于偶现元素,则无需抛出异常
else:
print('"设置"存在')
ActionChains(dr).move_to_element(settings).perform() # 鼠标移动到设置
# 或 使用 find_elements + 列表长度判断
elms = dr.find_elements('id', 's-usersetting-top')
if len(elms) > 0:
settings = elms[0]
print('"设置"存在')
ActionChains(dr).move_to_element(settings).perform()
else:
print('未定位到"设置"')
raise AssertionError('设置链接不存在')
# 对需要等待的元素,使用主动等待轮询
try:
WebDriverWait(dr, 10, 0.5).until(
lambda dr: dr.find_element('link text', '高级搜索')
).click()
except TimeoutException:
print('未定位到"高级搜索"')
# raise AssertionError('高级搜索不存在')
dr.quit()
图片及链接断言
图片是否显示以及链接是否能正常打开，我们可以通过dr.get('图片或链接地址')，看是否返回404页面来判断图片是否能正常打开，示例如下。
from selenium import webdriver
dr = webdriver.Chrome()
dr.get('https://www.baidu.com')
baidu_logo_url = dr.find_element('id', 's_lg_img').get_attribute('src')
print('百度Logo图片链接', baidu_logo_url)
dr.get(baidu_logo_url) # 尝试打开图片
assert 'Not Found' not in dr.page_source # 假设不存在报错页面包含Not Found字样
dr.back()
hao123_link_url = dr.find_element('link text', 'hao123').get_attribute('href')
print('hao123链接', hao123_link_url)
dr.get(hao123_link_url) # 尝试打开页面
assert 'Not Found' not in dr.page_source
dr.back()
dr.quit()
图片和链接使用接口请求来检查会更快，或者使用专门的404检查工具。
其他
- 颜色断言：Selenium支持
- 样式排列断言：比如标题是否折行，按钮重叠，错位等，可以通过保存一份页面（或局部）的截图，通过python-opencv图片对比等一系列算法实现。
- 断言图片是否指定图片：通过图片对比，涉及一些图像算法