由于UI層自動(dòng)化的不穩(wěn)定性，經(jīng)常會(huì)遇到執(zhí)行中斷或用例失敗的問題，以下是一些常見的措施。

1. 詳細(xì)的日志
2. 定位出錯(cuò)時(shí)截圖
3. Pytest的緩存機(jī)制(可以記錄成功了哪些失敗了哪些)
4. 自動(dòng)重試機(jī)制(如pytest-rerunfailures)
5. 用例錄像

用例錄像是最直觀的一種方式，可以查看到實(shí)時(shí)的操作情況。很多云平臺(tái)如Saucelabs等都提供了用例錄像的功能。
然而很遺憾selenium本身沒有這種能力。除了借助三方軟件如ffmpeg同步錄屏外，另外一種實(shí)現(xiàn)方法是另外啟動(dòng)一個(gè)線程，不斷截圖，最后將圖片拼接成git。
操作方法如下。

實(shí)時(shí)截圖

1. 編寫一個(gè)循環(huán)截圖函數(shù)shot

def shot(dr):
    i = 0
    while True:
        img_file = os.path.join(img_dir, f'{i}.png')
        try:
            dr.save_screenshot(img_file)
        except:
            return
        i += 1

由于webdriver接口執(zhí)行截圖命令的速度限制，每一輪截圖無需再sleep。

2. 操作web時(shí)，啟動(dòng)一個(gè)線程

img_dir = 'img'  # 臨時(shí)圖片目錄
dr = webdriver.Chrome()

t = threading.Thread(target=shot, args=(dr, img_dir))  # 新建線程
t.start()  # 啟動(dòng)截圖線程

dr.get('https://www.baidu.com')
dr.find_element('id', 'kw').send_keys('簡(jiǎn)書 韓志超')
dr.find_element('id', 'su').click()
time.sleep(1)
dr.get('https://www.qq.com')
dr.back()
time.sleep(2)
dr.quit()

3. 圖片拼接成gif

需要安裝pillow: pip install pillow

img_list = os.listdir(img_dir)  # 列出目錄所有圖片
img_list.sort(key=lambda x: int(x[:-4]))  # 排序

first_img = Image.open(os.path.join(img_dir, img_list[0]))  # 第一張圖片對(duì)象
else_imgs = [Image.open(os.path.join(img_dir, img)) for img in img_list[1:]]  # 剩余圖片對(duì)象

first_img.save("record.gif", append_images=else_imgs,
               duration=300,
               save_all=True) # 拼接保存

完整代碼

from selenium import webdriver
import threading
import os
import time
from PIL import Image


def clear_dir(path):
    """創(chuàng)建或清空目錄"""
    if not os.path.isdir(path):
        os.mkdir(path)  # 創(chuàng)建目錄
    else:  # 清空目錄
        [os.remove(os.path.join(path, file_name)) for file_name in os.listdir(path)]


def shot(dr, img_dir):
    """循環(huán)截圖函數(shù)"""
    i = 0
    clear_dir(img_dir)  # 清空目錄
    while True:
        img_file = os.path.join(img_dir, f'{i}.png')
        try:
            dr.save_screenshot(img_file)
        except:
            return
        i += 1


# Selenium操作
img_dir = 'img'  # 臨時(shí)圖片目錄
dr = webdriver.Chrome()

t = threading.Thread(target=shot, args=(dr, img_dir))  # 新建線程
t.start()  # 啟動(dòng)截圖線程

dr.get('https://www.baidu.com')
dr.find_element('id', 'kw').send_keys('簡(jiǎn)書 韓志超')
dr.find_element('id', 'su').click()
time.sleep(1)
dr.get('https://www.qq.com')
dr.back()
time.sleep(2)
dr.quit()

# 圖片拼接成gif
img_list = os.listdir(img_dir)  # 列出目錄所有圖片
img_list.sort(key=lambda x: int(x[:-4]))  # 排序

first_img = Image.open(os.path.join(img_dir, img_list[0]))  # 第一張圖片對(duì)象
else_imgs = [Image.open(os.path.join(img_dir, img)) for img in img_list[1:]]  # 剩余圖片對(duì)象

first_img.save("record.gif", append_images=else_imgs,
               duration=300,  # 每張圖片的過過渡時(shí)間
               save_all=True) # 拼接保存，如果想要循環(huán)播放可以加上loop=0

最終效果--有點(diǎn)像幻燈片

image