test_iteration regex

Author: NoahStapp

test/asynchronous/test_client.py

@@ -243,12 +243,9 @@ async def test_getattr(self, async_client):
         assert "has no attribute '_does_not_exist'" in str(context.value)
 
     async def test_iteration(self, async_client):
-        if _IS_SYNC or sys.version_info < (3, 10):
-            msg = "'AsyncMongoClient' object is not iterable"
-        else:
-            msg = "'AsyncMongoClient' object is not an async iterator"
+        msg = "'AsyncMongoClient' object is not iterable"
 
-        with pytest.raises(TypeError, match="'AsyncMongoClient' object is not iterable"):
+        with pytest.raises(TypeError, match=msg):
             for _ in async_client:
                 break
 
@@ -261,7 +258,7 @@ async def test_iteration(self, async_client):
             _ = await anext(async_client)
 
         # 'next()' method fails
-        with pytest.raises(TypeError, match="'AsyncMongoClient' object is not iterable"):
+        with pytest.raises(TypeError, match=msg):
             _ = await async_client.anext()
 
         # Do not implement typing.Iterable

test/test_client.py

@@ -240,12 +240,9 @@ def test_getattr(self, client):
         assert "has no attribute '_does_not_exist'" in str(context.value)
 
     def test_iteration(self, client):
-        if _IS_SYNC or sys.version_info < (3, 10):
-            msg = "'MongoClient' object is not iterable"
-        else:
-            msg = "'MongoClient' object is not an async iterator"
+        msg = "'MongoClient' object is not iterable"
 
-        with pytest.raises(TypeError, match="'MongoClient' object is not iterable"):
+        with pytest.raises(TypeError, match=msg):
             for _ in client:
                 break
 
@@ -258,7 +255,7 @@ def test_iteration(self, client):
             _ = next(client)
 
         # 'next()' method fails
-        with pytest.raises(TypeError, match="'MongoClient' object is not iterable"):
+        with pytest.raises(TypeError, match=msg):
             _ = client.next()
 
         # Do not implement typing.Iterable