Add method for deeper configuration of testgres

Author: ildus

.gitignore

@@ -5,3 +5,5 @@ tags
 Dockerfile
 .coverage
 env
+/build
+coverage.xml

testgres/testgres.py

@@ -75,6 +75,11 @@
 DEFAULT_XLOG_METHOD = "fetch"
 
 
+class TestgresConfig:
+cache_pg_config = True
+cache_initdb = True
+
+
 class TestgresException(Exception):
 """
 Base exception
@@ -1081,7 +1086,7 @@ def call_initdb(_data_dir):
 raise InitNodeException(str(e))
 
 # Call initdb if we have custom params
-if initdb_params:
+if initdb_params or not TestgresConfig.cache_initdb:
 call_initdb(data_dir)
 # Else we can use cached dir
 else:
@@ -1190,25 +1195,31 @@ def get_pg_config():
 
 global pg_config_data
 
-if not pg_config_data:
-pg_config_cmd = os.environ.get("PG_CONFIG") or "pg_config"
+if TestgresConfig.cache_pg_config and pg_config_data:
+return pg_config_data
+
+data = {}
+pg_config_cmd = os.environ.get("PG_CONFIG") or "pg_config"
+out = six.StringIO(subprocess.check_output([pg_config_cmd],
+universal_newlines=True))
+for line in out:
+if line and "=" in line:
+key, value = line.split("=", 1)
+data[key.strip()] = value.strip()
 
-out = six.StringIO(subprocess.check_output([pg_config_cmd],
-universal_newlines=True))
-for line in out:
-if line and "=" in line:
-key, value = line.split("=", 1)
-pg_config_data[key.strip()] = value.strip()
+# Fetch version of PostgreSQL and save it as VERSION_NUM
+version = data["VERSION"]
+version = version.split(" ")[-1] \
+.partition('devel')[0] \
+.partition('beta')[0] \
+.partition('rc')[0]
+data["VERSION_NUM"] = version
 
-# Fetch version of PostgreSQL and save it as VERSION_NUM
-version = pg_config_data["VERSION"]
-version = version.split(" ")[-1] \
-.partition('devel')[0] \
-.partition('beta')[0] \
-.partition('rc')[0]
-pg_config_data["VERSION_NUM"] = version
+if TestgresConfig.cache_pg_config:
+pg_config_data.clear()
+pg_config_data.update(data)
 
-return pg_config_data
+return data
 
 
 def get_new_node(name, base_dir=None, use_logging=False):
@@ -1225,3 +1236,11 @@ def get_new_node(name, base_dir=None, use_logging=False):
 """
 
 return PostgresNode(name=name, base_dir=base_dir, use_logging=use_logging)
+
+
+def configure_testgres(**options):
+'''
+Configure testgres. Look for TestgresConfig to check what can be changed.
+'''
+for key, option in options.items():
+setattr(TestgresConfig, key, option)

tests/test_simple.py

@@ -6,14 +6,15 @@
 import tempfile
 import logging.config
 import subprocess
+import testgres
 
 from distutils.version import LooseVersion
 
 from testgres import InitNodeException, \
 StartNodeException, ExecUtilException, \
 BackupException, QueryException, CatchUpException
 
-from testgres import get_new_node, get_pg_config
+from testgres import get_new_node, get_pg_config, configure_testgres
 from testgres import bound_ports
 from testgres import NodeStatus
 
@@ -414,6 +415,19 @@ def test_version_management(self):
 self.assertTrue(b > c)
 self.assertTrue(a > c)
 
+def test_configure(self):
+# set global if it wasn't set
+pg_config = get_pg_config()
+configure_testgres(cache_initdb=True, cache_pg_config=True)
+
+# check that is the same instance
+self.assertEqual(id(get_pg_config()), id(testgres.pg_config_data))
+configure_testgres(cache_initdb=True, cache_pg_config=False)
+self.assertNotEqual(id(get_pg_config()), id(testgres.pg_config_data))
+
+# return to the base state
+configure_testgres(cache_initdb=True, cache_pg_config=True)
+
 
 if __name__ == '__main__':
 unittest.main()