test mongodb cache index

pallets-eco · May 31, 2023 · 8bf41a8 · 8bf41a8
1 parent 2564a19
commit 8bf41a8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/cachelib/mongodb.py b/src/cachelib/mongodb.py
@@ -41,8 +41,13 @@ def __init__(
 
         if client is None or isinstance(client, str):
             client = pymongo.MongoClient(host=client)
-        client.ensure_index([("id", pymongo.ASCENDING)])
         self.client = client[db][collection]
+        index_info = self.client.index_information()
+        all_keys = {
+            subkey[0] for value in index_info.values() for subkey in value["key"]
+        }
+        if "id" not in all_keys:
+            self.client.create_index("id", unique=True)
         self.key_prefix = key_prefix or ""
         self.collection = collection
 

diff --git a/tests/test_mongodb_cache.py b/tests/test_mongodb_cache.py
@@ -14,6 +14,11 @@ def _factory(self, *args, **kwargs):
         kwargs["key_prefix"] = "prefix"
 
         rc = request.param(*args, **kwargs)
+        index_info = rc.client.index_information()
+        all_keys = {
+            subkey[0] for value in index_info.values() for subkey in value["key"]
+        }
+        assert "id" in all_keys, "Failed to create index on 'id' field"
         rc.clear()
         return rc