Use blockbuster to detect blocking calls

pyproject.toml

@@ -50,6 +50,7 @@ pytest-httpserver = "~1.0.8"
 testcontainers = "~3.7.1"
 ruff = "^0.6.8"
 types-toml = "^0.10.8.7"
+blockbuster = "~1.5.0"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

tests/conftest.py

@@ -20,9 +20,11 @@
 
 import functools
 import warnings
+from collections.abc import Iterator
 from typing import Any, Awaitable, Callable, TypedDict
 
 import pytest
+from blockbuster import BlockBuster, blockbuster_ctx
 from deprecation import UnsupportedWarning
 
 from astrapy import DataAPIClient
@@ -51,6 +53,31 @@
 )
 
 
+@pytest.fixture(autouse=True)
+def blockbuster() -> Iterator[BlockBuster]:
+    with blockbuster_ctx() as bb:
+        bb.functions["os.stat"].can_block_functions.extend(
+            [
+                # TODO: follow discussion in https://github.com/encode/httpx/discussions/3456
+                ("httpx/_client.py", {"_init_transport"}),
+                ("coverage/control.py", {"_should_trace"}),
+            ]
+        )
+        bb.functions["os.path.abspath"].can_block_functions.extend(
+            [
+                ("coverage/control.py", {"_should_trace"}),
+                ("_pytest/assertion/rewrite.py", {"_should_rewrite"}),
+            ]
+        )
+        bb.functions["os.mkdir"].can_block_functions.append(
+            ("_pytest/assertion/rewrite.py", {"try_makedirs"})
+        )
+        bb.functions["os.replace"].can_block_functions.append(
+            ("_pytest/assertion/rewrite.py", {"_write_pyc"})
+        )
+        yield bb
+
+
 class DataAPICredentials(TypedDict):
     token: str | TokenProvider
     api_endpoint: str

tests/idiomatic/integration/test_admin.py

@@ -14,6 +14,7 @@
 
 from __future__ import annotations
 
+import asyncio
 import time
 from typing import Any, Awaitable, Callable
 
@@ -73,7 +74,7 @@ async def await_until_true(
     while not (await acondition()):
         if time.time() - ini_time > max_seconds:
             raise ValueError("Timed out on condition.")
-        time.sleep(poll_interval)
+        await asyncio.sleep(poll_interval)
 
 
 @pytest.mark.skipif(not IS_ASTRA_DB, reason="Not supported outside of Astra DB")

tests/idiomatic/integration/test_ddl_async.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-import time
+import asyncio
 
 import pytest
 
@@ -146,7 +146,7 @@ async def test_collection_default_id_type_async(
         assert isinstance(doc["_id"], UUID)
         await acol.drop()
 
-        time.sleep(2)
+        await asyncio.sleep(2)
         acol = await async_database.create_collection(
             ID_TEST_COLLECTION_NAME_ROOT + DefaultIdType.UUIDV6,
             definition=CollectionDefinition(
@@ -168,7 +168,7 @@ async def test_collection_default_id_type_async(
         assert doc["_id"].version == 6
         await acol.drop()
 
-        time.sleep(2)
+        await asyncio.sleep(2)
         acol = await async_database.create_collection(
             ID_TEST_COLLECTION_NAME_ROOT + DefaultIdType.UUIDV7,
             definition=CollectionDefinition(
@@ -190,7 +190,7 @@ async def test_collection_default_id_type_async(
         assert doc["_id"].version == 7
         await acol.drop()
 
-        time.sleep(2)
+        await asyncio.sleep(2)
         acol = await async_database.create_collection(
             ID_TEST_COLLECTION_NAME_ROOT + DefaultIdType.DEFAULT,
             definition=CollectionDefinition(
@@ -205,7 +205,7 @@ async def test_collection_default_id_type_async(
         assert acol_options.default_id.default_id_type == DefaultIdType.DEFAULT
         await acol.drop()
 
-        time.sleep(2)
+        await asyncio.sleep(2)
         acol = await async_database.create_collection(
             ID_TEST_COLLECTION_NAME_ROOT + DefaultIdType.OBJECTID,
             definition=CollectionDefinition(
@@ -429,7 +429,7 @@ async def test_tokenless_client_async(
     @pytest.mark.describe(
         "test database-from-admin default keyspace per environment, async"
     )
-    async def test_database_from_admin_default_keyspace_per_environment_async(
+    def test_async_database_from_admin_default_keyspace_per_environment(
         self,
         data_api_credentials_kwargs: DataAPICredentials,
         data_api_credentials_info: DataAPICredentialsInfo,
@@ -448,7 +448,7 @@ async def test_database_from_admin_default_keyspace_per_environment_async(
     @pytest.mark.describe(
         "test database-from-astradbadmin default keyspace per environment, async"
     )
-    async def test_database_from_astradbadmin_default_keyspace_per_environment_async(
+    def test_async_database_from_astradbadmin_default_keyspace_per_environment(
         self,
         data_api_credentials_kwargs: DataAPICredentials,
         data_api_credentials_info: DataAPICredentialsInfo,

tests/idiomatic/integration/test_timeout_async.py

@@ -80,11 +80,11 @@ async def test_cursor_overalltimeout_exceptions_async(
         acol = async_empty_collection
         await acol.insert_many([{"a": 1}] * 1000)
 
-        await acol.distinct("a", timeout_ms=20000)
+        await acol.distinct("a", timeout_ms=60000)
         with pytest.raises(DataAPITimeoutException):
             await acol.distinct("a", timeout_ms=1)
 
-        await acol.distinct("a", timeout_ms=20000)
+        await acol.distinct("a", timeout_ms=60000)
         with pytest.raises(DataAPITimeoutException):
             await acol.distinct("a", timeout_ms=1)