关于python UI Selenium自动化的一个浏览器driver重复调用问题

最开始也是和大多数一样先写个方法去定义一个驱动每次都要传入一个driver对象

class PageObject(object):
    def __init__(self, driver):
        self.driver = driver

    def test(self):
        ele = self.driver.find_element_by_id("menu-user")
        return ele

    def login(self):
        self.driver.find_element_by_id("userAccount").send_keys("admin")
        self.driver.find_element_by_id("loginPwd").send_keys(123456)
        time.sleep(2)
        self.driver.find_element_by_id("loginBtn").click()

 

from selenium import webdriver
import unittest

class Test(unittest.TestCase):

    def setUp(self):
        self.driver=webdriver.Chrome()
        print("打开浏览器")
        self.driver.get("http://www.baidu.com")
        print(self.driver.current_url)
    def tearDown(self):
        self.driver.close()
        print("close浏览器")
    def test_1(self):
        self.driver.get("http://www.vip.com")
        print(self.driver.current_url)
if __name__ == "__main__":
    unittest.main()

class Test(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()