Fix tests

libs/core/pyproject.toml

@@ -109,7 +109,7 @@ grandalf = "^0.8"
 responses = "^0.25.0"
 pytest-socket = "^0.7.0"
 pytest-xdist = "^3.6.1"
-blockbuster = "~1.5.8"
+blockbuster = "~1.5.9"
 [[tool.poetry.group.test.dependencies.numpy]]
 version = "^1.24.0"
 python = "<3.12"

libs/core/tests/unit_tests/conftest.py

@@ -14,15 +14,22 @@
 def blockbuster() -> Iterator[BlockBuster]:
     with blockbuster_ctx() as bb:
         for func in ["os.stat", "os.path.abspath"]:
-            bb.functions[func].can_block_in(
-                "langchain_core/_api/internal.py", "is_caller_internal"
+            (
+                bb.functions[func]
+                .can_block_in("langchain_core/_api/internal.py", "is_caller_internal")
+                .can_block_in("langchain_core/runnables/base.py", "__repr__")
             )
 
         for func in ["os.stat", "io.TextIOWrapper.read"]:
             bb.functions[func].can_block_in(
                 "langsmith/client.py", "_default_retry_config"
             )
 
+        for bb_function in bb.functions.values():
+            bb_function.can_block_in(
+                "freezegun/api.py", "_get_cached_module_attributes"
+            )
+
         yield bb
 
 

libs/core/tests/unit_tests/language_models/chat_models/test_rate_limiting.py

@@ -1,11 +1,20 @@
 import time
 from typing import Optional as Optional
 
+import pytest
+from blockbuster import BlockBuster
+
 from langchain_core.caches import InMemoryCache
 from langchain_core.language_models import GenericFakeChatModel
 from langchain_core.rate_limiters import InMemoryRateLimiter
 
 
+@pytest.fixture(autouse=True)
+def deactivate_blockbuster(blockbuster: BlockBuster) -> None:
+    # Deactivate BlockBuster to not disturb the rate limiter timings
+    blockbuster.deactivate()
+
+
 def test_rate_limit_invoke() -> None:
     """Add rate limiter."""
 

libs/core/tests/unit_tests/runnables/test_fallbacks.py

@@ -96,9 +96,7 @@ def test_fallbacks(
     "runnable",
     ["llm", "llm_multi", "chain", "chain_pass_exceptions"],
 )
-async def test_fallbacks_async(
-    runnable: RunnableWithFallbacks, request: Any, snapshot: SnapshotAssertion
-) -> None:
+async def test_fallbacks_async(runnable: RunnableWithFallbacks, request: Any) -> None:
     runnable = request.getfixturevalue(runnable)
     assert await runnable.ainvoke("hello") == "bar"
     assert await runnable.abatch(["hi", "hey", "bye"]) == ["bar"] * 3

libs/core/tests/unit_tests/tracers/test_memory_stream.py

@@ -69,7 +69,7 @@ async def producer() -> None:
         """Produce items with slight delay."""
         tic = time.time()
         for i in range(3):
-            await asyncio.sleep(0.10)
+            await asyncio.sleep(0.2)
             toc = time.time()
             await writer.send(
                 {
@@ -96,6 +96,8 @@ async def consumer() -> AsyncIterator[dict]:
     items = [item async for item in consumer()]
     await task
 
+    assert len(items) == 3
+
     for item in items:
         delta_time = item["receive_time"] - item["produce_time"]
         # Allow a generous 10ms of delay