Edits

Author: WaVEV

django_mongodb_backend/cache.py

@@ -113,7 +113,6 @@ def add(self, key, value, timeout=DEFAULT_TIMEOUT, version=None):
                 True,
             )
         except DuplicateKeyError:
-            # Check the exception name to catch when the key exists.
             return False
         return True
 

django_mongodb_backend/creation.py

@@ -1,7 +1,9 @@
 from django.conf import settings
-from django.core.management import call_command
+from django.core.cache import caches
 from django.db.backends.base.creation import BaseDatabaseCreation
 
+from django_mongodb_backend.cache import MongoDBCache
+
 
 class DatabaseCreation(BaseDatabaseCreation):
     def _execute_create_test_db(self, cursor, parameters, keepdb=False):
@@ -20,5 +22,12 @@ def _destroy_test_db(self, test_database_name, verbosity):
 
     def create_test_db(self, *args, **kwargs):
         test_database_name = super().create_test_db(*args, **kwargs)
-        call_command("createcachecollection", database=self.connection.alias)
+        # Create cache collections
+        for cache_alias in settings.CACHES:
+            cache = caches[cache_alias]
+            connection = cache._db
+            if cache._collection_name in connection.introspection.table_names():
+                continue
+            cache = MongoDBCache(cache._collection_name, {})
+            cache.create_indexes()
         return test_database_name

django_mongodb_backend/management/commands/createcachecollection.py

@@ -50,7 +50,6 @@ def check_collection(self, database, collection_name):
         if not router.allow_migrate_model(database, cache.cache_model_class):
             return
         connection = connections[database]
-
         if collection_name in connection.introspection.table_names():
             if self.verbosity > 0:
                 self.stdout.write("Cache collection '%s' already exists." % collection_name)