Fixed warning about an unclosed MongoClient

agronholm · Sep 22, 2024 · 0bcb319 · 0bcb319
1 parent 90a775c
commit 0bcb319
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_datastores.py b/tests/test_datastores.py
@@ -857,5 +857,8 @@ async def test_psycopg(self) -> None:
         )
 
     async def test_mongodb(self) -> None:
-        data_store = MongoDBDataStore("mongodb://localhost")
-        assert repr(data_store) == "MongoDBDataStore(host=[('localhost', 27017)])"
+        from pymongo import MongoClient
+
+        with MongoClient() as client:
+            data_store = MongoDBDataStore(client)
+            assert repr(data_store) == "MongoDBDataStore(host=[('localhost', 27017)])"