Minor fix for multiprocessing unit test

Read from socket might have returned partial message.

diff --git a/Lib/test/test_multiprocessing.py b/Lib/test/test_multiprocessing.py
index 799be702fcda33076b6ba8afbc21b1a8138bc7c6..089b76fd7a293bafcc6353a48438b93f64a46c30 100644 (file)
--- a/Lib/test/test_multiprocessing.py
+++ b/Lib/test/test_multiprocessing.py
@@ -2034,7 +2034,14 @@ class _TestPicklingConnections(BaseTestCase):
         address = lconn.recv()
         rconn.send((address, msg))
         new_conn = lconn.recv()
-        self.assertEqual(new_conn.recv(100), msg.upper())
+        buf = []
+        while True:
+            s = new_conn.recv(100)
+            if not s:
+                break
+            buf.append(s)
+        buf = b''.join(buf)
+        self.assertEqual(buf, msg.upper())
         new_conn.close()
 
         lconn.send(None)