From d9f8479375fb6231db26e53a6549f2b96108faee Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Thu, 26 Apr 2018 17:37:24 +0300 Subject: Tests: response handling improved. --- test/unit.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'test/unit.py') diff --git a/test/unit.py b/test/unit.py index 90906ba8..e1bf2374 100644 --- a/test/unit.py +++ b/test/unit.py @@ -259,17 +259,21 @@ class TestUnitHTTP(TestUnit): return self.http('PUT', **kwargs) def _recvall(self, sock, buff_size=4096): - data = '' + data = b'' while select.select([sock], [], [], 1)[0]: - part = sock.recv(buff_size).decode() + part = sock.recv(buff_size) data += part - if part is '': + if not len(part): break - return data + return data.decode() def _resp_to_dict(self, resp): m = re.search('(.*?\x0d\x0a?)\x0d\x0a?(.*)', resp, re.M | re.S) + + if not m: + return {} + headers_text, body = m.group(1), m.group(2) p = re.compile('(.*?)\x0d\x0a?', re.M | re.S) -- cgit