selenium ---- 异常自动截图

ShouSi_DaiMa · 2024 年6 月 1 日 03:58

异常自动截图的作用

1、增加自动化测试代码的可测性
2、丰富报告

实现原理

装饰器
自动化关键数据记录

1、截图

image2122×1362 240 KB

2、日志

3、page_source

image2140×1363 428 KB

image2121×1364 228 KB

实现代码

# 目标1：实现代码异常的时候，截图/打印page_source
# 实现方法：try/catch 配合截图/ page_source操作
import time
import allure
from selenium import webdriver
from selenium.webdriver.common.by import By

# ====问题4：隐藏的小bug，一旦被装饰方法有返回值，会丢失返回值
# 解决方案：给被装饰器装饰的方法添加return返回值
def ui_exception_record(func):
    def inner(*args, **kwargs):
        """
        内函数，实现封装异常捕获处理的代码
        :param args: 不定长参数，是个元组，args[0] 是 TestBaidu类的实例对象，相当于self
        :param kwargs:
        :return:
        """
        # 获取TestBaidu类中被装饰方法的self，也就是实例对象,args[0]相当于self，args[0].driver = self.driver
        # 前提条件： TestBaidu类中的driver变量是一个实例变量 self.driver
        try:
            # 当被装饰方法/函数发生异常，就捕获并做数据记录
            func(*args, **kwargs)
        except Exception:
            # ====问题3： 被装饰的函数test_baidu()还没执行, 就先调用了装饰器ui_exception_record，此时还没有self.driver,导致args[0]还获取不到self
            # 解决方案一：获取self.driver的操作，放在装饰器中被装饰的函数func(*args, **kwargs)之后
            driver = args[0].driver
            # 出现异常情况
            # print("出现异常情况")
            # 截图操作
            timestamp = int(time.time())
            image_path = f"../images/image_{timestamp}.PNG"
            driver.save_screenshot(image_path)

            # page_source操作
            page_source_path = f"../page_source/page_source_{timestamp}.HTML"
            with open(page_source_path, "w", encoding="utf8") as f:
                f.write(driver.page_source)

            # 将截图放到allure报告中
            allure.attach.file(image_path, name="picture", attachment_type=allure.attachment_type.PNG)
            # 将page_source放到allure报告中
            # 如果想要HTML格式展示在报告中，则使用HTML
            allure.attach.file(page_source_path, name="page_source", attachment_type=allure.attachment_type.HTML)
            # 如果想要TEXT格式展示在报告中，则使用TEXT
            allure.attach.file(page_source_path, name="page_source", attachment_type=allure.attachment_type.HTML)

        # 待截图和page_source数据保存完成后，抛出异常，此时用例会按正常情况，执行失败
        raise Exception

    # 返回内函数对象
    return inner


class TestBaidu:
    def setup_class(self):
        # 解决方案二：在前置条件中对self.driver进行实例化操作
        self.driver = webdriver.Chrome()

    @ui_exception_record
    def find(self):
        return self.driver.find_element(By.ID, "su1")

    def test_baidu(self):
        self.driver.get("https://www.baidu.com")
        self.find().click()
        # self.driver.find_element(By.ID, "su1")