Issue #19857: Make sure that test_imaplib reaps server threads even in face of

error upon client disconnection.

diff --git a/Lib/test/test_imaplib.py b/Lib/test/test_imaplib.py
index c37ea1dccf10403db96090423a233db464c82da0..2435be5b0f9f3c73b443902e47a824615dda417b 100644 (file)
--- a/Lib/test/test_imaplib.py
+++ b/Lib/test/test_imaplib.py
@@ -210,13 +210,12 @@ class BaseThreadedNetworkedTests(unittest.TestCase):
 
     @contextmanager
     def reaped_pair(self, hdlr):
-        server, thread = self.make_server((support.HOST, 0), hdlr)
-        client = self.imap_class(*server.server_address)
-        try:
-            yield server, client
-        finally:
-            client.logout()
-            self.reap_server(server, thread)
+        with self.reaped_server(hdlr) as server:
+            client = self.imap_class(*server.server_address)
+            try:
+                yield server, client
+            finally:
+                client.logout()
 
     @reap_threads
     def test_connect(self):