summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--framework/ssh_connection.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/framework/ssh_connection.py b/framework/ssh_connection.py
index 3ee9665..63b8785 100644
--- a/framework/ssh_connection.py
+++ b/framework/ssh_connection.py
@@ -68,6 +68,12 @@ class SSHConnection(object):
def close(self):
self.session.close()
+ connection = {}
+ connection[self.name] = self.session
+ try:
+ CONNECTIONS.remove(connection)
+ except:
+ pass
def isalive(self):
return self.session.isalive()