fix(tests): Fix DB names to support tox parallel runner

Author: abhinand-c

mongoengine/fields.py

@@ -24,6 +24,14 @@
 else:
     import dateutil.parser
 
+try:
+    # Python 3.11+
+    from datetime import UTC
+except ImportError:
+    # Python ≤ 3.10
+    from datetime import timezone
+    UTC = timezone.utc
+
 from mongoengine.base import (
     BaseDocument,
     BaseField,
@@ -549,15 +557,15 @@ def to_mongo(self, value):
         if isinstance(value, datetime.datetime):
             # If naive: assume UTC
             if value.tzinfo is None:
-                value = value.replace(tzinfo=datetime.UTC)
+                value = value.replace(tzinfo=UTC)
             else:
                 # Normalize to UTC
-                value = value.astimezone(datetime.UTC)
+                value = value.astimezone(UTC)
             return value
 
         # A date without time
         if isinstance(value, datetime.date):
-            value = datetime.datetime(value.year, value.month, value.day, tzinfo=datetime.UTC)
+            value = datetime.datetime(value.year, value.month, value.day, tzinfo=UTC)
             return value
 
         # Strings
@@ -567,9 +575,9 @@ def to_mongo(self, value):
                 return None
             # Force to UTC
             if parsed.tzinfo is None:
-                parsed = parsed.replace(tzinfo=datetime.UTC)
+                parsed = parsed.replace(tzinfo=UTC)
             else:
-                parsed = parsed.astimezone(datetime.UTC)
+                parsed = parsed.astimezone(UTC)
             return parsed
 
         return None
@@ -673,9 +681,9 @@ def _convert_from_datetime(self, val):
         '2011,06,08,20,26,24,092284'
         """
         if val.tzinfo is None:
-            val = val.replace(tzinfo=datetime.UTC)
+            val = val.replace(tzinfo=UTC)
         else:
-            val = val.astimezone(datetime.UTC)
+            val = val.astimezone(UTC)
         return val.strftime(self.format)
 
     def _convert_from_string(self, data):
@@ -689,7 +697,7 @@ def _convert_from_string(self, data):
         datetime.datetime(2011, 6, 8, 20, 26, 24, 92284)
         """
         values = [int(d) for d in data.split(self.separator)]
-        return datetime.datetime(*values, tzinfo=datetime.UTC)
+        return datetime.datetime(*values, tzinfo=UTC)
 
     def __get__(self, instance, owner):
         if instance is None:

tests/asynchronous/fields/test_complex_datetime_field.py

@@ -8,6 +8,14 @@
 from mongoengine import *
 from tests.asynchronous.utils import MongoDBAsyncTestCase
 
+try:
+    # Python 3.11+
+    from datetime import UTC
+except ImportError:
+    # Python ≤ 3.10
+    from datetime import timezone
+    UTC = timezone.utc
+
 
 class ComplexDateTimeFieldTest(MongoDBAsyncTestCase):
     async def test_complexdatetime_storage(self):
@@ -23,7 +31,7 @@ class LogEntry(Document):
 
         # Post UTC - microseconds are rounded (down) nearest millisecond and
         # dropped - with default datetime fields
-        d1 = datetime.datetime(1970, 1, 1, 0, 0, 1, 999,tzinfo=datetime.UTC)
+        d1 = datetime.datetime(1970, 1, 1, 0, 0, 1, 999,tzinfo=UTC)
         log = LogEntry()
         log.date = d1
         await log.asave()
@@ -32,15 +40,15 @@ class LogEntry(Document):
 
         # Post UTC - microseconds are rounded (down) nearest millisecond - with
         # default datetime fields
-        d1 = datetime.datetime(1970, 1, 1, 0, 0, 1, 9999,tzinfo=datetime.UTC)
+        d1 = datetime.datetime(1970, 1, 1, 0, 0, 1, 9999,tzinfo=UTC)
         log.date = d1
         await log.asave()
         await log.areload()
         assert log.date == d1
 
         # Pre UTC dates microseconds below 1000 are dropped - with default
         # datetime fields
-        d1 = datetime.datetime(1969, 12, 31, 23, 59, 59, 999,tzinfo=datetime.UTC)
+        d1 = datetime.datetime(1969, 12, 31, 23, 59, 59, 999,tzinfo=UTC)
         log.date = d1
         await log.asave()
         await log.areload()
@@ -50,7 +58,7 @@ class LogEntry(Document):
         # log.date has an invalid microsecond value, so I can't construct
         # a date to compare.
         for i in range(1001, 3113, 33):
-            d1 = datetime.datetime(1969, 12, 31, 23, 59, 59, i,tzinfo=datetime.UTC)
+            d1 = datetime.datetime(1969, 12, 31, 23, 59, 59, i,tzinfo=UTC)
             log = LogEntry(
                 date=d1
             )
@@ -164,7 +172,7 @@ class Log(Document):
         assert fetched_log.timestamp is None
 
     async def test_default_static_value(self):
-        NOW = datetime.datetime.now(datetime.UTC)
+        NOW = datetime.datetime.now(UTC)
 
         class Log(Document):
             timestamp = ComplexDateTimeField(default=NOW)
@@ -179,7 +187,7 @@ class Log(Document):
         assert fetched_log.timestamp == NOW
 
     async def test_default_callable(self):
-        NOW = datetime.datetime.now(datetime.UTC)
+        NOW = datetime.datetime.now(UTC)
 
         class Log(Document):
             timestamp = ComplexDateTimeField(default=NOW)

tests/asynchronous/fields/test_datetime_field.py

@@ -13,6 +13,14 @@
 except ImportError:
     dateutil = None
 
+try:
+    # Python 3.11+
+    from datetime import UTC
+except ImportError:
+    # Python ≤ 3.10
+    from datetime import timezone
+    UTC = timezone.utc
+
 
 class TestDateTimeField(MongoDBAsyncTestCase):
     async def test_datetime_from_empty_string(self):
@@ -47,9 +55,9 @@ async def test_default_value_utcnow(self):
         """
 
         class Person(Document):
-            created = DateTimeField(default=dt.datetime.now(datetime.UTC))
+            created = DateTimeField(default=dt.datetime.now(UTC))
 
-        utcnow = dt.datetime.now(datetime.UTC)
+        utcnow = dt.datetime.now(UTC)
         person = Person()
         person.validate()
         person_created_t0 = person.created

tests/asynchronous/fields/test_fields.py

@@ -35,6 +35,14 @@
 from mongoengine.errors import DeprecatedError
 from tests.asynchronous.utils import MongoDBAsyncTestCase
 
+try:
+    # Python 3.11+
+    from datetime import UTC
+except ImportError:
+    # Python ≤ 3.10
+    from datetime import timezone
+    UTC = timezone.utc
+
 
 class TestField(MongoDBAsyncTestCase):
     async def test_constructor_set_historical_behavior_is_kept(self):
@@ -200,7 +208,7 @@ class Person(Document):
             name = StringField()
             age = IntField(default=30, required=False)
             userid = StringField(default=lambda: "test", required=True)
-            created = DateTimeField(default=datetime.datetime.now(datetime.UTC))
+            created = DateTimeField(default=datetime.datetime.now(UTC))
 
         person = Person()
         person.name = None
@@ -369,7 +377,7 @@ class HandleNoneFields(Document):
         doc.str_fld = "spam ham egg"
         doc.int_fld = 42
         doc.flt_fld = 4.2
-        doc.comp_dt_fld = datetime.datetime.now(datetime.UTC)
+        doc.comp_dt_fld = datetime.datetime.now(UTC)
         await doc.asave()
 
         # Unset all the fields

tests/asynchronous/fields/test_file_field.py

@@ -20,6 +20,8 @@
     HAS_PIL = False
 
 from tests.asynchronous.utils import MongoDBAsyncTestCase
+from tests.utils import MONGO_TEST_DB
+
 
 require_pil = pytest.mark.skipif(not HAS_PIL, reason="PIL not installed")
 
@@ -514,7 +516,7 @@ class TestImage(Document):
         await t.image.adelete()
 
     async def test_file_multidb(self):
-        await async_register_connection("test_files", "test_files")
+        await async_register_connection("test_files", f"{MONGO_TEST_DB}_test_files")
 
         class TestFile(Document):
             name = StringField()

tests/asynchronous/queryset/test_queryset.py

@@ -30,6 +30,13 @@
 )
 from tests.utils import MONGO_TEST_DB
 
+try:
+    # Python 3.11+
+    from datetime import UTC
+except ImportError:
+    # Python ≤ 3.10
+    from datetime import timezone
+    UTC = timezone.utc
 
 def get_key_compat(mongo_ver):
     ORDER_BY_KEY = "sort"
@@ -1455,7 +1462,7 @@ class BlogPost(Document):
 
         await BlogPost.adrop_collection()
         await BlogPost.aobjects.create(
-            title="whatever", published_date=datetime.datetime.now(datetime.UTC)
+            title="whatever", published_date=datetime.datetime.now(UTC)
         )
 
         async with async_db_ops_tracker() as q:

tests/asynchronous/test_connection.py

@@ -81,8 +81,8 @@ class History2(Document):
             name = StringField()
             meta = {"db_alias": "db2"}
 
-        await async_connect("db1", alias="db1")
-        await async_connect("db2", alias="db2")
+        await async_connect(f"{MONGO_TEST_DB}_db1", alias="db1")
+        await async_connect(f"{MONGO_TEST_DB}_db2", alias="db2")
 
         await History1.adrop_collection()
         await History2.adrop_collection()
@@ -104,8 +104,8 @@ class History2(Document):
         with pytest.raises(ConnectionFailure):
             await History2.aobjects().as_pymongo().to_list()
 
-        await async_connect("db1", alias="db1")
-        await async_connect("db2", alias="db2")
+        await async_connect(f"{MONGO_TEST_DB}_db1", alias="db1")
+        await async_connect(f"{MONGO_TEST_DB}_db2", alias="db2")
 
         assert await History1.aobjects().as_pymongo().to_list() == [
             {"_id": h.id, "name": "default"}
@@ -125,9 +125,9 @@ class History2(Document):
             name = StringField()
             meta = {"db_alias": "db2"}
 
-        await async_connect()
-        await async_connect("db1", alias="db1")
-        await async_connect("db2", alias="db2")
+        await async_connect(MONGO_TEST_DB)
+        await async_connect(f"{MONGO_TEST_DB}_db1", alias="db1")
+        await async_connect(f"{MONGO_TEST_DB}_db2", alias="db2")
 
         await History.adrop_collection()
         await History1.adrop_collection()
@@ -137,9 +137,9 @@ class History2(Document):
         h1 = await History1(name="db1").asave()
         h2 = await History2(name="db2").asave()
 
-        assert (await History._aget_collection()).database.name == DEFAULT_DATABASE_NAME
-        assert (await History1._aget_collection()).database.name == "db1"
-        assert (await History2._aget_collection()).database.name == "db2"
+        assert (await History._aget_collection()).database.name == MONGO_TEST_DB
+        assert (await History1._aget_collection()).database.name == f"{MONGO_TEST_DB}_db1"
+        assert (await History2._aget_collection()).database.name == f"{MONGO_TEST_DB}_db2"
 
         assert await History.aobjects().as_pymongo().to_list() == [
             {"_id": h.id, "name": "default"}
@@ -276,8 +276,8 @@ class History(Document):
     @pytest.mark.asyncio
     async def test_async_connect_disconnect_works_on_same_document(self):
         """Ensure that the async_connect/async_disconnect works properly with a single Document"""
-        db1 = "db1"
-        db2 = "db2"
+        db1 = f"{MONGO_TEST_DB}_db1"
+        db2 = f"{MONGO_TEST_DB}_db2"
 
         # Ensure freshness of the 2 databases through pymongo
         client = AsyncMongoClient("localhost", 27017)

tests/asynchronous/test_context_managers.py

@@ -424,8 +424,8 @@ class A(Document):
         assert await A.aobjects.count() == 0
 
     async def test_transaction_updates_across_databases(self):
-        await async_connect()
-        await async_connect("test2", "test2")
+        await async_connect(MONGO_TEST_DB)
+        await async_connect(f"{MONGO_TEST_DB}_2", "test2")
 
         class A(Document):
             name = StringField()
@@ -449,8 +449,8 @@ class B(Document):
 
     @requires_mongodb_gte_44
     async def test_collection_creation_via_upserts_across_databases_in_transaction(self):
-        await async_connect()
-        await async_connect("test2", "test2")
+        await async_connect(MONGO_TEST_DB)
+        await async_connect(f"{MONGO_TEST_DB}_test2", "test2")
 
         class A(Document):
             name = StringField()
@@ -482,8 +482,8 @@ class B(Document):
     async def test_an_exception_raised_in_transactions_across_databases_rolls_back_updates(
             self,
     ):
-        await async_connect()
-        await async_connect("test2", "test2")
+        await async_connect(MONGO_TEST_DB)
+        await async_connect(f"{MONGO_TEST_DB}_2", "test2")
 
         class A(Document):
             name = StringField()

tests/asynchronous/test_dereference.py

@@ -15,7 +15,7 @@ async def asyncSetUp(self):
         self.db = await async_connect(db=MONGO_TEST_DB)
 
     async def asyncTearDown(self):
-        await self.db.drop_database()
+        await self.db.drop_database(MONGO_TEST_DB)
         await async_disconnect_all()
         await reset_async_connections()