INTPYTHON-770 Make ttl test more reliable (#224)

Author: blink1073

libs/langgraph-store-mongodb/tests/integration_tests/test_store_semantic_async.py

@@ -22,7 +22,7 @@
 DB_NAME = os.environ.get("DB_NAME", "langgraph-test")
 COLLECTION_NAME = "semantic_search_async"
 INDEX_NAME = "vector_index"
-TIMEOUT, INTERVAL = 30, 1  # timeout to index new data
+TIMEOUT, INTERVAL = 60, 1  # timeout to index new data
 
 DIMENSIONS = 5  # Dimensions of embedding model
 

libs/langgraph-store-mongodb/tests/unit_tests/test_store.py

@@ -1,4 +1,5 @@
 import os
+import time
 from collections.abc import Generator
 from datetime import datetime
 
@@ -180,6 +181,8 @@ def test_ttl() -> None:
         res = store.collection.find_one({})
         assert res is not None
         orig_updated_at = res["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        time.sleep(0.1)
         res = store.get(namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})
@@ -200,6 +203,8 @@ def test_ttl() -> None:
         found = store.collection.find_one({})
         assert found is not None
         orig_updated_at = found["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        time.sleep(0.1)
         res = store.get(namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})
@@ -220,6 +225,8 @@ def test_ttl() -> None:
         found = store.collection.find_one({})
         assert found is not None
         orig_updated_at = found["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        time.sleep(0.1)
         res = store.get(namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})
@@ -240,6 +247,8 @@ def test_ttl() -> None:
         found = store.collection.find_one({})
         assert found is not None
         orig_updated_at = found["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        time.sleep(0.1)
         res = store.get(refresh_ttl=False, namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})

libs/langgraph-store-mongodb/tests/unit_tests/test_store_async.py

@@ -1,3 +1,4 @@
+import asyncio
 import os
 from collections.abc import AsyncGenerator
 from datetime import datetime
@@ -184,6 +185,8 @@ async def test_ttl() -> None:
         res = store.collection.find_one({})
         assert res is not None
         orig_updated_at = res["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        await asyncio.sleep(0.1)
         res = await store.aget(namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})
@@ -204,6 +207,8 @@ async def test_ttl() -> None:
         found = store.collection.find_one({})
         assert found is not None
         orig_updated_at = found["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        await asyncio.sleep(0.1)
         res = await store.aget(namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})
@@ -224,6 +229,8 @@ async def test_ttl() -> None:
         found = store.collection.find_one({})
         assert found is not None
         orig_updated_at = found["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        await asyncio.sleep(0.1)
         res = await store.aget(namespace=namespace, key=key)
         assert res is not None
         found = store.collection.find_one({})
@@ -244,9 +251,11 @@ async def test_ttl() -> None:
         found = store.collection.find_one({})
         assert found is not None
         orig_updated_at = found["updated_at"]
+        # Add a delay to ensure a different timestamp.
+        await asyncio.sleep(0.1)
+        found = store.collection.find_one({})
         res = await store.aget(refresh_ttl=False, namespace=namespace, key=key)
         assert res is not None
-        found = store.collection.find_one({})
         assert found is not None
         new_updated_at = found["updated_at"]
         assert new_updated_at == orig_updated_at