Add repair command

vshepard · vshepard · commit 6b95083bb914 · 2025-01-05T15:23:59.000+01:00
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/app.py b/testgres/plugins/pg_probackup2/pg_probackup2/app.py
@@ -811,5 +811,22 @@ def archive_get(self, instance, wal_file_name, wal_file_path, options=None, expe
         ]
         return self.run(cmd + options, expect_error=expect_error)
 
+    def repair(
+            self, instance=None, backup_id=None,
+            options=None, old_binary=False, gdb=False, expect_error=False
+    ):
+        if options is None:
+            options = []
+        cmd_list = [
+            'repair',
+        ]
+        if instance:
+            cmd_list += ['--instance={0}'.format(instance)]
+        if backup_id:
+            cmd_list += ['-i', backup_id]
+
+        return self.run(cmd_list + options, old_binary=old_binary, gdb=gdb,
+                        expect_error=expect_error)
+
     def build_backup_dir(self, backup='backup'):
         return fs_backup_class(rel_path=self.rel_path, backup=backup)
