Validate async paginated callbacks return awaitables

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

Author: cursoragent

hyperbrowser/client/polling.py

@@ -420,7 +420,13 @@ async def collect_paginated_results_async(
         should_sleep = True
         try:
             previous_page_batch = current_page_batch
-            page_response = await get_next_page(current_page_batch + 1)
+            page_result = get_next_page(current_page_batch + 1)
+            page_awaitable = _ensure_awaitable(
+                page_result,
+                callback_name="get_next_page",
+                operation_name=operation_name,
+            )
+            page_response = await page_awaitable
             on_page_success(page_response)
             current_page_batch = get_current_page_batch(page_response)
             total_page_batches = get_total_page_batches(page_response)

tests/test_polling.py

@@ -332,6 +332,25 @@ async def run() -> None:
     asyncio.run(run())
 
 
+def test_collect_paginated_results_async_rejects_non_awaitable_page_callback_result():
+    async def run() -> None:
+        with pytest.raises(
+            HyperbrowserError, match="get_next_page must return an awaitable"
+        ):
+            await collect_paginated_results_async(
+                operation_name="async paginated awaitable validation",
+                get_next_page=lambda page: {"current": 1, "total": 1, "items": []},  # type: ignore[return-value]
+                get_current_page_batch=lambda response: response["current"],
+                get_total_page_batches=lambda response: response["total"],
+                on_page_success=lambda response: None,
+                max_wait_seconds=1.0,
+                max_attempts=2,
+                retry_delay_seconds=0.0001,
+            )
+
+    asyncio.run(run())
+
+
 def test_collect_paginated_results_async_allows_single_page_on_zero_max_wait():
     async def run() -> None:
         collected = []