fix: count_document issue

openedx · Apr 26, 2024 · 218b310 · 218b310
1 parent 591010d
commit 218b310
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -14,7 +14,7 @@ jobs:
         os: [ubuntu-20.04]
         python-version:
         - "3.8"
-        - '3.11'
+        # - '3.11'
         toxenv: [celery53-django42]
 
     steps:

diff --git a/eventtracking/backends/tests/test_mongodb_integration.py b/eventtracking/backends/tests/test_mongodb_integration.py
@@ -55,8 +55,9 @@ def test_sequential_events(self):
             mem_events[event['data']['sequence']] = event
         self.assertEqual(len(mem_events), 10)
 
-        cursor = self.mongo_backend.collection.find()
-        self.assertEqual(cursor.count_documents({}), 10)
+        collection = self.mongo_backend.collection
+        cursor = collection.find()
+        self.assertEqual(collection.count_documents({}), 10)
 
         for event in cursor:
             mem_event = mem_events[event['data']['sequence']]