Merge pull request #119 from postgrespro/Add_major_version_value_into_init_helper

egarbuz · web-flow · commit 8e06f913e6a3 · 2024-04-10T16:13:05.000+03:00
Value pg_probackup major_version added to the class Init
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/init_helpers.py b/testgres/plugins/pg_probackup2/pg_probackup2/init_helpers.py
@@ -204,6 +204,12 @@ def __init__(self):
         os.environ["PGAPPNAME"] = "pg_probackup"
         self.delete_logs = delete_logs
 
+        if self.probackup_version.split('.')[0].isdigit():
+            self.major_version = int(self.probackup_version.split('.')[0])
+        else:
+            print('Can\'t process pg_probackup version \"{}\": the major version is expected to be a number'.format(self.probackup_version))
+            sys.exit(1)
+
     def test_env(self):
         return self._test_env.copy()
 
