Validate async retry operations return awaitables

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

Author: cursoragent

hyperbrowser/client/polling.py

@@ -298,7 +298,24 @@ async def retry_operation_async(
     failures = 0
     while True:
         try:
-            return await operation()
+            operation_result = operation()
+        except Exception as exc:
+            failures += 1
+            if failures >= max_attempts:
+                raise HyperbrowserError(
+                    f"{operation_name} failed after {max_attempts} attempts: {exc}"
+                ) from exc
+            await asyncio.sleep(retry_delay_seconds)
+            continue
+
+        operation_awaitable = _ensure_awaitable(
+            operation_result,
+            callback_name="operation",
+            operation_name=operation_name,
+        )
+
+        try:
+            return await operation_awaitable
         except Exception as exc:
             failures += 1
             if failures >= max_attempts:

tests/test_polling.py

@@ -177,6 +177,21 @@ async def operation() -> str:
     asyncio.run(run())
 
 
+def test_retry_operation_async_rejects_non_awaitable_operation_result() -> None:
+    async def run() -> None:
+        with pytest.raises(
+            HyperbrowserError, match="operation must return an awaitable"
+        ):
+            await retry_operation_async(
+                operation_name="invalid-async-retry-awaitable",
+                operation=lambda: "ok",  # type: ignore[return-value]
+                max_attempts=2,
+                retry_delay_seconds=0.0001,
+            )
+
+    asyncio.run(run())
+
+
 def test_async_poll_until_terminal_status_allows_immediate_terminal_on_zero_max_wait():
     async def run() -> None:
         status = await poll_until_terminal_status_async(