[py] Fix test

Author: cgoldberg

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

@@ -17,6 +17,7 @@
 
 import pytest
 
+from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.bidi.browsing_context import ReadinessState
 from selenium.webdriver.common.bidi.network import Request
 from selenium.webdriver.common.by import By
@@ -101,17 +102,19 @@ def test_remove_auth_handler(driver):
 @pytest.mark.xfail_firefox(reason="Data URLs in Network requests are not implemented in Firefox yet")
 def test_handler_with_data_url_request(driver, pages):
     data_requests = []
+    exceptions = []
 
     def callback(request: Request):
         if request.url.startswith("data:"):
             data_requests.append(request)
-        request.continue_request()
+        try:
+            request.continue_request()
+        except WebDriverException as e:
+            exceptions.append(e)
 
     driver.network.add_request_handler("before_request", callback)
     url = pages.url("data_url.html")
     driver.browsing_context.navigate(context=driver.current_window_handle, url=url, wait=ReadinessState.COMPLETE)
-
-    # Assert that the BiDi event was captured.
-    assert len(data_requests) > 0
-    # Assert that the image is displayed.
     assert driver.find_element(By.ID, "data-url-image").is_displayed()
+    assert len(data_requests) > 0, "BiDi event not captured"
+    assert len(exceptions) == 0, "Exception raised when continuing request in callback"