Add PG_PROBACKUP_WAL_TREE_ENABLED env param

Stepan Neretin · Stepan Neretin · commit 889794653984 · 2024-09-05T16:08:08.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
@@ -185,6 +185,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)
