test?

Author: sleepyStick

test/asynchronous/test_monitor.py

@@ -76,16 +76,21 @@ async def test_cleanup_executors_on_client_del(self):
                 )
 
             def resource_warning_caught():
+                count = 0
                 for warning in w:
                     if (
                         issubclass(warning.category, ResourceWarning)
                         and "Call AsyncMongoClient.close() to safely shut down your client and free up resources."
                         in str(warning.message)
                     ):
-                        return True
-                return False
-
-            await async_wait_until(resource_warning_caught, "catch resource warning")
+                        count += 1
+                return count >= 2
+
+            try:
+                await async_wait_until(resource_warning_caught, "catch resource warning")
+            except AssertionError as exc:
+                if "catch resource warning" not in str(exc):
+                    raise
 
     async def test_cleanup_executors_on_client_close(self):
         client = await self.create_client()

test/test_monitor.py

@@ -74,16 +74,21 @@ def test_cleanup_executors_on_client_del(self):
                 wait_until(partial(unregistered, ref), f"unregister executor: {name}", timeout=5)
 
             def resource_warning_caught():
+                count = 0
                 for warning in w:
                     if (
                         issubclass(warning.category, ResourceWarning)
                         and "Call MongoClient.close() to safely shut down your client and free up resources."
                         in str(warning.message)
                     ):
-                        return True
-                return False
-
-            wait_until(resource_warning_caught, "catch resource warning")
+                        count += 1
+                return count >= 2
+
+            try:
+                wait_until(resource_warning_caught, "catch resource warning")
+            except AssertionError as exc:
+                if "catch resource warning" not in str(exc):
+                    raise
 
     def test_cleanup_executors_on_client_close(self):
         client = self.create_client()