fix(test): Fix DB connection name to support tox parallel runners

Author: abhinand-c

tests/asynchronous/all_warnings/test_warnings.py

@@ -10,12 +10,13 @@
 from mongoengine import *
 from mongoengine.base.common import _document_registry
 from tests.asynchronous.utils import reset_async_connections
+from tests.utils import MONGO_TEST_DB
 
 
 class TestAllWarnings(unittest.IsolatedAsyncioTestCase):
 
     async def asyncSetUp(self):
-        await async_connect(db="mongoenginetest")
+        await async_connect(db=MONGO_TEST_DB)
         self.warning_list = []
         self.showwarning_default = warnings.showwarning
         warnings.showwarning = self.append_to_warning_list

tests/asynchronous/document/test_class_methods.py

@@ -4,12 +4,13 @@
 from mongoengine.pymongo_support import async_list_collection_names
 from mongoengine.base.queryset import NULLIFY, PULL
 from tests.asynchronous.utils import reset_async_connections
+from tests.utils import MONGO_TEST_DB
 
 
 class TestClassMethods(unittest.IsolatedAsyncioTestCase):
 
     async def asyncSetUp(self):
-        await async_connect(db="mongoenginetest")
+        await async_connect(db=MONGO_TEST_DB)
         self.db = await async_get_db()
 
         class Person(Document):

tests/asynchronous/document/test_indexes.py

@@ -15,11 +15,12 @@
 )
 from mongoengine.registry import _CollectionRegistry
 from tests.asynchronous.utils import reset_async_connections
