test_backup_via_unprivileged_user - test with ptrack correctly

funny-falcon · funny-falcon · commit e674202eac14 · 2022-11-24T14:42:12.000+03:00
diff --git a/tests/auth_test.py b/tests/auth_test.py
@@ -33,6 +33,7 @@ def test_backup_via_unprivileged_user(self):
         node = self.make_simple_node(
             base_dir=os.path.join(self.module_name, self.fname, 'node'),
             set_replication=True,
+            ptrack_enable=self.ptrack,
             initdb_params=['--data-checksums'])
 
         backup_dir = os.path.join(self.tmp_path, self.module_name, self.fname, 'backup')
@@ -41,6 +42,11 @@ def test_backup_via_unprivileged_user(self):
         self.set_archiving(backup_dir, 'node', node)
         node.slow_start()
 
+        if self.ptrack:
+            node.safe_psql(
+                "postgres",
+                "CREATE EXTENSION ptrack")
+
         node.safe_psql("postgres", "CREATE ROLE backup with LOGIN")
 
         try:
@@ -160,9 +166,10 @@ def test_backup_via_unprivileged_user(self):
             backup_dir, 'node', node, options=['-U', 'backup'])
 
         # PTRACK
-#        self.backup_node(
-#            backup_dir, 'node', node,
-#            backup_type='ptrack', options=['-U', 'backup'])
+        if self.ptrack:
+            self.backup_node(
+                backup_dir, 'node', node,
+                backup_type='ptrack', options=['-U', 'backup'])
 
 
 class AuthTest(unittest.TestCase):
