Selenium中的断言 Selenium中的断言

【Selenium中的断言】Selenium常用的断言包括

页面属性断言：断言标题、url或页面源码中是否包含或不包含特定字符
元素存在断言：断言指定元素存在
图片及链接断言：断言图片正常显示、链接可以正常打开

页面属性断言这是最常用的断言方式，可以用来断言页面是否正常打开，是否在指定的窗口及页面上，示例如下。

from selenium import webdriverdr = 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, TimeoutExceptiondr = 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 webdriverdr = 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()