Selenium：设置无头浏览器

在使用selenium执行测试用例时，每条用例执行过程中都会打开一个浏览器，如果用例数量过多时，每次运行均需要调用一次浏览器，增加了服务器压力，而无头模式就可以解决这种问题，他可以让运行速度更快，占用的资源也更少，让浏览器偷偷的在后台工作。

无头模式，是webdriver浏览器驱动的一个功能，可以支持不打开浏览器，直接跟网页进行交互，能够模拟真实得到浏览器进行操作。

操作环境

Windows 10
Python 3.9.1
Selenium 4.0.0
Chrome 浏览器

使用方法

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

chrome_options = Options()
chrome_options.add_argument('--headless')
driver = webdriver.Chrome(options=chrome_options)

# 通过driver.get()来打开对应链接
driver.get(url="https://www.baidu.com/")

示例

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


chrome_options = Options()
chrome_options.add_argument('--headless')


class TestDemo:

    # driver = webdriver.Chrome(options=chrome_options)

    @classmethod
    def setup(cls):
        cls.driver = webdriver.Chrome(options=chrome_options)
        cls.driver.implicitly_wait(20)
        # cls.driver.maximize_window()
        cls.driver.get(url="https://www.baidu.com/")

    @classmethod
    def teardown(cls):
        cls.driver.quit()
        
    def test_1_a(self):
        self.driver.find_element('id', 'kw').send_keys("QQ")
        self.driver.find_element('id', 'su').click()
        time.sleep(3)
        assert self.driver.title == 'QQ_百度搜索'

    def test_2_a(self):
        assert self.driver.current_url == 'https://www.baidu.com/'


if __name__ == '__main__':
    pytest.main(['-s', 'case.py'])