Fix tests on Travis

Author: 1st1

tests/test_tcp.py

@@ -40,12 +40,18 @@ async def test_client(addr):
                 await self.loop.sock_connect(sock, addr)
 
                 await self.loop.sock_sendall(sock, A_DATA)
-                data = await self.loop.sock_recv(sock, 2)
-                self.assertEqual(data, b'OK')
+
+                buf = b''
+                while len(buf) != 2:
+                    buf += await self.loop.sock_recv(sock, 1)
+                self.assertEqual(buf, b'OK')
 
                 await self.loop.sock_sendall(sock, B_DATA)
-                data = await self.loop.sock_recv(sock, 4)
-                self.assertEqual(data, b'SPAM')
+
+                buf = b''
+                while len(buf) != 4:
+                    buf += await self.loop.sock_recv(sock, 1)
+                self.assertEqual(buf, b'SPAM')
 
         async def start_server():
             nonlocal CNT

tests/test_unix.py

@@ -37,12 +37,18 @@ async def test_client(addr):
                 await self.loop.sock_connect(sock, addr)
 
                 await self.loop.sock_sendall(sock, b'AAAA')
-                data = await self.loop.sock_recv(sock, 2)
-                self.assertEqual(data, b'OK')
+
+                buf = b''
+                while len(buf) != 2:
+                    buf += await self.loop.sock_recv(sock, 1)
+                self.assertEqual(buf, b'OK')
 
                 await self.loop.sock_sendall(sock, b'BBBB')
-                data = await self.loop.sock_recv(sock, 4)
-                self.assertEqual(data, b'SPAM')
+
+                buf = b''
+                while len(buf) != 4:
+                    buf += await self.loop.sock_recv(sock, 1)
+                self.assertEqual(buf, b'SPAM')
 
         async def start_server():
             nonlocal CNT