1 files changed, 7 insertions, 0 deletions
diff --git a/tests/torture.c b/tests/torture.c
index 2ba5556c..74640031 100644
--- a/tests/torture.c
+++ b/tests/torture.c
@@ -315,6 +315,7 @@ int torture_isdir(const char *path) {
 }
 
 ssh_session torture_ssh_session(const char *host,
+                                const unsigned int *port,
                                 const char *user,
                                 const char *password) {
     ssh_session session;
@@ -334,6 +335,12 @@ ssh_session torture_ssh_session(const char *host,
         goto failed;
     }
 
+    if (port != NULL) {
+      if (ssh_options_set(session, SSH_OPTIONS_PORT, port) < 0) {
+        goto failed;
+      }
+    }
+
     if (user != NULL) {
         if (ssh_options_set(session, SSH_OPTIONS_USER, user) < 0) {
             goto failed;