Skip to content

Commit 3580671

Browse files
committed
Fix ruff import order and unused lambda args
Lint failed on the stale-JWT retry change.
1 parent f525a24 commit 3580671

2 files changed

Lines changed: 8 additions & 8 deletions

File tree

src/kernel/_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@
4141
strip_direct_vm_auth,
4242
rewrite_direct_vm_options,
4343
browser_routing_config_from_env,
44-
maybe_evict_browser_route_from_response,
4544
should_retry_stale_direct_vm_auth,
45+
maybe_evict_browser_route_from_response,
4646
maybe_populate_browser_route_cache_from_response,
4747
)
4848

tests/test_browser_routing.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)