为了在测试用例执行失败时自动截图，写了一个装饰器，代码如下：
class ScreenShoot:
def __init__(self, driver):
self.driver = driver
def __call__(self, func):
def inner(*args, **kwargs):
try:
return func(*args, **kwargs)
except:
now_time = time.strftime('%Y_%m_%d_%H_%M_%S') # 异常时，截图
self.driver.get_screenshot_as_file(os.path.dirname(os.path.dirname(__file__)) + '/report/{}.png'.format(now_time))
raise
return inner
正如代码所示，这个截图类需要传入一个driver
在看我的测试类：
......
@ddt
class TestNewPartsSale(unittest.TestCase):
"""新建配件销售单并销售"""
driver = webdriver.Chrome()
@classmethod
def setUpClass(cls) -> None:
# cls.driver = webdriver.Chrome()
cls.driver.maximize_window()
cls.page = LoginPage(cls.driver)
cls.page.get('http://xxxxxx')
cls.page.username = 'xxxx'
cls.page.password = 'xxxx'
cls.page.verify_code = 'xxx'
cls.page.login_button.click()
cls.page = PartsSalePage(cls.driver)
@ScreenShoot(driver)
def dispatch_orders(self, batch=False):
"""派单操作"""
# 服务管理-服务单
webdriver.ActionChains(self.driver).click(self.page.service_manage).perform()
self.page.work_order.click()
time.sleep(1)
# 待派工-第一单服务单直接派工
.....
想要给ScreenShoot传入driver，就必须把 driver的声明放到这个测试类下，并注释掉类方法setUpClass中的cls.driver,这就相当于一个全局变量，然后问题就出现了：当我用
unnittest.defaultTestLoader.discover(test_dir, pattern='test*.py')方法转载测试文件时执行，浏览器会启动n个空白浏览器driver，n的值为我的测试文件数量
如果将空白浏览器关闭，当执行到此测试文件时会直接报错selenium.common.exceptions.WebDriverException: Message: chrome not reachable
不知道是什么原因，runner.py代码如下：
import unittest
import os
import time
from BeautifulReport import BeautifulReport as BR
test_dir = os.path.dirname(os.getcwd()) + '/test_case'
suits = unittest.defaultTestLoader.discover(test_dir, pattern='test*.py')
if __name__ == '__main__':
current_time = time.strftime('%Y-%m-%d %H_%M_%S')
runner = BR(suits)
report_dir = os.path.dirname(os.path.dirname(__file__)) + '/report'
runner.report(filename='test_report' + current_time, description='done', report_dir=report_dir)
这里怀疑是BR这个类会实例化所有的测试类，而测试类中driver = webdriver.Chrome()就会被全部加载
如果将webdriver的启动放到setUpClass中，截图类就无法传入driver，水平暂时不够，只能换一种自动截图的方法，此问题暂时放一放。
待更。。。
