tests(pymongo) support pymongo 4.0 in tests (backport #3037) (#3038)

* tests(pymongo) support pymongo 4.0 in tests (#3037)

(cherry picked from commit b684836)

# Conflicts:
#	riotfile.py

* Apply suggestions from code review

Co-authored-by: Brett Langdon <[email protected]>

Author: mergify[bot]

riotfile.py

@@ -417,8 +417,16 @@ def select_pys(min_version=MIN_PYTHON_VERSION, max_version=MAX_PYTHON_VERSION):
                             ">=3.7,<3.8",
                             ">=3.8,<3.9",
                             ">=3.9,<3.10",
+                        ],
+                    },
+                ),
+                Venv(
+                    pys=select_pys(min_version="3.6"),
+                    pkgs={
+                        "pymongo": [
                             ">=3.10,<3.11",
                             ">=3.12,<3.13",
+                            ">=4.0,<4.1",
                             latest,
                         ],
                     },

tests/contrib/pymongo/test.py

@@ -140,17 +140,22 @@ def test_delete(self):
         songs = db[collection_name]
         songs.insert_many(input_songs)
 
+        def count(col, *args, **kwargs):
+            if pymongo.version_tuple >= (4, 0):
+                return col.count_documents(*args, **kwargs)
+            return col.count(*args, **kwargs)
+
         # test delete one
         af = {"artist": "Neil"}
-        assert songs.count(af) == 2
+        assert count(songs, af) == 2
         songs.delete_one(af)
-        assert songs.count(af) == 1
+        assert count(songs, af) == 1
 
         # test delete many
         af = {"artist": "Leonard"}
-        assert songs.count(af) == 2
+        assert count(songs, af) == 2
         songs.delete_many(af)
-        assert songs.count(af) == 0
+        assert count(songs, af) == 0
 
         # ensure all is traced.
         spans = tracer.pop()
@@ -165,16 +170,28 @@ def test_delete(self):
             assert span.meta.get("out.host")
             assert span.metrics.get("out.port")
 
-        expected_resources = [
-            "drop here.are.songs",
-            "count here.are.songs",
-            "count here.are.songs",
-            "count here.are.songs",
-            "count here.are.songs",
-            'delete here.are.songs {"artist": "?"}',
-            'delete here.are.songs {"artist": "?"}',
-            "insert here.are.songs",
-        ]
+        if pymongo.version_tuple >= (4, 0):
+            expected_resources = [
+                "drop here.are.songs",
+                "aggregate here.are.songs",
+                "aggregate here.are.songs",
+                "aggregate here.are.songs",
+                "aggregate here.are.songs",
+                'delete here.are.songs {"artist": "?"}',
+                'delete here.are.songs {"artist": "?"}',
+                "insert here.are.songs",
+            ]
+        else:
+            expected_resources = [
+                "drop here.are.songs",
+                "count here.are.songs",
+                "count here.are.songs",
+                "count here.are.songs",
+                "count here.are.songs",
+                'delete here.are.songs {"artist": "?"}',
+                'delete here.are.songs {"artist": "?"}',
+                "insert here.are.songs",
+            ]
 
         assert sorted(expected_resources) == sorted(s.resource for s in spans)