From: Tommi Virtanen <tommi.virtanen@dreamhost.com>
Date: Mon, 5 Dec 2011 17:55:02 +0000 (-0800)
Subject: Fix unit tests for SSH keep-alive setting.
X-Git-Tag: 1.1.0~2713
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=0dd4d69ffee854ad39f5371fa05c2e19d42d2f24;p=teuthology.git

Fix unit tests for SSH keep-alive setting.

Commit 6e3e0d7cdcb5ba70f938f0850a8828aca2753ab5 failed to pass
unit tests.
---

diff --git a/teuthology/orchestra/test/test_connection.py b/teuthology/orchestra/test/test_connection.py
index 66e0dcaec..371722bd9 100644
--- a/teuthology/orchestra/test/test_connection.py
+++ b/teuthology/orchestra/test/test_connection.py
@@ -36,6 +36,9 @@ def test_connect():
         username='jdoe',
         timeout=60,
         )
+    transport = ssh.expects('get_transport').with_args().returns_fake()
+    transport.remember_order()
+    transport.expects('set_keepalive').with_args(False)
     got = connection.connect(
         'jdoe@orchestra.test.newdream.net.invalid',
         _SSHClient=sshclient,
@@ -60,6 +63,9 @@ def test_connect_override_hostkeys():
         username='jdoe',
         timeout=60,
         )
+    transport = ssh.expects('get_transport').with_args().returns_fake()
+    transport.remember_order()
+    transport.expects('set_keepalive').with_args(False)
     create_key = fudge.Fake('create_key')
     create_key.expects_call().with_args('ssh-rsa', 'testkey').returns('frobnitz')
     got = connection.connect(