Tests: HTTP settings.

This commit is contained in:
Andrey Zelenkov
2018-07-13 20:15:50 +03:00
parent 92a7d9f0c6
commit cfe91315eb
2 changed files with 124 additions and 3 deletions

View File

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