自动化测试过程中，是否遇到过脚本执行中途出错却不知道原因的情况？测试人员面临的不仅是问题的复现，还有对错误的快速定位和分析。而异常截图与页面源码（Page Source）的结合，正是解决这一难题的利器。
在实际的自动化测试中，您是否清楚异常发生时的具体页面状态？单凭日志信息往往无法全面还原错误背景。通过自动化实现异常截图和保存页面源码的功能，您可以大幅提升定位问题的效率和准确性。
在UI自动化测试行业，随着业务复杂性增加，定位问题的效率和深度成为测试团队竞争力的核心。以异常截图和页面源码为代表的增强调试工具，已成为优秀测试框架的标配。例如 Allure 的报告集成截图功能，TestNG 的详细异常日志，都表明这种趋势在逐步标准化。
为什么需要异常截图和Page Source？
- 截图直观定位问题：UI界面异常、元素定位失败等问题可通过截图直接查看。
- 源码辅助排查：页面源码保存可帮助分析DOM结构变化、动态加载等问题。
- 联动使用效果更佳：截图展示表面，源码剖析细节，二者结合为测试提供全方位视角。
实际案例分享
例如，使用 Selenium 进行电商网站的自动化测试时，在支付页面的验证中，偶尔会出现按钮未加载的问题。通过启用异常捕获机制，截图显示支付按钮缺失，而源码内容揭示了JS脚本未执行的原因，最终确认问题出在接口超时。
01 场景
-
增加自动化测试代码的可测性
-
丰富报告
02 实现代码异常时
实现代码异常的时候，实现截图和打印page_source
实现方法：try except 配合截图和page_source操作
特别注意1：
-
在保存截图和页面源码时，一定先创建好images、source_path路径
-
保存截图：driver.save_screenshot(路径名称)
-
获取页面源码：driver.page_source()
-
异常处理会影响用例本身的结果；
解决办法：在except之后，再把异常抛出
代码最后加上：raise Exception；
如果用例失败，抛出异常；否则即使捕获到异常，用例也会通过
特别注意2：
-
将截图保存到allure报告中
allure.attach.file(截图路径,name=‘image’,attachment_type=allure.attachment_type.PNG)
-
将页面源码保存到allure中,以文本的形式存储
allure.attach.file(源码路径,name=‘text’,attachment_type=allure.attachment_type.TEXT)
import sys
import time
import allure
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.action_chains import ActionChains
class TestBaidu:
def setup_class(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(2)
def teardown_class(self):
self.driver.quit()
def test_baidu(self):
self.driver.get('https://www.baidu.com')
try:
self.driver.find_element(By.ID, 'su1')
except Exception:
# 时间戳
time_stamp = int(time.time())
# 注意：一定要创建好images路径、source_path路径
image_path = f'./images/image_{time_stamp}.PNG'
page_source_path = f'./page_source/page_source_{time_stamp}.html'
# 保存截图
self.driver.save_screenshot(image_path)
# 保存获取到的页面源码
with open(page_source_path, 'w', encoding='utf-8') as f:
f.write(self.driver.page_source)
# 将截图添加到allure报告中
allure.attach.file(image_path,
name='image',
attachment_type=allure.attachment_type.PNG)
# 将页面源码添加到allure报告中
allure.attach.file(page_source_path,
name='text',
attachment_type=allure.attachment_type.TEXT)
# 如果用例失败，抛出异常；否则即使捕获到异常，用例也会通过
raise Exception03 代码优化
异常捕获处理代码是公共方法和业务代码无关，不能耦合。
解决办法，使用装饰器装饰用例或者相关方法。
思路
-
先把装饰器架子搭建好
-
把相关逻辑嵌套进来
特别注意
使用装饰器执行用例，被装饰函数还没有执行，所以还没有self.driver，获取被装饰方法的self，也就是实例对象，通过self就可以拿到声明的实例变量driver
driver = args[0].driver
前提条件：被装饰的方法是一个实例方法，实例需要有实例变量self.driver
-
解决方法1：获取driver放到函数执行之后
import sys
import time
import allure
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
# 采用装饰器
def ui_exception_record(func):
def inner(*args, **kwargs):
try:
return func(*args, **kwargs)
except Exception:
driver = args[0].driver
# 时间戳
time_stamp = int(time.time())
# 注意：一定要创建好images路径、source_path路径
image_path = f'./images/image_{time_stamp}.PNG'
page_source_path = f'./page_source/page_source_{time_stamp}.html'
# 保存截图
driver.save_screenshot(image_path)
# 保存获取到的页面源码
with open(page_source_path, 'w', encoding='utf-8') as f:
f.write(driver.page_source)
# 将截图添加到allure报告中
allure.attach.file(image_path,
name='image',
attachment_type=allure.attachment_type.PNG)
# 将页面源码添加到allure报告中
allure.attach.file(page_source_path,
name='text',
attachment_type=allure.attachment_type.TEXT)
# 如果用例失败，抛出异常；否则即使捕获到异常，用例也会通过
raise Exception
return innerclass TestBaidu2:
def setup_class(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(2)
def teardown_class(self):
self.driver.quit()
@ui_exception_record
def test_baidu(self):
self.driver.get('https://www.baidu.com')
self.driver.find_element(By.ID, 'su1')-
解决方法2：保证使用装饰器的时候，driver已经声明：driver = args[0].driver
import sys
import time
import allure
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
# 采用装饰器
def ui_exception_record(func):
def inner(*args, **kwargs):
driver = args[0].driver
try:
return func(*args, **kwargs)
except Exception:
# 时间戳
time_stamp = int(time.time())
# 注意：一定要创建好images路径、source_path路径
image_path = f'./images/image_{time_stamp}.PNG'
page_source_path = f'./page_source/page_source_{time_stamp}.html'
# 保存截图
driver.save_screenshot(image_path)
# 保存获取到的页面源码
with open(page_source_path, 'w', encoding='utf-8') as f:
f.write(driver.page_source)
# 将截图添加到allure报告中
allure.attach.file(image_path,
name='image',
attachment_type=allure.attachment_type.PNG)
# 将页面源码添加到allure报告中
allure.attach.file(page_source_path,
name='text',
attachment_type=allure.attachment_type.TEXT)
# 如果用例失败，抛出异常；否则即使捕获到异常，用例也会通过
raise Exception
return innerreturn inner
class TestBaidu2:
def setup_class(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(2)
def teardown_class(self):
self.driver.quit()
@ui_exception_record
def test_baidu(self):
self.driver.get('https://www.baidu.com')
self.driver.find_element(By.ID, 'su1')-
一旦被装饰的方法有返回值，会丢失返回值
解决方案：return func(*args, **kwargs)
当用例执行失败allure报告中可以查看截图
当用例执行失败allure报告中可以查看page_source源码
异常截图和Page Source的结合，不仅帮助测试人员快速还原问题背景，更为问题定位提供了关键依据。这种看似简单的操作，却是提升测试效率的关键一步。
测试的目标从来不是证明程序正确，而是帮助发现问题。学会利用工具，还原问题真相，才是真正的测试之道！
