Procházet zdrojové kódy

CTS: handle socket exceptions better

Signed-off-by: Angus Salkeld <asalkeld@redhat.com>
Reviewed-by: Steven Dake <sdake@redhat.com>
Angus Salkeld před 14 roky
rodič
revize
df06e98298
1 změnil soubory, kde provedl 5 přidání a 1 odebrání
  1. 5 1
      cts/corosync.py

+ 5 - 1
cts/corosync.py

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