[未解决]unittest通过discover方法执行测试文件时会打开空的浏览器窗口

为了在测试用例执行失败时自动截图，写了一个装饰器，代码如下：

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，水平暂时不够，只能换一种自动截图的方法，此问题暂时放一放。

待更。。。