CTS: handle socket exceptions better

Signed-off-by: Angus Salkeld <asalkeld@redhat.com>
Reviewed-by: Steven Dake <sdake@redhat.com>
This commit is contained in:
Angus Salkeld 2012-01-25 10:56:26 +11:00
parent 1331c43075
commit df06e98298

View File

@ -558,7 +558,11 @@ class TestAgent(object):
def read (self):
msg = self.sock.recv (4096)
try:
msg = self.sock.recv (4096)
except socket.error, msg:
raise RuntimeError(msg)
if msg == '':
raise RuntimeError("socket connection broken")
return msg