+from tests.utils import MONGO_TEST_DB
 
 
 class TestIndexes(unittest.IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        self.connection = await async_connect(db="mongoenginetest")
+        self.connection = await async_connect(db=MONGO_TEST_DB)
         self.db = async_get_db()
 
         class Person(Document):
@@ -981,7 +982,7 @@ async def test_indexes_after_database_drop(self):
         # Use a new connection and database since dropping the database could
         # cause concurrent tests to fail.
         tmp_alias = "test_indexes_after_database_drop"
-        connection = await async_connect(db="tempdatabase", alias=tmp_alias)
+        connection = await async_connect(db=f"{MONGO_TEST_DB}_tempdb", alias=tmp_alias)
 
         class BlogPost(Document):
             slug = StringField(unique=True)
@@ -993,7 +994,7 @@ class BlogPost(Document):
             await BlogPost(slug="test").asave()
 
         # Drop the Database
-        await connection.drop_database("tempdatabase")
+        await connection.drop_database(f"{MONGO_TEST_DB}_tempdb")
         await BlogPost(slug="test").asave()
         # No error because the index was not recreated after dropping the database.
         await BlogPost(slug="test").asave()
@@ -1012,12 +1013,12 @@ class BlogPost2(Document):
             await BlogPost2(slug="test").asave()
 
         # Drop the Database
-        await connection.drop_database("tempdatabase")
+        await connection.drop_database(f"{MONGO_TEST_DB}_tempdb")
         await BlogPost2(slug="test").asave()
         # Error because ensure_indexes is run on every save().
         with pytest.raises(NotUniqueError):
             await BlogPost2(slug="test").asave()
-        await connection.drop_database("tempdatabase")
+        await connection.drop_database(f"{MONGO_TEST_DB}_tempdb")
 
     async def test_index_dont_send_cls_option(self):
         """

tests/asynchronous/document/test_instance.py

@@ -49,6 +49,7 @@
     async_get_as_pymongo,
     requires_mongodb_gte_44, reset_async_connections,
 )
+from tests.utils import MONGO_TEST_DB
 
 TEST_IMAGE_PATH = os.path.join(os.path.dirname(__file__), "../fields/mongoengine.png")
 
@@ -491,7 +492,7 @@ class Animal(Document):
         CMD_QUERY_KEY = "command"
         async with async_query_counter() as q:
             await doc.areload()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.animal"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.animal"})).to_list())[0]
             assert set(query_op[CMD_QUERY_KEY]["filter"].keys()) == {
                 "_id",
                 "superphylum",
@@ -508,7 +509,7 @@ class Person(Document):
         CMD_QUERY_KEY = "command"
         async with async_query_counter() as q:
             await doc.areload()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.person"})).to_list())[0]
             assert set(query_op[CMD_QUERY_KEY]["filter"].keys()) == {"_id", "country"}
 
     async def test_reload_sharded_nested(self):
@@ -542,7 +543,7 @@ class Animal(Document):
         async with async_query_counter() as q:
             doc.name = "Cat"
             await doc.asave()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.animal"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.animal"})).to_list())[0]
             assert query_op["op"] == "update"
             assert set(query_op["command"]["q"].keys()) == {"_id", "is_mammal"}
 
@@ -564,7 +565,7 @@ class Animal(Document):
 
         async with async_query_counter() as q:
             await doc.asave()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.animal"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.animal"})).to_list())[0]
             assert query_op["op"] == "command"
             assert query_op["command"]["findAndModify"] == "animal"
             assert set(query_op["command"]["query"].keys()) == {"_id", "is_mammal"}
@@ -1894,23 +1895,23 @@ class AggPerson(Document):
 
         async with async_db_ops_tracker() as q:
             _ = await AggPerson.aobjects.comment(comment).update_one(name="something")
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert "hint" not in query_op[CMD_QUERY_KEY]
             assert query_op[CMD_QUERY_KEY]["comment"] == comment
             assert "collation" not in query_op[CMD_QUERY_KEY]
 
         async with async_db_ops_tracker() as q:
             _ = await AggPerson.aobjects.hint(index_name).update_one(name="something")
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert query_op[CMD_QUERY_KEY]["hint"] == {"$hint": index_name}
             assert "comment" not in query_op[CMD_QUERY_KEY]
             assert "collation" not in query_op[CMD_QUERY_KEY]
 
         async with async_db_ops_tracker() as q:
             _ = await AggPerson.aobjects.collation(base).update_one(name="something")
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert "hint" not in query_op[CMD_QUERY_KEY]
             assert "comment" not in query_op[CMD_QUERY_KEY]
@@ -1943,23 +1944,23 @@ class AggPerson(Document):
 
         async with async_db_ops_tracker() as q:
             _ = await AggPerson.aobjects().comment(comment).delete()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert "hint" not in query_op[CMD_QUERY_KEY]
             assert query_op[CMD_QUERY_KEY]["comment"] == comment
             assert "collation" not in query_op[CMD_QUERY_KEY]
 
         async with async_db_ops_tracker() as q:
             _ = await AggPerson.aobjects.hint(index_name).delete()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert query_op[CMD_QUERY_KEY]["hint"] == {"$hint": index_name}
             assert "comment" not in query_op[CMD_QUERY_KEY]
             assert "collation" not in query_op[CMD_QUERY_KEY]
 
         async with async_db_ops_tracker() as q:
             _ = await AggPerson.aobjects.collation(base).delete()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert "hint" not in query_op[CMD_QUERY_KEY]
             assert "comment" not in query_op[CMD_QUERY_KEY]
@@ -2867,9 +2868,9 @@ async def test_db_alias_tests(self):
         """DB Alias tests."""
         # mongoenginetest - Is default connection alias from setUp()
         # Register Aliases
-        await async_register_connection("testdb-1", "mongoenginetest2")
-        await async_register_connection("testdb-2", "mongoenginetest3")
-        await async_register_connection("testdb-3", "mongoenginetest4")
+        await async_register_connection("testdb-1", f"{MONGO_TEST_DB}_2")
+        await async_register_connection("testdb-2", f"{MONGO_TEST_DB}_3")
+        await async_register_connection("testdb-3", f"{MONGO_TEST_DB}_4")
 
         class User(Document):
             name = StringField()
@@ -2931,7 +2932,7 @@ class AuthorBooks(Document):
     async def test_db_alias_overrides(self):
         """Test db_alias can be overriden."""
         # Register a connection with db_alias testdb-2
-        await async_register_connection("testdb-2", "mongoenginetest2")
+        await async_register_connection("testdb-2", f"{MONGO_TEST_DB}_2")
 
         class A(Document):
             """Uses default db_alias"""
@@ -2947,13 +2948,13 @@ class B(A):
         A.aobjects.all()
 
         assert "testdb-2" == B._meta.get("db_alias")
-        assert "mongoenginetest" == (await A._aget_collection()).database.name
-        assert "mongoenginetest2" == (await B._aget_collection()).database.name
+        assert MONGO_TEST_DB == (await A._aget_collection()).database.name
+        assert f"{MONGO_TEST_DB}_2" == (await B._aget_collection()).database.name
         await async_disconnect("testdb-2")
 
     async def test_db_alias_propagates(self):
         """db_alias propagates?"""
-        await async_register_connection("testdb-1", "mongoenginetest2")
+        await async_register_connection("testdb-1", f"{MONGO_TEST_DB}_2")
 
         class A(Document):
             name = StringField()
@@ -3049,7 +3050,7 @@ def __str__(self):
         assert [str(b) async for b in custom_qs] == ["1", "2"]
 
     async def test_switch_db_instance(self):
-        await async_register_connection("testdb-1", "mongoenginetest2")
+        await async_register_connection("testdb-1", f"{MONGO_TEST_DB}_2")
 
         class Group(Document):
             name = StringField()
@@ -3098,8 +3099,8 @@ class Group(Document):
         assert "hello - default" == group.name
 
     async def test_switch_db_multiple_documents_same_context(self):
-        await async_register_connection("testdb-1", "mongoenginetest2")
-        await async_register_connection("testdb-2", "mongoenginetest3")
+        await async_register_connection("testdb-1", f"{MONGO_TEST_DB}_2")
+        await async_register_connection("testdb-2", f"{MONGO_TEST_DB}_3")
 
         class Group(Document):
             name = StringField()
@@ -3155,7 +3156,7 @@ class Post(Document):
             assert p2.title == "post-testdb-2"
 
     async def test_switch_db_and_switch_collection_instance(self):
-        await async_register_connection("testdb-1", "mongoenginetest2")
+        await async_register_connection("testdb-1", f"{MONGO_TEST_DB}_2")
 
         class Group(Document):
             name = StringField()
@@ -3218,8 +3219,8 @@ class Group(Document):
         assert "hello - default" == g0.name
 
     async def test_switch_multiple_db_and_multiple_collection_same_time(self):
-        await async_register_connection("tenantA", "mongoenginetest2")
-        await async_register_connection("tenantB", "mongoenginetest2")
+        await async_register_connection("tenantA", f"{MONGO_TEST_DB}_2")
+        await async_register_connection("tenantB", f"{MONGO_TEST_DB}_2")
 
         class User(Document):
             name = StringField()

tests/asynchronous/document/test_timeseries_collection.py

@@ -1,6 +1,7 @@
 import asyncio
 import unittest
 from datetime import datetime, timedelta
+from tests.utils import MONGO_TEST_DB
 
 try:
     # Python 3.11+
@@ -23,7 +24,7 @@
 
 class TestTimeSeriesCollections(unittest.IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        await async_connect(db="mongoenginetest")
+        await async_connect(db=MONGO_TEST_DB)
         self.db = await async_get_db()
 
         class SensorData(Document):

tests/asynchronous/fields/test_datetime_field.py

@@ -6,6 +6,7 @@
 from mongoengine import *
 from mongoengine.asynchronous import async_connect, connection
 from tests.asynchronous.utils import MongoDBAsyncTestCase, async_get_as_pymongo
+from tests.utils import MONGO_TEST_DB
 
 try:
     import dateutil
@@ -234,7 +235,7 @@ async def test_datetime_tz_aware_mark_as_changed(self):
         connection._connections = {}
         connection._dbs = {}
 
-        await async_connect(db="mongoenginetest", tz_aware=True)
+        await async_connect(db=MONGO_TEST_DB, tz_aware=True)
 
         class LogEntry(Document):
             time = DateTimeField()

tests/asynchronous/queryset/test_field_list.py

@@ -6,11 +6,12 @@
 from mongoengine.asynchronous import async_connect, async_disconnect
 from mongoengine.registry import _CollectionRegistry
 from tests.asynchronous.utils import reset_async_connections
+from tests.utils import MONGO_TEST_DB
 
 
 class TestOnlyExcludeAll(unittest.IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        await async_connect(db="mongoenginetest")
+        await async_connect(db=MONGO_TEST_DB)
 
         class Person(Document):
             name = StringField()

tests/asynchronous/queryset/test_modify.py

@@ -9,7 +9,7 @@
 from mongoengine.asynchronous import async_connect, async_disconnect
 from mongoengine.registry import _CollectionRegistry
 from tests.asynchronous.utils import reset_async_connections
-
+from tests.utils import MONGO_TEST_DB
 
 class Doc(Document):
     id = IntField(primary_key=True)
@@ -18,7 +18,7 @@ class Doc(Document):
 
 class TestOnlyExcludeAll(unittest.IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        await async_connect(db="mongoenginetest")
+        await async_connect(db=MONGO_TEST_DB)
         await Doc.adrop_collection()
 
     async def asyncTearDown(self):

tests/asynchronous/queryset/test_queryset.py

@@ -28,6 +28,7 @@
     async_get_as_pymongo,
     reset_async_connections,
 )
+from tests.utils import MONGO_TEST_DB
 
 
 def get_key_compat(mongo_ver):
@@ -38,8 +39,8 @@ def get_key_compat(mongo_ver):
 
 class TestQueryset(unittest.IsolatedAsyncioTestCase):
     async def asyncSetUp(self):
-        await async_connect(db="mongoenginetest")
-        await async_connect(db="mongoenginetest2", alias="test2")
+        await async_connect(db=MONGO_TEST_DB)
+        await async_connect(db=f"{MONGO_TEST_DB}_2", alias="test2")
 
         class PersonMeta(EmbeddedDocument):
             weight = IntField()
@@ -3515,7 +3516,7 @@ class News(Document):
             assert await qs1.to_list() == await qs2.to_list()
 
     async def test_distinct_handles_references_to_alias(self):
-        await async_register_connection("testdb", "mongoenginetest2")
+        await async_register_connection("testdb", f"{MONGO_TEST_DB}_2")
 
         class Bar(Document):
             text = StringField()

tests/asynchronous/queryset/test_queryset_aggregation.py

@@ -6,6 +6,7 @@
     async_get_mongodb_version
 )
 from tests.asynchronous.utils import async_db_ops_tracker, MongoDBAsyncTestCase
+from tests.utils import MONGO_TEST_DB
 
 
 class TestQuerysetAggregate(MongoDBAsyncTestCase):
@@ -107,23 +108,23 @@ class AggPerson(Document):
 
         async with async_db_ops_tracker() as q:
             _ = await (await AggPerson.aobjects.comment(comment).aggregate(pipeline)).to_list()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert "hint" not in query_op[CMD_QUERY_KEY]
             assert query_op[CMD_QUERY_KEY]["comment"] == comment
             assert "collation" not in query_op[CMD_QUERY_KEY]
 
         async with async_db_ops_tracker() as q:
             _ = await (await AggPerson.aobjects.hint(index_name).aggregate(pipeline)).to_list()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert query_op[CMD_QUERY_KEY]["hint"] == "name_1"
             assert "comment" not in query_op[CMD_QUERY_KEY]
             assert "collation" not in query_op[CMD_QUERY_KEY]
 
         async with async_db_ops_tracker() as q:
             _ = await (await AggPerson.aobjects.collation(base).aggregate(pipeline)).to_list()
-            query_op = (await ((await q.db).system.profile.find({"ns": "mongoenginetest.agg_person"})).to_list())[0]
+            query_op = (await ((await q.db).system.profile.find({"ns": f"{MONGO_TEST_DB}.agg_person"})).to_list())[0]
             CMD_QUERY_KEY = "command"
             assert "hint" not in query_op[CMD_QUERY_KEY]
             assert "comment" not in query_op[CMD_QUERY_KEY]