From 6e3e0d7cdcb5ba70f938f0850a8828aca2753ab5 Mon Sep 17 00:00:00 2001
From: Josh Durgin <josh.durgin@dreamhost.com>
Date: Thu, 3 Nov 2011 13:07:21 -0700
Subject: [PATCH] connection: allow the caller to specify whether keep-alive
 should be used

---
 teuthology/orchestra/connection.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/teuthology/orchestra/connection.py b/teuthology/orchestra/connection.py
index fb9d0bd182..3515b8b3b2 100644
--- a/teuthology/orchestra/connection.py
+++ b/teuthology/orchestra/connection.py
@@ -18,7 +18,8 @@ def create_key(keytype, key):
     else:
         raise ValueError('keytype must be ssh-rsa or ssh-dsa')
 
-def connect(user_at_host, host_key=None, _SSHClient=None, _create_key=None):
+def connect(user_at_host, host_key=None, keep_alive=False,
+            _SSHClient=None, _create_key=None):
     user, host = split_user(user_at_host)
     if _SSHClient is None:
         _SSHClient = paramiko.SSHClient
@@ -41,4 +42,5 @@ def connect(user_at_host, host_key=None, _SSHClient=None, _create_key=None):
         username=user,
         timeout=60,
         )
+    ssh.get_transport().set_keepalive(keep_alive)
     return ssh
-- 
2.39.5