Update establish_ssh_tunnel

Author: vshepard

testgres/operations/remote_ops.py

@@ -74,11 +74,11 @@ def is_port_open(host, port):
             except socket.error:
                 return False
 
-    def establish_ssh_tunnel(self, local_port, remote_port):
+    def establish_ssh_tunnel(self, local_port, remote_port, host='localhost'):
         """
         Establish an SSH tunnel from a local port to a remote PostgreSQL port.
         """
-        ssh_cmd = ['-N', '-L', f"{local_port}:localhost:{remote_port}"]
+        ssh_cmd = ['-N', '-L', f"{local_port}:{host}:{remote_port}"]
         self.tunnel_process = self.exec_command(ssh_cmd, get_process=True, timeout=300)
         timeout = 10
         start_time = time.time()
@@ -410,7 +410,7 @@ def db_connect(self, dbname, user, password=None, host="localhost", port=5432):
         """
         local_port = reserve_port()
         self.tunnel_port = local_port
-        self.establish_ssh_tunnel(local_port=local_port, remote_port=port)
+        self.establish_ssh_tunnel(local_port=local_port, remote_port=port, host=host)
         try:
             conn = pglib.connect(
                 host=host,
@@ -423,8 +423,8 @@ def db_connect(self, dbname, user, password=None, host="localhost", port=5432):
             print("Database connection established successfully.")
             return conn
         except Exception as e:
-            print(f"Error connecting to the database: {str(e)}")
+            print(f"!!!Error connecting to the database: {str(e)}")
             if self.tunnel_process:
                 self.tunnel_process.terminate()
-                print("SSH tunnel closed due to connection failure.")
+                print("!!!SSH tunnel closed due to connection failure.")
             raise