introduce default_conf()

Author: funbringer

testgres/testgres.py

@@ -475,7 +475,6 @@ def init(self, allow_streaming=False, fsync=False, initdb_params=[]):
         """
 
         postgres_conf = os.path.join(self.data_dir, "postgresql.conf")
-        hba_conf = os.path.join(self.data_dir, "pg_hba.conf")
 
         # We don't have to reinit it if data directory exists
         if os.path.isfile(postgres_conf):
@@ -485,6 +484,26 @@ def init(self, allow_streaming=False, fsync=False, initdb_params=[]):
         initdb_log = os.path.join(self.logs_dir, "initdb.log")
         _cached_initdb(self.data_dir, initdb_log, initdb_params)
 
+        # initialize default config files
+        self.default_conf(fsync=fsync)
+
+        return self
+
+    def default_conf(self, allow_streaming=True, fsync=False):
+        """
+        Apply default settings to this node.
+
+        Args:
+            allow_streaming: should this node add a hba entry for replication?
+            fsync: should this node use fsync to keep data safe?
+
+        Returns:
+            This instance of PostgresNode.
+        """
+
+        postgres_conf = os.path.join(self.data_dir, "postgresql.conf")
+        hba_conf = os.path.join(self.data_dir, "pg_hba.conf")
+
         # add parameters to hba file
         with open(hba_conf, "w") as conf:
             conf.write("# TYPE\tDATABASE\tUSER\tADDRESS\t\tMETHOD\n"
@@ -502,9 +521,9 @@ def init(self, allow_streaming=False, fsync=False, initdb_params=[]):
                 conf.write("fsync = off\n")
 
             conf.write("log_statement = all\n"
-                       "port = {}\n".format(self.port))
-
-            conf.write("listen_addresses = '{}'\n".format(self.host))
+                       "listen_addresses = '{}'\n"
+                       "port = {}\n".format(self.host,
+                                            self.port))
 
             if allow_streaming:
                 cur_ver = LooseVersion(get_pg_config()["VERSION_NUM"])
@@ -516,7 +535,6 @@ def init(self, allow_streaming=False, fsync=False, initdb_params=[]):
                 conf.write("max_wal_senders = 5\n"
                            "wal_keep_segments = 20\n"
                            "hot_standby = on\n"
-                           "max_connections = 10\n"
                            "wal_level = {}\n".format(wal_level))
 
         return self