设定浏览器驱动

主要是做测试前的准备和测试结束后的收尾工作。。。。

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import unittest


class Browser(unittest.TestCase):
    """
    加载浏览器和测试前后准备工作,利用chrome无头浏览器
    """

    @classmethod
    def setUpClass(cls):
        chrome_options = Options()
        chrome_options.add_argument('--headless')
        chrome_options.add_argument('--disable-gpu')

        cls.driver = webdriver.Chrome(chrome_options=chrome_options)
        # cls.driver = webdriver.Chrome()    在设定一个界面的chrome,便于调试
        cls.driver.implicitly_wait(5)
        cls.driver.delete_all_cookies()

    @classmethod
    def tearDownClass(cls):
        # cls.driver.close()
        cls.driver.quit()


if __name__ == "__main__":
    unittest.main()

browser