Add port to connection parameters

vshepard · vshepard · commit e8f4da9f63ac · 2024-06-06T07:51:43.000+02:00
diff --git a/testgres/operations/os_ops.py b/testgres/operations/os_ops.py
@@ -10,8 +10,9 @@
 
 
 class ConnectionParams:
-    def __init__(self, host='127.0.0.1', ssh_key=None, username=None):
+    def __init__(self, host='127.0.0.1', port=None, ssh_key=None, username=None):
         self.host = host
+        self.port = port
         self.ssh_key = ssh_key
         self.username = username
 
diff --git a/testgres/operations/remote_ops.py b/testgres/operations/remote_ops.py
@@ -47,10 +47,13 @@ def __init__(self, conn_params: ConnectionParams):
         self.conn_params = conn_params
         self.host = conn_params.host
         self.ssh_key = conn_params.ssh_key
+        self.port = conn_params.port
         if self.ssh_key:
             self.ssh_cmd = ["-i", self.ssh_key]
         else:
             self.ssh_cmd = []
+        if self.port:
+            self.ssh_cmd = ["-p", self.port]
         self.remote = True
         self.username = conn_params.username or self.get_user()
         self.add_known_host(self.host)
