@@ -127,7 +127,7 @@ def __repr__(self):
127
127
128
128
129
129
class PostgresNode (object ):
130
- def __init__ (self , name = None , port = None , base_dir = None , conn_params : ConnectionParams = ConnectionParams (), bin_dir = None , prefix = None ):
130
+ def __init__ (self , name = None , base_dir = None , port = None , conn_params : ConnectionParams = ConnectionParams (), bin_dir = None , prefix = None ):
131
131
"""
132
132
PostgresNode constructor.
133
133
@@ -156,9 +156,9 @@ def __init__(self, name=None, port=None, base_dir=None, conn_params: ConnectionP
156
156
else :
157
157
self .os_ops = LocalOperations (conn_params )
158
158
159
+ self .host = self .os_ops .host
159
160
self .port = port or reserve_port ()
160
161
161
- self .host = self .os_ops .host
162
162
self .ssh_key = self .os_ops .ssh_key
163
163
164
164
# defaults for __exit__()
@@ -1690,12 +1690,13 @@ def __init__(self, test_path, nodes_to_cleanup, os_ops=LocalOperations()):
1690
1690
1691
1691
def make_empty (
1692
1692
self ,
1693
- base_dir = None ):
1693
+ base_dir = None ,
1694
+ port = None ):
1694
1695
real_base_dir = os .path .join (self .test_path , base_dir )
1695
1696
self .os_ops .rmdirs (real_base_dir , ignore_errors = True )
1696
1697
self .os_ops .makedirs (real_base_dir )
1697
1698
1698
- node = PostgresNode (base_dir = real_base_dir )
1699
+ node = PostgresNode (base_dir = real_base_dir , port = port )
1699
1700
node .should_rm_dirs = True
1700
1701
self .nodes_to_cleanup .append (node )
1701
1702
@@ -1704,14 +1705,15 @@ def make_empty(
1704
1705
def make_simple (
1705
1706
self ,
1706
1707
base_dir = None ,
1708
+ port = None ,
1707
1709
set_replication = False ,
1708
1710
ptrack_enable = False ,
1709
1711
initdb_params = [],
1710
1712
pg_options = {},
1711
1713
checksum = True ):
1712
1714
if checksum and '--data-checksums' not in initdb_params :
1713
1715
initdb_params .append ('--data-checksums' )
1714
- node = self .make_empty (base_dir )
1716
+ node = self .make_empty (base_dir , port )
1715
1717
node .init (
1716
1718
initdb_params = initdb_params , allow_streaming = set_replication )
1717
1719
0 commit comments