Python + Selenium自动化测试 -- 异常捕捉与错误截图

最新推荐文章于 2025-01-13 09:53:44 发布

原创

最新推荐文章于 2025-01-13 09:53:44 发布 · 3.3k 阅读

14 ·

CC 4.0 BY-SA版权

本文介绍了如何在使用Python和Selenium进行自动化测试时，通过异常捕捉来处理错误，并实现错误发生时的截图保存。利用此功能，可以更方便地定位和解决测试过程中遇到的问题。

异常捕捉与错误截图

用例不可能每一次运行都成功，肯定运行时候有不成功的时候。如果可以捕捉到错误，并且把错误截图保存，这将是一个非常棒的功能，也会给我们错误定位带来方便。

例如编写一个函数，关键语句为driver.get_screenshot_as_file：

    def savescreenshot(self,driver,file_name):
        if not os.path.exists('./image'):
            os.makedirs('./image')
        now=time.strftime("%Y%m%d-%H%M%S",time.localtime(time.time()))
        #截图保存
        driver.get_screenshot_as_file('./image/'+now+'-'+file_name)
        time.sleep(1)

一个引用的例子：

testscreenshot.py

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
import os

class Baidu1(unittest.TestCase):
#test fixture，初始化环境
    def setUp(self):
        s