как я могу использовать драйвер init в setupModule

Question

когда я использую unittest в python3, я пытался так:

import unittest
from selenium import webdriver

def setupModule():
    driver = webdriver.Firefox
    driver.maximize_window()
    driver.get('www.google.com')

def teardownModule():
    driver.close()

class test_01(unittest.TestCase):
    def setUp(self):
        driver.xxxx

    def tearDown(self):
        driver.xxxx

    def test_0001(self):
        driver.yyyy

    def test_0002(self):
        driver.zzzz

class test_02(unittest.TestCase):
    def setUp(self):
        driver.xxxx

    def tearDown(self):
        driver.xxxx

    def test_0001(self):
        driver.yyyy

    def test_0002(self):
        driver.zzzz

драйвер в классе и teardownModule не может быть распознан.Есть ли способ сделать его доступным?

Я не хочу выводить driver = webdriver.Firefox из def, как если бы у меня был 2 py-файл для разных случаев, он сначала запустит 2 или более файлов Firefox, и ни один из случаев в этом файле не будет запущен илинет, браузер всегда будет открыт.

Answer 1

Я бы порекомендовал иметь базовый класс для управления настройкой и отключением веб-драйвера, т. Е.

class BaseTest(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox

    def tearDown(self):
        self.driver.quit()

class test_01(BaseTest):
    def test_0001(self):
        self.driver.xxx

class test_02(BaseTest):
    def test_0002(self):
        self.driver.xxx