Add port to connection parameters

Author: vshepard

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
 

testgres/operations/remote_ops.py

@@ -45,10 +45,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)