testgres-pg-probackup2 0.3.0__tar.gz → 0.5.0__tar.gz

{testgres_pg_probackup2-0.3.0 → testgres_pg_probackup2-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: testgres_pg_probackup2
-Version: 0.3.0
+Version: 0.5.0
 Summary: Plugin for testgres that manages pg_probackup2
 Home-page: https://github.com/postgrespro/testgres
 Author: Postgres Professional

{testgres_pg_probackup2-0.3.0 → testgres_pg_probackup2-0.5.0}/pg_probackup2/app.py

@@ -41,15 +41,24 @@ class ProbackupException(Exception):
 fs_backup_class = FSTestBackupDir
 
 
+class _OutputContainer:
+    """Internal container for output and cmd when test_class is not provided"""
+    def __init__(self):
+        self.output = None
+        self.cmd = None
+
+
 class ProbackupApp:
 
-    def __init__(self, test_class: unittest.TestCase,
-                 pg_node, pb_log_path, test_env, auto_compress_alg, backup_dir, probackup_path=None):
+    def __init__(self, test_class: unittest.TestCase = None,
+                 pg_node=None, pb_log_path=None, test_env=None, auto_compress_alg=None,
+                 backup_dir=None, probackup_path=None):
         self.process = None
-        self.test_class = test_class
+        # Use provided test_class or create internal container
+        self.test_class = test_class if test_class is not None else _OutputContainer()
         self.pg_node = pg_node
         self.pb_log_path = pb_log_path
-        self.test_env = test_env
+        self.test_env = test_env if test_env is not None else os.environ.copy()
         self.auto_compress_alg = auto_compress_alg
         self.backup_dir = backup_dir
         self.probackup_path = probackup_path or init_params.probackup_path
@@ -571,7 +580,7 @@ class ProbackupApp:
                         backup_list.append(backup)
 
             if backup_id is not None:
-                assert False, "Failed to find backup with ID: {0}".format(backup_id)
+                raise RuntimeError("Failed to find backup with ID: {0}".format(backup_id))
 
             return backup_list
         else:
@@ -629,7 +638,7 @@ class ProbackupApp:
                     specific_record[name.strip()] = var
 
                 if not specific_record:
-                    assert False, "Failed to find backup with ID: {0}".format(backup_id)
+                    raise RuntimeError("Failed to find backup with ID: {0}".format(backup_id))
 
                 return specific_record
 
@@ -778,6 +787,13 @@ class ProbackupApp:
             log_level=False, archive_timeout=False,
             custom_archive_command=None):
 
+        # check instance existing
+        instances_json = self.show(instance=None, as_json=True, expect_error=False, as_text=True)
+        instances_data = json.loads(instances_json)
+        if not any(inst_data.get('instance') == instance for inst_data in instances_data):
+            raise RuntimeError("Instance '{0}' does not exist. " \
+                          "Please add the instance first using add_instance() or init_pb_node().".format(instance))
+
         # parse postgresql.auto.conf
         options = {}
         if replica:

{testgres_pg_probackup2-0.3.0 → testgres_pg_probackup2-0.5.0}/setup.py

@@ -4,7 +4,7 @@ except ImportError:
     from distutils.core import setup
 
 setup(
-    version='0.3.0',
+    version='0.5.0',
     name='testgres_pg_probackup2',
     packages=['pg_probackup2', 'pg_probackup2.storage'],
     description='Plugin for testgres that manages pg_probackup2',

{testgres_pg_probackup2-0.3.0 → testgres_pg_probackup2-0.5.0}/testgres_pg_probackup2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: testgres_pg_probackup2
-Version: 0.3.0
+Version: 0.5.0
 Summary: Plugin for testgres that manages pg_probackup2
 Home-page: https://github.com/postgrespro/testgres
 Author: Postgres Professional