Preserve HTTP success status codes in transport JSON responses

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

hyperbrowser/transport/async_transport.py

@@ -70,7 +70,7 @@ async def _handle_response(self, response: httpx.Response) -> APIResponse:
             try:
                 if not response.content:
                     return APIResponse.from_status(normalized_status_code)
-                return APIResponse(response.json())
+                return APIResponse(response.json(), status_code=normalized_status_code)
             except Exception as e:
                 if normalized_status_code >= 400:
                     try:

hyperbrowser/transport/sync.py

@@ -58,7 +58,7 @@ def _handle_response(self, response: httpx.Response) -> APIResponse:
             try:
                 if not response.content:
                     return APIResponse.from_status(normalized_status_code)
-                return APIResponse(response.json())
+                return APIResponse(response.json(), status_code=normalized_status_code)
             except Exception as e:
                 if normalized_status_code >= 400:
                     try:

tests/test_transport_response_handling.py

@@ -37,6 +37,18 @@ def json(self):
         raise RuntimeError("broken json")
 
 
+class _JsonCreatedResponse:
+    status_code = 201
+    content = b'{"created": true}'
+    text = '{"created": true}'
+
+    def raise_for_status(self) -> None:
+        return None
+
+    def json(self):
+        return {"created": True}
+
+
 class _BrokenJsonErrorResponse:
     status_code = 500
     content = b"{broken-json}"
@@ -151,6 +163,19 @@ def test_sync_handle_response_with_non_json_success_body_returns_status_only():
         transport.close()
 
 
+def test_sync_handle_response_preserves_non_200_success_status_codes():
+    transport = SyncTransport(api_key="test-key")
+    try:
+        api_response = transport._handle_response(
+            _JsonCreatedResponse()  # type: ignore[arg-type]
+        )
+
+        assert api_response.status_code == 201
+        assert api_response.data == {"created": True}
+    finally:
+        transport.close()
+
+
 def test_sync_handle_response_with_broken_json_success_payload_returns_status_only():
     transport = SyncTransport(api_key="test-key")
     try:
@@ -332,6 +357,22 @@ async def run() -> None:
     asyncio.run(run())
 
 
+def test_async_handle_response_preserves_non_200_success_status_codes():
+    async def run() -> None:
+        transport = AsyncTransport(api_key="test-key")
+        try:
+            api_response = await transport._handle_response(
+                _JsonCreatedResponse()  # type: ignore[arg-type]
+            )
+
+            assert api_response.status_code == 201
+            assert api_response.data == {"created": True}
+        finally:
+            await transport.close()
+
+    asyncio.run(run())
+
+
 def test_async_handle_response_with_broken_json_success_payload_returns_status_only():
     async def run() -> None:
         transport = AsyncTransport(api_key="test-key")