@@ -432,9 +432,7 @@ def test_rewrite_direct_vm_options_keeps_telemetry_events_on_control_plane() ->
432432 )
433433
434434 cache = BrowserRouteCache ()
435- cache .set (
436- BrowserRoute (session_id = "sess-1" , base_url = "http://browser-session.test/browser/kernel" , jwt = "token-abc" )
437- )
435+ cache .set (BrowserRoute (session_id = "sess-1" , base_url = "http://browser-session.test/browser/kernel" , jwt = "token-abc" ))
438436 config = BrowserRoutingConfig (subresources = ("curl" , "telemetry/stream" , "computer" , "playwright" ))
439437
440438 events = rewrite_direct_vm_options (
@@ -511,9 +509,7 @@ def test_process_fs_and_telemetry_events_stay_on_api_origin_by_default(
511509 fs_read = respx .get (f"{ base_url } /browsers/sess-1/fs/read_file" ).mock (
512510 return_value = httpx .Response (200 , content = b"x" , headers = {"content-type" : "application/octet-stream" })
513511 )
514- events = respx .get (f"{ base_url } /browsers/sess-1/telemetry/events" ).mock (
515- return_value = httpx .Response (200 , json = [])
516- )
512+ events = respx .get (f"{ base_url } /browsers/sess-1/telemetry/events" ).mock (return_value = httpx .Response (200 , json = []))
517513 with Kernel (base_url = base_url , api_key = api_key , _strict_response_validation = True ) as client :
518514 _cache_browser (client )
519515 client .browsers .process .exec ("sess-1" , command = "echo" )
@@ -530,7 +526,11 @@ def test_stale_direct_vm_jwt_evicts_cache_and_retries_control_plane(
530526 monkeypatch : pytest .MonkeyPatch ,
531527) -> None :
532528 monkeypatch .delenv ("KERNEL_BROWSER_ROUTING_SUBRESOURCES" , raising = False )
533- monkeypatch .setattr ("kernel._base_client.SyncAPIClient._sleep_for_retry" , lambda * args , ** kwargs : None )
529+
530+ def _skip_retry_sleep (_self : object , ** _kwargs : object ) -> None :
531+ return None
532+
533+ monkeypatch .setattr ("kernel._base_client.SyncAPIClient._sleep_for_retry" , _skip_retry_sleep )
534534 vm = respx .post ("http://browser-session.test/browser/kernel/computer/screenshot" ).mock (
535535 return_value = httpx .Response (401 , text = "Invalid JWT" )
536536 )
0 commit comments