refactor: use class attribute for SSH and Telnet client instantiation (#1068)

Author: NtAlexio2

nettacker/core/lib/ssh.py

@@ -9,13 +9,15 @@
 
 
 class SshLibrary(BaseLibrary):
+    client = SSHClient
+
     def brute_force(self, *args, **kwargs):
         host = kwargs["host"]
         port = kwargs["port"]
         username = kwargs["username"]
         password = kwargs["password"]
 
-        connection = SSHClient()
+        connection = self.client()
         connection.set_missing_host_key_policy(AutoAddPolicy())
         connection.connect(
             **{

nettacker/core/lib/telnet.py

@@ -7,7 +7,7 @@ class TelnetLibrary(BaseLibrary):
     client = telnetlib.Telnet
 
     def brute_force(self, host, port, username, password, timeout):
-        connection = telnetlib.Telnet(host, port, timeout)
+        connection = self.client(host, port, timeout)
         connection.read_until(b"login: ")
         connection.write(username.encode("utf-8") + b"\n")
         connection.read_until(b"Password: ")