Add PG_PROBACKUP_WAL_TREE_ENABLED env param

Stepan Neretin · Stepan Neretin · commit a660ab187e59 · 2024-09-05T16:11:52.000+07:00
PG_PROBACKUP_WAL_TREE_ENABLED used to set to probackup setting when creating instance how wals should be located
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/app.py b/testgres/plugins/pg_probackup2/pg_probackup2/app.py
@@ -54,6 +54,7 @@ def __init__(self, test_class: unittest.TestCase,
         self.probackup_path = probackup_path or init_params.probackup_path
         self.probackup_old_path = init_params.probackup_old_path
         self.remote = init_params.remote
+        self.wal_tree_enabled = init_params.wal_tree_enabled
         self.verbose = init_params.verbose
         self.archive_compress = init_params.archive_compress
         self.test_class.output = None
@@ -185,6 +186,9 @@ def add_instance(self, instance, node, old_binary=False, options=None, expect_er
                 '--remote-proto=ssh',
                 '--remote-host=localhost']
 
+        if self.wal_tree_enabled:
+            options = options + ['--wal-tree']
+
         return self.run(cmd + options, old_binary=old_binary, expect_error=expect_error)
 
     def set_config(self, instance, old_binary=False, options=None, expect_error=False):
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/init_helpers.py b/testgres/plugins/pg_probackup2/pg_probackup2/init_helpers.py
@@ -170,6 +170,7 @@ def __init__(self):
 
         self.remote = test_env.get('PGPROBACKUP_SSH_REMOTE', None) == 'ON'
         self.ptrack = test_env.get('PG_PROBACKUP_PTRACK', None) == 'ON' and self.pg_config_version >= 110000
+        self.wal_tree_enabled = test_env.get('PG_PROBACKUP_WAL_TREE_ENABLED', None) == 'ON'
 
         self.paranoia = test_env.get('PG_PROBACKUP_PARANOIA', None) == 'ON'
         env_compress = test_env.get('ARCHIVE_COMPRESSION', None)
