[create_archive_push] archive_push_command in testgres_pg_probackup added

z-kasymalieva · z-kasymalieva · commit b4e17e10de62 · 2024-02-02T18:24:04.000+03:00
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/app.py b/testgres/plugins/pg_probackup2/pg_probackup2/app.py
@@ -748,5 +748,28 @@ 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'))]
+        return self.run(cmd + options, expect_error=expect_error)
+
+
+# Local or S3 backup
+fs_backup_class = FSTestBackupDir
+if os.environ.get('PG_PROBACKUP_S3_TEST', os.environ.get('PROBACKUP_S3_TYPE_FULL_TEST')):
+    root = os.path.realpath(os.path.join(os.path.dirname(__file__), '../..'))
+    if root not in sys.path:
+        sys.path.append(root)
+    from pg_probackup2.storage.s3_backup import S3TestBackupDir
+
+    fs_backup_class = S3TestBackupDir
+
     def build_backup_dir(self, backup='backup'):
         return fs_backup_class(rel_path=self.rel_path, backup=backup)
