From: Matus Fabian <matfabia@cisco.com>
Date: Thu, 17 Mar 2016 13:14:22 +0000 (+0100)
Subject: Add SSH disconnect
X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=commitdiff_plain;h=2bd0f8299dffb7818d87081f2f7b06d38a8ab620

Add SSH disconnect

Change-Id: Ib0efec083aa9ee33b7d15680e5a15989827b8e57
Signed-off-by: Matus Fabian <matfabia@cisco.com>
---

diff --git a/resources/libraries/python/ssh.py b/resources/libraries/python/ssh.py
index ad5fb27b85..a94eec4e91 100644
--- a/resources/libraries/python/ssh.py
+++ b/resources/libraries/python/ssh.py
@@ -67,6 +67,17 @@ class SSH(object):
                 format(self._ssh.get_transport().getpeername()))
         logger.debug('Connections: {0}'.format(str(SSH.__existing_connections)))
 
+    def disconnect(self, node):
+        """Close SSH connection to the node.
+
+        :param node: The node to disconnect from.
+        :type node: dict
+        """
+        node_hash = self._node_hash(node)
+        if node_hash in SSH.__existing_connections:
+            ssh = SSH.__existing_connections.pop(node_hash)
+            ssh.close()
+
     def exec_command(self, cmd, timeout=10):
         """Execute SSH command on a new channel on the connected Node.