Skip to content

Commit 820fee6

Browse files
committed
Update tests
1 parent 7d5208e commit 820fee6

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

py/test/selenium/webdriver/common/bidi_network_tests.py

+13-4
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@
1717
import pytest
1818

1919
from selenium.webdriver.common.bidi.network import BeforeRequestSentParameters, ContinueRequestParameters
20-
20+
from selenium.webdriver.common.bidi.cdp import open_cdp
21+
import trio
2122

2223
@pytest.mark.xfail_safari
2324
@pytest.mark.xfail_firefox
@@ -35,6 +36,14 @@ def request_handler(params: BeforeRequestSentParameters):
3536
}
3637
return ContinueRequestParameters(**json)
3738

38-
await driver.network.add_request_handler(request_filter, request_handler)
39-
driver.get("https://www.example.com/")
40-
assert driver.current_url == "https://www.selenium.dev/about/"
39+
ws_url = driver.caps.get("webSocketUrl")
40+
async with open_cdp(ws_url) as conn:
41+
async with trio.open_nursery() as nursery:
42+
nursery.start_soon(driver.network.add_request_handler, request_filter, request_handler, conn)
43+
await trio.sleep(1)
44+
await driver.network.get("https://www.example.com",conn)
45+
assert "Selenium" in driver.title
46+
await trio.sleep(1)
47+
await driver.network.remove_request_handler()
48+
await driver.network.get("https://www.example.com",conn)
49+
assert "Example" in driver.title

0 commit comments

Comments
 (0)