tests: added tests.merge.MergeTest.test_missing_data_file

gsmolk · gsmolk · commit c90b45ae3b05 · 2020-08-27T20:40:16.000+03:00
diff --git a/tests/merge.py b/tests/merge.py
@@ -2655,5 +2655,69 @@ def test_merge_page_header_map_retry(self):
         # Clean after yourself
         self.del_test_dir(module_name, fname)
 
+    # @unittest.skip("skip")
+    def test_missing_data_file(self):
+        """
+        """
+        fname = self.id().split('.')[3]
+        backup_dir = os.path.join(self.tmp_path, module_name, fname, 'backup')
+        node = self.make_simple_node(
+            base_dir=os.path.join(module_name, fname, 'node'),
+            set_replication=True,
+            initdb_params=['--data-checksums'],
+            pg_options={'autovacuum': 'off'})
+
+        self.init_pb(backup_dir)
+        self.add_instance(backup_dir, 'node', node)
+        self.set_archiving(backup_dir, 'node', node)
+        node.slow_start()
+
+        # Add data
+        node.pgbench_init(scale=1)
+
+        # FULL backup
+        self.backup_node(backup_dir, 'node', node)
+
+        # Change data
+        pgbench = node.pgbench(options=['-T', '5', '-c', '1'])
+        pgbench.wait()
+
+        # DELTA backup
+        delta_id = self.backup_node(backup_dir, 'node', node, backup_type='delta')
+
+        path = node.safe_psql(
+            'postgres',
+            "select pg_relation_filepath('pgbench_accounts')").rstrip()
+
+        pgdata = self.pgdata_content(node.data_dir)
+
+        print(self.show_pb(
+            backup_dir, 'node', as_json=False, as_text=True))
+
+        gdb = self.merge_backup(
+            backup_dir, "node", delta_id,
+            options=['--log-level-file=VERBOSE'], gdb=True)
+        gdb.set_breakpoint('merge_files')
+        gdb.run_until_break()
+
+        # remove data file in incremental backup
+        file_to_remove = os.path.join(
+            backup_dir, 'backups',
+            'node', delta_id, 'database', path)
+
+        os.remove(file_to_remove)
+
+        gdb.continue_execution_until_error()
+
+        logfile = os.path.join(backup_dir, 'log', 'pg_probackup.log')
+        with open(logfile, 'r') as f:
+                logfile_content = f.read()
+
+        self.assertIn(
+            'ERROR: Cannot open backup file "{0}": No such file or directory'.format(file_to_remove),
+            logfile_content)
+
+        self.del_test_dir(module_name, fname)
+
 # 1. Need new test with corrupted FULL backup
 # 2. different compression levels
