roles: use correct ssh host and port

SSSD · Aug 9, 2024 · 5afdc33 · 5afdc33
1 parent 5668acf
commit 5afdc33
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/sssd_test_framework/roles/base.py b/sssd_test_framework/roles/base.py
@@ -123,8 +123,16 @@ def ssh(self, user: str, password: str, *, shell: Shell | None = None) -> SSHCli
         if shell is None:
             shell = Bash()
 
+        host = self.host.hostname
+        port = 22
+
+        if isinstance(self.host.conn, SSHClient):
+            host = getattr(self.host.conn, "host", host)
+            port = getattr(self.host.conn, "port", 22)
+
         return SSHClient(
-            self.host.hostname,
+            host=host,
+            port=port,
             user=user,
             password=password,
             shell=shell,