Python 集成视频录制(Selenium):让 UI 自动化问题无处隐藏
📝 面试求职: 「面试试题小程序」 ,内容涵盖 测试基础、Linux操作系统、MySQL数据库、Web功能测试、接口测试、APPium移动端测试、Python知识、Selenium自动化测试相关、性能测试、性能测试、计算机网络知识、Jmeter、HR面试,命中率杠杠的。(大家刷起来…)
📝 职场经验干货:
在 Web UI 自动化测试中,“偶发性失败”是最头疼的问题:
元素突然找不到?
按钮点击无反应?
页面渲染错乱?
仅靠截图和日志,往往无法还原完整操作过程。
而 视频录制 能提供 连续、直观、可回放 的上下文,是定位 UI 问题的终极武器。
今天,我们就手把手教你用 纯 Python 方案,在 Pytest + Selenium 项目中集成 智能视频录制 —— 仅录制失败用例,节省 90% 存储空间。
🎯 核心目标
|
需求 |
实现方案 |
|---|---|
|
✅ 按需录制 |
仅当用例失败时保存视频 |
|
✅ 轻量高效 |
不拖慢测试速度 |
|
✅ 无缝集成 |
自动附加到 Allure 报告 |
|
✅ 跨平台支持 |
Windows / macOS / Linux |
🧱 技术选型:为什么用 FFmpeg?
|
工具 |
缺点 |
本方案优势 |
|---|---|---|
|
Selenium 截图序列 |
占用空间大、需手动合成视频 |
实时编码,直接生成 MP4 |
|
BrowserStack 录屏 |
依赖云服务、成本高 |
本地录制,零额外费用 |
|
手动 OBS 录制 |
无法自动化、精度低 |
精准绑定单个用例 |
💡 FFmpeg 是行业标准音视频处理工具,支持:
实时屏幕捕获(Windows/macOS/Linux)
硬件加速编码(H.264)
极低 CPU 开销
🛠️ 第一步:安装依赖
1. 安装 FFmpeg
Windows:https://www.gyan.dev/ffmpeg/builds/ → 解压后将 bin 加入 PATH
macOS:brew install ffmpeg
Linux (Ubuntu):sudo apt install ffmpeg
2. Python 依赖
pip install selenium pytest allure-pytest opencv-python
✅ 验证安装:终端执行 ffmpeg -version 应显示版本信息
🎥 第二步:封装视频录制器(核心类)
# utils/video_recorder.py
import os
import subprocess
import threading
import time
from pathlib import Path
class VideoRecorder:
def __init__(self, output_path: str, fps: int = 10):
self.output_path = Path(output_path)
self.fps = fps
self.process = None
self._stop_event = threading.Event()
self._thread = None
def start(self):
"""启动录制(后台线程)"""
if os.name == 'nt': # Windows
cmd = [
'ffmpeg',
'-f', 'gdigrab',
'-framerate', str(self.fps),
'-i', 'desktop',
'-vcodec', 'libx264',
'-preset', 'ultrafast',
'-pix_fmt', 'yuv420p',
'-y', str(self.output_path)
]
else: # macOS / Linux
display = os.environ.get('DISPLAY', ':0.0')
cmd = [
'ffmpeg',
'-f', 'x11grab',
'-framerate', str(self.fps),
'-s', '1920x1080', # 可根据实际分辨率调整
'-i', display,
'-vcodec', 'libx264',
'-preset', 'ultrafast',
'-pix_fmt', 'yuv420p',
'-y', str(self.output_path)
]
# 启动 FFmpeg 进程(不显示控制台窗口)
startupinfo = None
if os.name == 'nt':
startupinfo = subprocess.STARTUPINFO()
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
self.process = subprocess.Popen(
cmd,
stdin=subprocess.PIPE,
stdout=subprocess.DEVNULL,
stderr=subprocess.DEVNULL,
startupinfo=startupinfo
)
# 启动监控线程
self._stop_event.clear()
self._thread = threading.Thread(target=self._monitor)
self._thread.daemon = True
self._thread.start()
def stop(self):
"""停止录制"""
if self.process:
self._stop_event.set()
# 发送 'q' 命令给 FFmpeg 正常退出
try:
self.process.stdin.write(b'q')
self.process.stdin.flush()
except:
pass
self.process.wait(timeout=5)
if self.process.poll() is None:
self.process.kill()
self.process = None
def _monitor(self):
"""监控录制状态"""
while not self._stop_event.is_set():
time.sleep(0.1)
⚠️ 关键参数说明:
-preset ultrafast:牺牲压缩率换取速度
-pix_fmt yuv420p:确保视频能在浏览器播放
fps=10:平衡清晰度与文件大小(UI 操作无需高帧率)
🔗 第三步:集成到 Pytest + Selenium
1. 在 conftest.py 中管理录制生命周期
# conftest.py
import pytest
import allure
from utils.video_recorder import VideoRecorder
from selenium import webdriver
import os
import tempfile
@pytest.fixture(scope="function")
def driver():
# 初始化 WebDriver
options = webdriver.ChromeOptions()
options.add_argument("--headless=new") # 如需可视化可注释
driver = webdriver.Chrome(options=options)
yield driver
driver.quit()
@pytest.fixture(scope="function")
def video_recorder(request):
"""视频录制 fixture"""
test_name = request.node.name
video_path = os.path.join(tempfile.gettempdir(), f"{test_name}.mp4")
recorder = VideoRecorder(video_path)
# 开始录制
recorder.start()
yield recorder
# 停止录制(无论成功失败)
recorder.stop()
# 仅当用例失败时附加到 Allure
if hasattr(request.node, 'rep_call') and request.node.rep_call.failed:
if os.path.exists(video_path) and os.path.getsize(video_path) > 0:
with open(video_path, "rb") as f:
allure.attach(
f.read(),
name="操作视频",
attachment_type=allure.attachment_type.MP4
)
# 清理临时文件(可选)
# os.remove(video_path)
# Hook: 捕获用例执行结果
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
def pytest_runtest_makereport(item, call):
outcome = yield
rep = outcome.get_result()
setattr(item, f"rep_{rep.when}", rep)
2. 在测试用例中使用
# test_login.py
def test_login_failure(driver, video_recorder):
"""故意制造失败:输入错误密码"""
driver.get("https://example.com/login")
driver.find_element("id", "username").send_keys("admin")
driver.find_element("id", "password").send_keys("wrong_password")
driver.find_element("id", "submit").click()
# 断言失败(触发视频保存)
assert "Welcome" in driver.page_source
🌟 高级优化:提升体验与性能
优化 1:仅录制浏览器区域(非全屏)
# 修改 VideoRecorder.start() 中的 FFmpeg 命令
# Windows 示例(需先获取浏览器窗口位置)
import pygetwindow as gw
def get_browser_window_rect(driver):
# 获取 Chrome 窗口(需安装 pygetwindow)
windows = gw.getWindowsWithTitle('Chrome')
if windows:
win = windows[0]
return (win.left, win.top, win.width, win.height)
return (0, 0, 1920, 1080)
# FFmpeg 命令添加 -offset_x, -offset_y, -video_size
cmd = [
'ffmpeg',
'-f', 'gdigrab',
'-framerate', str(self.fps),
'-offset_x', str(x),
'-offset_y', str(y),
'-video_size', f'{width}x{height}',
'-i', 'desktop',
...
]
✅ 效果:视频只包含浏览器,文件更小,隐私更安全
优化 2:动态调整录制质量
# 根据环境选择参数
if os.getenv("CI"):
fps = 5 # CI 环境降低帧率
preset = "superfast"
else:
fps = 10
preset = "ultrafast"
优化 3:自动清理旧视频
# 在 video_recorder fixture 中
import shutil
from datetime import datetime, timedelta
def cleanup_old_videos():
tmp_dir = Path(tempfile.gettempdir())
for video in tmp_dir.glob("test_*.mp4"):
if datetime.now() - datetime.fromtimestamp(video.stat().st_mtime) > timedelta(hours=24):
video.unlink()
📊 效果验证:Allure 报告中的视频
当用例失败时,Allure 报告会自动显示:
(实际报告中可直接播放 MP4 视频)
✅ 价值:
开发无需复现,直接看视频定位问题
减少“在我机器上能跑”的扯皮
提升缺陷描述专业度
⚠️ 注意事项与避坑指南
|
问题 |
解决方案 |
|---|---|
|
FFmpeg 未找到 |
确保已加入系统 PATH,或指定绝对路径 |
|
视频文件为空 |
检查 FFmpeg 命令权限(Linux/macOS 需允许屏幕录制) |
|
录制卡顿 |
降低 FPS(5~10 足够),使用 |
|
CI 环境无 GUI |
在 headless 模式下无法录制 → 改用截图序列合成视频 |
|
文件过大 |
限制录制时长(如最多 60 秒),或转码压缩 |
✅ 总结:视频录制 = UI 自动化的“黑匣子”
当文字和截图无法说清问题时,一段 10 秒的视频胜过千言万语。
通过本方案,你将获得:
🔍 精准问题复现:失败瞬间完整操作流
💾 智能存储管理:仅保存有价值视频
📤 无缝报告集成:Allure 中一键播放
🚀 团队效率提升:减少沟通成本,加速修复
📣 行动建议
在本地环境安装 FFmpeg 并验证
将 VideoRecorder 类复制到你的项目
在一个失败用例中启用录制,查看 Allure 效果
好的自动化,不仅要发现问题,更要让问题无处可藏。
最后: 下方这份完整的软件测试视频教程已经整理上传完成,需要的朋友们可以自行领取【保证100%免费】
更多推荐


所有评论(0)