support socket timeout for connections

add getter and setter for remoteconnection timeout

Signed-off-by: Luke Inman-Semerau <[email protected]>

Author: Sergey Chipiga

py/selenium/webdriver/remote/remote_connection.py

@@ -133,6 +133,21 @@ class RemoteConnection(object):
 
     Communicates with the server using the WebDriver wire protocol:
     https://github.com/SeleniumHQ/selenium/wiki/JsonWireProtocol"""
+
+    _timeout = socket._GLOBAL_DEFAULT_TIMEOUT
+
+    @classmethod
+    def get_timeout(cls):
+        return None if cls._timeout == socket._GLOBAL_DEFAULT_TIMEOUT or cls._timeout
+
+    @classmethod
+    def set_timeout(cls, timeout):
+        cls._timeout = timeout
+
+    @classmethod
+    def reset_timeout(cls):
+        cls._timeout = socket._GLOBAL_DEFAULT_TIMEOUT
+
     def __init__(self, remote_server_addr, keep_alive=False):
         # Attempt to resolve the hostname and get an IP address.
         self.keep_alive = keep_alive
@@ -157,7 +172,9 @@ def __init__(self, remote_server_addr, keep_alive=False):
 
         self._url = remote_server_addr
         if keep_alive:
-            self._conn = httplib.HTTPConnection(str(addr), str(parsed_url.port))
+            self._conn = httplib.HTTPConnection(
+                str(addr), str(parsed_url.port), timeout=self._timeout)
+
         self._commands = {
             Command.STATUS: ('GET', '/status'),
             Command.NEW_SESSION: ('POST', '/session'),
@@ -386,7 +403,7 @@ def _request(self, method, url, body=None):
             try:
                 self._conn.request(method, parsed_url.path, body, headers)
                 resp = self._conn.getresponse()
-            except httplib.HTTPException:
+            except (httplib.HTTPException, socket.error):
                 self._conn.close()
                 raise
 
@@ -422,7 +439,7 @@ def _request(self, method, url, body=None):
             else:
                 opener = url_request.build_opener(url_request.HTTPRedirectHandler(),
                                                   HttpErrorHandler())
-            resp = opener.open(request)
+            resp = opener.open(request, timeout=self._timeout)
             statuscode = resp.code
             if not hasattr(resp, 'getheader'):
                 if hasattr(resp.headers, 'getheader'):