2727from selenium .webdriver .remote .client_config import ClientConfig
2828
2929
30- def test_browser_specific_method (driver , pages ):
30+ def test_browser_specific_method (firefox_options , webserver ):
3131 """This only works on Firefox"""
32- pages .load ("simpleTest.html" )
33- screenshot = driver .execute ("FULL_PAGE_SCREENSHOT" )["value" ]
34- result = base64 .b64decode (screenshot )
35- kind = filetype .guess (result )
36- assert kind is not None and kind .mime == "image/png"
32+ server_addr = f"http://{ webserver .host } :{ webserver .port } "
33+ with webdriver .Remote (options = firefox_options ) as driver :
34+ driver .get (f"{ server_addr } /simpleTest.html" )
35+ screenshot = driver .execute ("FULL_PAGE_SCREENSHOT" )["value" ]
36+ result = base64 .b64decode (screenshot )
37+ kind = filetype .guess (result )
38+ assert kind is not None and kind .mime == "image/png"
3739
3840
39- def test_remote_webdriver_with_http_timeout (firefox_options , webserver ):
41+ def test_remote_webdriver_with_http_timeout (chromium_options , webserver ):
4042 """This test starts a remote webdriver with an http client timeout
4143 set less than the implicit wait timeout, and verifies the http timeout
4244 is triggered first when waiting for an element.
@@ -46,14 +48,14 @@ def test_remote_webdriver_with_http_timeout(firefox_options, webserver):
4648 server_addr = f"http://{ webserver .host } :{ webserver .port } "
4749 client_config = ClientConfig (remote_server_addr = server_addr , timeout = http_timeout )
4850 assert client_config .timeout == http_timeout
49- with webdriver .Remote (options = firefox_options , client_config = client_config ) as driver :
51+ with webdriver .Remote (options = chromium_options , client_config = client_config ) as driver :
5052 driver .get (f"{ server_addr } /simpleTest.html" )
5153 driver .implicitly_wait (wait_timeout )
5254 with pytest .raises (ReadTimeoutError ):
5355 driver .find_element (By .ID , "no_element_to_be_found" )
5456
5557
56- def test_remote_webdriver_with_websocket_timeout (firefox_options , webserver ):
58+ def test_remote_webdriver_with_websocket_timeout (chromium_options , webserver ):
5759 """This test starts a remote webdriver that uses websockets, and has a websocket
5860 client timeout less than the default. It verifies the websocket times out according
5961 to this value.
@@ -66,8 +68,8 @@ def test_remote_webdriver_with_websocket_timeout(firefox_options, webserver):
6668 remote_server_addr = server_addr , websocket_timeout = websocket_timeout , websocket_interval = websocket_interval
6769 )
6870 assert client_config .websocket_timeout == websocket_timeout
69- firefox_options .enable_bidi = True
70- with webdriver .Remote (options = firefox_options , client_config = client_config ) as driver :
71+ chromium_options .enable_bidi = True
72+ with webdriver .Remote (options = chromium_options , client_config = client_config ) as driver :
7173 driver ._start_bidi ()
7274 assert driver ._websocket_connection .response_wait_timeout == websocket_timeout
7375 assert driver ._websocket_connection .response_wait_interval == websocket_interval
0 commit comments