Handle socket connection error in SSH client

Catch socket error raised when connecting to the server over SSH
and raise a GerritError.

Change-Id: I140a26be4bc04ec229d95faaa80b94477cf96645

1 files changed, 8 insertions, 4 deletions

diff --git a/pygerrit/ssh.py b/pygerrit/ssh.py
index 2a019f5..270a63f 100644
--- a/pygerrit/ssh.py
+++ b/pygerrit/ssh.py
@@ -26,6 +26,7 @@ THE SOFTWARE.
 
 from os.path import abspath, expanduser, isfile
 import re
+import socket
 
 from pygerrit.error import GerritError
 
@@ -84,10 +85,13 @@ class GerritSSHClient(SSHClient):
             port = int(data['port'])
         except ValueError:
             raise GerritError("Invalid port: %s" % data['port'])
-        self.connect(hostname=data['hostname'],
-                     port=port,
-                     username=data['user'],
-                     key_filename=key_filename)
+        try:
+            self.connect(hostname=data['hostname'],
+                         port=port,
+                         username=data['user'],
+                         key_filename=key_filename)
+        except socket.error as e:
+            raise GerritError("Failed to connect to server: %s" % e)
 
     def get_remote_version(self):
         """ Return the version of the remote Gerrit server. """