orchestra.run: fix generating ConnectionLostError

Previously you could get AttributeError if
a node went away, because it tried to call
getpeername after get_transport() had
returned None.

Signed-off-by: John Spray <john.spray@redhat.com>

diff --git a/teuthology/orchestra/run.py b/teuthology/orchestra/run.py
index 43c9eb8ece702c9b66ef359bfa1f496f53f1dfa9..cccb8d38970b91fadee92f042182ab397858b532 100644 (file)
--- a/teuthology/orchestra/run.py
+++ b/teuthology/orchestra/run.py
@@ -380,7 +380,11 @@ def run(
                     paramiko
     """
     try:
-        (host, port) = client.get_transport().getpeername()
+        transport = client.get_transport()
+        if transport:
+            (host, port) = transport.getpeername()
+        else:
+            raise ConnectionLostError(command=quote(args), node=name)
     except socket.error:
         raise ConnectionLostError(command=quote(args), node=name)