Wrap missing computer action endpoint attribute errors

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

Author: cursoragent

hyperbrowser/client/managers/async_manager/computer_action.py

@@ -37,7 +37,17 @@ async def _execute_request(
                 "session must be a plain string session ID or SessionDetail"
             )
 
-        if not session.computer_action_endpoint:
+        try:
+            computer_action_endpoint = session.computer_action_endpoint
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "session must include computer_action_endpoint",
+                original_error=exc,
+            ) from exc
+
+        if not computer_action_endpoint:
             raise HyperbrowserError(
                 "Computer action endpoint not available for this session"
             )
@@ -48,7 +58,7 @@ async def _execute_request(
             payload = params
 
         response = await self._client.transport.post(
-            session.computer_action_endpoint,
+            computer_action_endpoint,
             data=payload,
         )
         return parse_response_model(

hyperbrowser/client/managers/sync_manager/computer_action.py

@@ -37,7 +37,17 @@ def _execute_request(
                 "session must be a plain string session ID or SessionDetail"
             )
 
-        if not session.computer_action_endpoint:
+        try:
+            computer_action_endpoint = session.computer_action_endpoint
+        except HyperbrowserError:
+            raise
+        except Exception as exc:
+            raise HyperbrowserError(
+                "session must include computer_action_endpoint",
+                original_error=exc,
+            ) from exc
+
+        if not computer_action_endpoint:
             raise HyperbrowserError(
                 "Computer action endpoint not available for this session"
             )
@@ -48,7 +58,7 @@ def _execute_request(
             payload = params
 
         response = self._client.transport.post(
-            session.computer_action_endpoint,
+            computer_action_endpoint,
             data=payload,
         )
         return parse_response_model(

tests/test_computer_action_manager.py

@@ -41,6 +41,29 @@ async def run() -> None:
     asyncio.run(run())
 
 
+def test_sync_computer_action_manager_wraps_missing_endpoint_attribute():
+    manager = SyncComputerActionManager(_DummyClient())
+
+    with pytest.raises(
+        HyperbrowserError, match="session must include computer_action_endpoint"
+    ) as exc_info:
+        manager.screenshot(SimpleNamespace())
+
+    assert isinstance(exc_info.value.original_error, AttributeError)
+
+
+def test_async_computer_action_manager_wraps_missing_endpoint_attribute():
+    async def run() -> None:
+        manager = AsyncComputerActionManager(_DummyClient())
+        with pytest.raises(
+            HyperbrowserError, match="session must include computer_action_endpoint"
+        ) as exc_info:
+            await manager.screenshot(SimpleNamespace())
+        assert isinstance(exc_info.value.original_error, AttributeError)
+
+    asyncio.run(run())
+
+
 def test_sync_computer_action_manager_rejects_string_subclass_session_ids():
     class _SessionId(str):
         pass