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()