File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change 66from _pytest .fixtures import FixtureRequest
77from selenium import webdriver
88from selenium .common import WebDriverException
9+ from selenium .webdriver .support .wait import WebDriverWait
10+ from selenium .webdriver .support import expected_conditions as EC
911
1012import requestium
1113
@@ -39,7 +41,9 @@ def _create_chrome_driver(headless: bool) -> webdriver.Chrome:
3941 options .add_argument ("--disable-dev-shm-usage" )
4042 if headless :
4143 options .add_argument ("--headless=new" )
42- return webdriver .Chrome (options = options )
44+ driver = webdriver .Chrome (options = options )
45+ WebDriverWait (driver , 5 ).until (EC .number_of_windows_to_be (1 ))
46+ return driver
4347
4448
4549def _create_firefox_driver (headless : bool ) -> webdriver .Firefox :
@@ -50,7 +54,9 @@ def _create_firefox_driver(headless: bool) -> webdriver.Firefox:
5054 options .set_preference ("network.http.use-cache" , False )
5155 if headless :
5256 options .add_argument ("--headless" )
53- return webdriver .Firefox (options = options )
57+ driver = webdriver .Firefox (options = options )
58+ WebDriverWait (driver , 5 ).until (EC .number_of_windows_to_be (1 ))
59+ return driver
5460
5561
5662@pytest .fixture (
You can’t perform that action at this time.
0 commit comments