Add tunnel_port to remote_ops.py

vshepard · vshepard · commit fffb23c9b00e · 2024-06-06T07:52:16.000+02:00
diff --git a/testgres/operations/remote_ops.py b/testgres/operations/remote_ops.py
@@ -58,6 +58,7 @@ def __init__(self, conn_params: ConnectionParams):
         self.remote = True
         self.username = conn_params.username or self.get_user()
         self.tunnel_process = None
+        self.tunnel_port = None
 
     def __enter__(self):
         return self
@@ -410,6 +411,7 @@ def db_connect(self, dbname, user, password=None, host="localhost", port=5432):
         Establish SSH tunnel and connect to a PostgreSQL database.
         """
         local_port = reserve_port()
+        self.tunnel_port = local_port
         self.establish_ssh_tunnel(local_port=local_port, remote_port=port)
         try:
             conn = pglib.connect(
