From df06e98298720207726e9dea8d8722b116693a64 Mon Sep 17 00:00:00 2001
From: Angus Salkeld <asalkeld@redhat.com>
Date: Wed, 25 Jan 2012 10:56:26 +1100
Subject: [PATCH] CTS: handle socket exceptions better

Signed-off-by: Angus Salkeld <asalkeld@redhat.com>
Reviewed-by: Steven Dake <sdake@redhat.com>
---
 cts/corosync.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/cts/corosync.py b/cts/corosync.py
index 89b08088..b8c4edbf 100644
--- a/cts/corosync.py
+++ b/cts/corosync.py
@@ -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