tests: added truncate_every_file_in_dir

gsmolk · gsmolk · commit dd2d85d11ff7 · 2019-06-10T20:56:01.000+03:00
diff --git a/tests/helpers/ptrack_helpers.py b/tests/helpers/ptrack_helpers.py
@@ -584,6 +584,12 @@ def get_backup_filelist_diff(self, filelist_A, filelist_B):
 
         return filelist_diff
 
+    # used for partial restore
+    def truncate_every_file_in_dir(self, path):
+        for file in os.listdir(path):
+            with open(os.path.join(path, file), "w") as f:
+                f.close()
+
     def check_ptrack_recovery(self, idx_dict):
         size = idx_dict['size']
         for PageNum in range(size):
@@ -734,20 +740,24 @@ def clean_pb(self, backup_dir):
 
     def backup_node(
             self, backup_dir, instance, node, data_dir=False,
-            backup_type='full', options=[], asynchronous=False, gdb=False,
+            backup_type='full', datname=False, options=[],
+            asynchronous=False, gdb=False,
             old_binary=False, return_id=True
             ):
         if not node and not data_dir:
             print('You must provide ether node or data_dir for backup')
             exit(1)
 
+        if not datname:
+            datname = 'postgres'
+
         cmd_list = [
             'backup',
             '-B', backup_dir,
             '--instance={0}'.format(instance),
             # "-D", pgdata,
             '-p', '%i' % node.port,
-            '-d', 'postgres'
+            '-d', datname
         ]
 
         if data_dir:
