tests: fixes for ptrack

Author: gsmolk

tests/ptrack.py

@@ -3,10 +3,10 @@
 from .helpers.ptrack_helpers import ProbackupTest, ProbackupException, idx_ptrack
 from datetime import datetime, timedelta
 import subprocess
-from testgres import QueryException
+from testgres import QueryException, StartNodeException
 import shutil
 import sys
-import time
+from time import sleep
 from threading import Thread
 
 
@@ -3848,7 +3848,7 @@ def test_ptrack_zero_changes(self):
         self.del_test_dir(module_name, fname)
 
     # @unittest.skip("skip")
-    # @unittest.expectedFailure
+    @unittest.expectedFailure
     def test_ptrack_pg_resetxlog(self):
         fname = self.id().split('.')[3]
         node = self.make_simple_node(
@@ -4016,28 +4016,115 @@ def test_corrupt_ptrack_map(self):
 
         node.stop(['-m', 'immediate', '-D', node.data_dir])
 
+        ptrack_map = os.path.join(node.data_dir, 'global', 'ptrack.map')
+        ptrack_map_mmap = os.path.join(node.data_dir, 'global', 'ptrack.map.mmap')
+
         # Let`s do index corruption. ptrack.map, ptrack.map.mmap
-        with open(os.path.join(node.data_dir, 'global', 'ptrack.map'), "rb+", 0) as f:
+        with open(ptrack_map, "rb+", 0) as f:
             f.seek(42)
             f.write(b"blablahblahs")
             f.flush()
             f.close
 
-        with open(os.path.join(node.data_dir, 'global', 'ptrack.map.mmap'), "rb+", 0) as f:
+        with open(ptrack_map_mmap, "rb+", 0) as f:
             f.seek(42)
             f.write(b"blablahblahs")
             f.flush()
             f.close
 
 #        os.remove(os.path.join(node.logs_dir, node.pg_log_name))
 
+        try:
+            node.slow_start()
+            # we should die here because exception is what we expect to happen
+            self.assertEqual(
+                1, 0,
+                "Expecting Error because ptrack.map is corrupted"
+                "\n Output: {0} \n CMD: {1}".format(
+                    repr(self.output), self.cmd))
+        except StartNodeException as e:
+            self.assertIn(
+                'Cannot start node',
+                e.message,
+                '\n Unexpected Error Message: {0}\n'
+                ' CMD: {1}'.format(repr(e.message), self.cmd))
+
+        log_file = os.path.join(node.logs_dir, 'postgresql.log')
+        with open(log_file, 'r') as f:
+            log_content = f.read()
+
+        self.assertIn(
+            'FATAL:  incorrect checksum of file "{0}"'.format(ptrack_map),
+            log_content)
+
+        self.set_auto_conf(node, {'ptrack_map_size': '0'})
+
+        node.slow_start()
+
+        try:
+            self.backup_node(
+                backup_dir, 'node', node,
+                backup_type='ptrack', options=['--stream'])
+            # we should die here because exception is what we expect to happen
+            self.assertEqual(
+                1, 0,
+                "Expecting Error because instance ptrack is disabled"
+                "\n Output: {0} \n CMD: {1}".format(
+                    repr(self.output), self.cmd))
+        except ProbackupException as e:
+            self.assertIn(
+                'ERROR: Ptrack is disabled',
+                e.message,
+                '\n Unexpected Error Message: {0}\n'
+                ' CMD: {1}'.format(repr(e.message), self.cmd))
+
+        node.safe_psql(
+            'postgres',
+            "update t_heap set id = nextval('t_seq'), text = md5(text), "
+            "tsvector = md5(repeat(tsvector::text, 10))::tsvector")
+
+        node.stop(['-m', 'immediate', '-D', node.data_dir])
+
+        self.set_auto_conf(node, {'ptrack_map_size': '32'})
+
         node.slow_start()
 
+        sleep(1)
+
+        try:
+            self.backup_node(
+                backup_dir, 'node', node,
+                backup_type='ptrack', options=['--stream'])
+            # we should die here because exception is what we expect to happen
+            self.assertEqual(
+                1, 0,
+                "Expecting Error because ptrack map is from future"
+                "\n Output: {0} \n CMD: {1}".format(
+                    repr(self.output), self.cmd))
+        except ProbackupException as e:
+            self.assertIn(
+                'ERROR: LSN from ptrack_control',
+                e.message,
+                '\n Unexpected Error Message: {0}\n'
+                ' CMD: {1}'.format(repr(e.message), self.cmd))
+
+        sleep(1)
+
+        self.backup_node(
+            backup_dir, 'node', node,
+            backup_type='delta', options=['--stream'])
+
+        node.safe_psql(
+            'postgres',
+            "update t_heap set id = nextval('t_seq'), text = md5(text), "
+            "tsvector = md5(repeat(tsvector::text, 10))::tsvector")
+
         self.backup_node(
             backup_dir, 'node', node,
             backup_type='ptrack', options=['--stream'])
 
         pgdata = self.pgdata_content(node.data_dir)
+
         node.cleanup()
 
         self.restore_node(backup_dir, 'node', node)