Skip to content

Commit b0a9b91

Browse files
PYTHON-5599 Fixes [BuildFailure] [AI-ML] test_time_travel on remote (#219)
Removed drop_database. Renamed db and collections to be in line with others in buildvariant.
1 parent caace80 commit b0a9b91

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

libs/langgraph-checkpoint-mongodb/tests/unit_tests/test_time_travel.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -43,16 +43,13 @@ def validate_expense_node(state: ExpenseState) -> dict[str, Any]:
4343
ids=["ttl_none", "ttl_3600"],
4444
)
4545
def checkpointer(request: Any) -> Generator[MongoDBSaver]:
46-
db_name = "langgraph_timetravel_db"
47-
checkpoint_collection_name = "checkpoints"
48-
writes_collection_name = "checkpoint_writes"
46+
db_name = "langgraph-test"
47+
checkpoint_collection_name = "timetravel_checkpoints"
48+
writes_collection_name = "timetravel_writes"
4949

5050
# Initialize MongoDB checkpointer
5151
client: MongoClient = MongoClient(MONGODB_URI)
5252

53-
# Clean up any existing test data.
54-
client.drop_database(db_name)
55-
5653
saver = MongoDBSaver(
5754
client=client,
5855
db_name=db_name,

0 commit comments

Comments
 (0)