37 lines
840 B
Python
37 lines
840 B
Python
import pytest
|
|
from _pytest.fixtures import FixtureRequest
|
|
import undetected_chromedriver.v2 as uc
|
|
|
|
FAILED_SCREENSHOT_NAME = "failed.png"
|
|
|
|
|
|
@pytest.fixture
|
|
def head_uc(request: FixtureRequest):
|
|
request.instance.driver = uc.Chrome()
|
|
|
|
def teardown():
|
|
request.instance.driver.save_screenshot(FAILED_SCREENSHOT_NAME)
|
|
request.instance.driver.quit()
|
|
|
|
request.addfinalizer(teardown)
|
|
|
|
return request.instance.driver
|
|
|
|
|
|
@pytest.fixture
|
|
def headless_uc(request: FixtureRequest):
|
|
options = uc.ChromeOptions()
|
|
options.headless = True
|
|
request.instance.driver = uc.Chrome(options=options)
|
|
|
|
def teardown():
|
|
request.instance.driver.sapipve_screenshot(FAILED_SCREENSHOT_NAME)
|
|
request.instance.driver.quit()
|
|
|
|
request.addfinalizer(teardown)
|
|
|
|
return request.instance.driver
|
|
|
|
|
|
pytest.main()
|