From cfe91315eb6a9473db532800d9a18576536c8ee0 Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Fri, 13 Jul 2018 20:15:50 +0300 Subject: Tests: HTTP settings. --- test/unit.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'test/unit.py') diff --git a/test/unit.py b/test/unit.py index f9716366..c8820280 100644 --- a/test/unit.py +++ b/test/unit.py @@ -250,7 +250,7 @@ class TestUnitHTTP(TestUnit): if '--verbose' in sys.argv: print('>>>', req, sep='\n') - resp = self._recvall(sock) + resp = self.recvall(sock) if '--verbose' in sys.argv: print('<<<', resp.encode('utf-8'), sep='\n') @@ -276,11 +276,16 @@ class TestUnitHTTP(TestUnit): def put(self, **kwargs): return self.http('PUT', **kwargs) - def _recvall(self, sock, buff_size=4096): + def recvall(self, sock, buff_size=4096): data = b'' while select.select([sock], [], [], 1)[0]: - part = sock.recv(buff_size) + try: + part = sock.recv(buff_size) + except: + break + data += part + if not len(part): break -- cgit