chore: mark Registry test helpers; fix pyright on pool cleanup

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: Reflex

src/runloop_api_client/_base_client.py

@@ -161,13 +161,14 @@ def acquire(self, shard: int) -> _SharedTransport:
                 return transport
 
         def take_all(self) -> list[_SharedTransport]:
-            """Remove and return all transports (test cleanup)."""
+            """Test-only: remove and return all transports for fixture cleanup."""
             with _pool_lock:
                 values = list(self._shards.values())
                 self._shards.clear()
                 return values
 
         def shard_ids(self) -> set[int]:
+            """Test-only: which shard indexes currently hold a shared transport."""
             with _pool_lock:
                 return set(self._shards)
 
@@ -229,6 +230,7 @@ def acquire(self, loop: asyncio.AbstractEventLoop, shard: int) -> _SharedAsyncTr
                 return transport
 
         def clear(self) -> None:
+            """Test-only: drop all per-loop shard maps (does not close transports)."""
             with _pool_lock:
                 self._by_loop.clear()
 

tests/test_h2_bulkhead_integration.py

@@ -36,7 +36,7 @@
 
 
 def _clear_pool_state() -> None:
-    old = []
+    old: list[_base_mod._SharedTransport] = []
     old.extend(_base_mod._shared_sync_api_transports.take_all())
     old.extend(_base_mod._shared_sync_background_transports.take_all())
     old.extend(_base_mod._shared_sync_transfer_transports.take_all())
@@ -49,6 +49,7 @@ def _clear_pool_state() -> None:
         except Exception:
             pass
 
+
 @pytest.fixture(autouse=True)
 def _reset_pools() -> Iterator[None]:  # pyright: ignore[reportUnusedFunction]
     _clear_pool_state()

tests/test_shared_pool.py

@@ -28,7 +28,7 @@ def _reset_shared_pool() -> Iterator[None]:  # pyright: ignore[reportUnusedFunct
 
 
 def _clear_pool_state() -> None:
-    old = []
+    old: list[_base_mod._SharedTransport] = []
     old.extend(_base_mod._shared_sync_api_transports.take_all())
     old.extend(_base_mod._shared_sync_background_transports.take_all())
     old.extend(_base_mod._shared_sync_transfer_transports.take_all())
@@ -41,6 +41,7 @@ def _clear_pool_state() -> None:
         except Exception:
             pass
 
+
 def _make_client(**kwargs: Any) -> Runloop:
     kwargs.setdefault("base_url", base_url)
     kwargs.setdefault("bearer_token", bearer_token)

tests/test_transfer_client.py

@@ -42,7 +42,7 @@ def _reset_shared_pool() -> Iterator[None]:  # pyright: ignore[reportUnusedFunct
 
 
 def _clear_pool_state() -> None:
-    old = []
+    old: list[_base_mod._SharedTransport] = []
     old.extend(_base_mod._shared_sync_api_transports.take_all())
     old.extend(_base_mod._shared_sync_background_transports.take_all())
     old.extend(_base_mod._shared_sync_transfer_transports.take_all())