kevinheyer/bootstrap-vz
1
0
Fork 0
mirror of https://github.com/kevingruesser/bootstrap-vz.git synced 2025-10-10 09:09:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve read_from_socket, a lot...

Browse source
This commit is contained in:
Anders Ingemann 2015-01-16 01:52:42 +01:00
parent 7ef88d284d
commit 767b32d20e
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/integration/tools/__init__.py
View file

@ -14,7 +14,7 @@ def waituntil(predicate, timeout=5, interval=0.05):
return False
def read_from_socket(socket_path, termination_string, timeout):
def read_from_socket(socket_path, termination_string, timeout, read_timeout=0.5):
import socket
import select
import errno
@ -22,15 +22,15 @@ def read_from_socket(socket_path, termination_string, timeout):
console.connect(socket_path)
console.setblocking(0)
from timeit import default_timer
start = default_timer()
output = ''
ptr = 0
continue_select = True
nooutput_for = 0
select_timeout = .5
while continue_select:
read_ready, _, _ = select.select([console], [], [], select_timeout)
read_ready, _, _ = select.select([console], [], [], read_timeout)
if console in read_ready:
nooutput_for = 0
while True:
try:
output += console.recv(1024)
@ -43,12 +43,11 @@ def read_from_socket(socket_path, termination_string, timeout):
if e.errno != errno.EWOULDBLOCK:
raise Exception(e)
continue_select = False
else:
nooutput_for += select_timeout
if nooutput_for > timeout:
if default_timer() - start > timeout:
from exceptions import SocketReadTimeout
msg = ('Reading from socket `{path}\' timed out after {seconds} seconds.'
.format(path=socket_path, timeout=nooutput_for))
msg = ('Reading from socket `{path}\' timed out after {seconds} seconds.\n'
'Here is the output so far:\n{output}'
.format(path=socket_path, seconds=timeout, output=output))
raise SocketReadTimeout(msg)
console.close()
return output