wait for resource warning

Author: sleepyStick

test/asynchronous/test_monitor.py

@@ -58,8 +58,8 @@ async def create_client(self):
         return client
 
     async def test_cleanup_executors_on_client_del(self):
-        with warnings.catch_warnings():
-            warnings.simplefilter("ignore")
+        with warnings.catch_warnings(record=True) as w:
+            warnings.simplefilter("always")
             client = await self.create_client()
             executors = get_executors(client)
             self.assertEqual(len(executors), 4)
@@ -75,6 +75,14 @@ async def test_cleanup_executors_on_client_del(self):
                     partial(unregistered, ref), f"unregister executor: {name}", timeout=5
                 )
 
+            def resource_warning_caught():
+                for warning in w:
+                    if isinstance(warning.message, ResourceWarning):
+                        return True
+                return False
+
+            await async_wait_until(resource_warning_caught, "catch resource warning")
+
     async def test_cleanup_executors_on_client_close(self):
         client = await self.create_client()
         executors = get_executors(client)

test/test_monitor.py

@@ -58,8 +58,8 @@ def create_client(self):
         return client
 
     def test_cleanup_executors_on_client_del(self):
-        with warnings.catch_warnings():
-            warnings.simplefilter("ignore")
+        with warnings.catch_warnings(record=True) as w:
+            warnings.simplefilter("always")
             client = self.create_client()
             executors = get_executors(client)
             self.assertEqual(len(executors), 4)
@@ -73,6 +73,14 @@ def test_cleanup_executors_on_client_del(self):
             for ref, name in executor_refs:
                 wait_until(partial(unregistered, ref), f"unregister executor: {name}", timeout=5)
 
+            def resource_warning_caught():
+                for warning in w:
+                    if isinstance(warning.message, ResourceWarning):
+                        return True
+                return False
+
+            wait_until(resource_warning_caught, "catch resource warning")
+
     def test_cleanup_executors_on_client_close(self):
         client = self.create_client()
         executors = get_executors(client)