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圖片對比等一系列算法實現。
- 斷言圖片是否指定圖片：通過圖片對比，涉及一些圖像算法