From f8e941c5239b1184d0f0d44228bda6ccf77785f2 Mon Sep 17 00:00:00 2001
From: "stainless-app[bot]"
 <142633134+stainless-app[bot]@users.noreply.github.com>
Date: Thu, 13 Feb 2025 03:21:35 +0000
Subject: [PATCH] chore(internal): update client tests

---
 tests/test_client.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/tests/test_client.py b/tests/test_client.py
index 20ac94b..1c703f0 100644
--- a/tests/test_client.py
+++ b/tests/test_client.py
@@ -23,6 +23,7 @@
 
 from browserbase import Browserbase, AsyncBrowserbase, APIResponseValidationError
 from browserbase._types import Omit
+from browserbase._utils import maybe_transform
 from browserbase._models import BaseModel, FinalRequestOptions
 from browserbase._constants import RAW_RESPONSE_HEADER
 from browserbase._exceptions import APIStatusError, APITimeoutError, BrowserbaseError, APIResponseValidationError
@@ -32,6 +33,7 @@
     BaseClient,
     make_request_options,
 )
+from browserbase.types.session_create_params import SessionCreateParams
 
 from .utils import update_env
 
@@ -727,7 +729,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
         with pytest.raises(APITimeoutError):
             self.client.post(
                 "/v1/sessions",
-                body=cast(object, dict(project_id="your_project_id")),
+                body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )
@@ -742,7 +744,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
         with pytest.raises(APIStatusError):
             self.client.post(
                 "/v1/sessions",
-                body=cast(object, dict(project_id="your_project_id")),
+                body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )
@@ -1507,7 +1509,7 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
         with pytest.raises(APITimeoutError):
             await self.client.post(
                 "/v1/sessions",
-                body=cast(object, dict(project_id="your_project_id")),
+                body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )
@@ -1522,7 +1524,7 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
         with pytest.raises(APIStatusError):
             await self.client.post(
                 "/v1/sessions",
-                body=cast(object, dict(project_id="your_project_id")),
+                body=cast(object, maybe_transform(dict(project_id="your_project_id"), SessionCreateParams)),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
             )