Simplify TestMonitor.

ajdavis · ajdavis · commit b633af82a91d · 2015-01-12T18:03:15.000-05:00
The executor is now registered immediately, don't need to wait.
diff --git a/test/test_monitor.py b/test/test_monitor.py
@@ -25,14 +25,6 @@
 from test.utils import single_client, one, connected, wait_until
 
 
-def registered(executor):
-    for ref in _EXECUTORS.copy():
-        if ref() is executor:
-            return ref
-
-    return None
-
-
 def unregistered(ref):
     gc.collect()
     return ref not in _EXECUTORS
@@ -45,7 +37,7 @@ def test_atexit_hook(self):
         connected(client)
 
         # The executor stores a weakref to itself in _EXECUTORS.
-        ref = wait_until(partial(registered, executor), 'register executor')
+        ref = one(r for r in _EXECUTORS.copy() if r() is executor)
 
         del executor
         del client
