Tests: response handling improved.

This commit is contained in:
Andrey Zelenkov
2018-04-26 17:37:24 +03:00
parent a44d358f29
commit d9f8479375
3 changed files with 14 additions and 10 deletions

View File

@@ -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)