Skip to content

Commit 862cd7e

Browse files
chore(internal): update client tests (#121)
1 parent 797daee commit 862cd7e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

Diff for: tests/test_client.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323

2424
from browserbase import Browserbase, AsyncBrowserbase, APIResponseValidationError
2525
from browserbase._types import Omit
26+
from browserbase._utils import maybe_transform
2627
from browserbase._models import BaseModel, FinalRequestOptions
2728
from browserbase._constants import RAW_RESPONSE_HEADER
2829
from browserbase._exceptions import APIStatusError, APITimeoutError, BrowserbaseError, APIResponseValidationError
@@ -32,6 +33,7 @@
3233
BaseClient,
3334
make_request_options,
3435
)
36+
from browserbase.types.session_create_params import SessionCreateParams
3537

3638
from .utils import update_env
3739

@@ -727,7 +729,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
727729
with pytest.raises(APITimeoutError):
728730
self.client.post(
729731
"/v1/sessions",
730-
body=cast(object, dict(project_id="your_project_id")),
732+
body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
731733
cast_to=httpx.Response,
732734
options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
733735
)
@@ -742,7 +744,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
742744
with pytest.raises(APIStatusError):
743745
self.client.post(
744746
"/v1/sessions",
745-
body=cast(object, dict(project_id="your_project_id")),
747+
body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
746748
cast_to=httpx.Response,
747749
options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
748750
)
@@ -1507,7 +1509,7 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
15071509
with pytest.raises(APITimeoutError):
15081510
await self.client.post(
15091511
"/v1/sessions",
1510-
body=cast(object, dict(project_id="your_project_id")),
1512+
body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
15111513
cast_to=httpx.Response,
15121514
options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
15131515
)
@@ -1522,7 +1524,7 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
15221524
with pytest.raises(APIStatusError):
15231525
await self.client.post(
15241526
"/v1/sessions",
1525-
body=cast(object, dict(project_id="your_project_id")),
1527+
body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
15261528
cast_to=httpx.Response,
15271529
options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
15281530
)

0 commit comments

Comments
 (0)