From 4460c3476d9099d0e05db8a763053c973d2d8be8 Mon Sep 17 00:00:00 2001 From: Richard Oudkerk Date: Mon, 30 Apr 2012 14:48:50 +0100 Subject: [PATCH] Minor fix for multiprocessing unit test Read from socket might have returned partial message. --- Lib/test/test_multiprocessing.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/Lib/test/test_multiprocessing.py b/Lib/test/test_multiprocessing.py index 799be702fcd..089b76fd7a2 100644 --- 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)