Selenium中的断言

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图片对比等一系列算法实现。
断言图片是否指定图片：通过图片对比，涉及一些图像算法