From 218b310ed0d329ea05215ade0f580971d9c88fef Mon Sep 17 00:00:00 2001 From: Muhammad Umar Khan Date: Fri, 26 Apr 2024 16:50:59 +0500 Subject: [PATCH] fix: count_document issue --- .github/workflows/ci.yml | 2 +- eventtracking/backends/tests/test_mongodb_integration.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index a78feea..69fe83b 100644 --- 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 index e0f60ea..c8ab503 100644 --- 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']]