postgrespro · z-kasymalieva · Feb 15, 2024 · Feb 2, 2024
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/app.py b/testgres/plugins/pg_probackup2/pg_probackup2/app.py
@@ -748,5 +748,19 @@ def load_backup_class(fs_type):
 
         return getattr(module, class_name)
 
+    def archive_push(self, instance, node, wal_file_name, wal_file_path=None, options=None, expect_error=False):
+        if options is None:
+            options = []
+        cmd = [
+            'archive-push',
+            '--instance={0}'.format(instance),
+            '--wal-file-name={0}'.format(wal_file_name),
+        ]
+        if wal_file_path is None:
+            cmd = cmd + ['--wal-file-path={0}'.format(os.path.join(node.data_dir, 'pg_wal'))]
+        else:
+            cmd = cmd + ['--wal-file-path={0}'.format(wal_file_path)]
+        return self.run(cmd + options, expect_error=expect_error)
+
     def build_backup_dir(self, backup='backup'):
         return fs_backup_class(rel_path=self.rel_path, backup=backup)