[PR #386] test coverage

Author: gsmolk

tests/ptrack.py

@@ -4436,3 +4436,78 @@ def test_corrupt_ptrack_map(self):
 
         # Clean after yourself
         self.del_test_dir(module_name, fname)
+
+        # @unittest.skip("skip")
+    def test_horizon_lsn_ptrack(self):
+        """
+        https://github.com/postgrespro/pg_probackup/pull/386
+        """
+
+        if self.pg_config_version < self.version_to_num('11.0'):
+            return unittest.skip("You need PostgreSQL >= 11 for this test")
+
+        self.assertLessEqual(
+            self.version_to_num(self.old_probackup_version),
+            self.version_to_num('2.4.15'),
+            'You need pg_probackup old_binary =< 2.4.15 for this test')
+
+        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,
+            ptrack_enable=True,
+            initdb_params=['--data-checksums'])
+
+        self.init_pb(backup_dir)
+        self.add_instance(backup_dir, 'node', node)
+        node.slow_start()
+
+        if node.major_version >= 11:
+            node.safe_psql(
+                "postgres",
+                "CREATE EXTENSION ptrack")
+
+        # TODO: ptrack version must be 2.1
+        ptrack_version = node.safe_psql(
+            "postgres",
+            "SELECT extversion "
+			"FROM pg_catalog.pg_extension WHERE extname = 'ptrack'").decode('utf-8').rstrip()
+
+        self.assertEqual(
+            ptrack_version,
+            "2.1",
+            "You need ptrack 2.1 for this test")
+
+        # set map_size to a minimal value
+        self.set_auto_conf(node, {'ptrack.map_size': '1'})
+        node.restart()
+
+        node.pgbench_init(scale=100)
+
+        # FULL backup
+        full_id = self.backup_node(backup_dir, 'node', node, options=['--stream'], old_binary=True)
+
+        # enable archiving so the WAL size to do interfere with data bytes comparison later
+        self.set_archiving(backup_dir, 'node', node)
+        node.restart()
+
+        # change data
+        pgbench = node.pgbench(options=['-T', '10', '-c', '1', '--no-vacuum'])
+        pgbench.wait()
+
+        # DELTA is exemplar
+        delta_id = self.backup_node(
+            backup_dir, 'node', node, backup_type='delta')
+        delta_bytes = self.show_pb(backup_dir, 'node', backup_id=delta_id)["data-bytes"]
+        self.delete_pb(backup_dir, 'node', backup_id=delta_id)
+
+        # PTRACK with current binary
+        ptrack_id = self.backup_node(backup_dir, 'node', node, backup_type='ptrack')
+        ptrack_bytes = self.show_pb(backup_dir, 'node', backup_id=ptrack_id)["data-bytes"]
+
+        # make sure that backup size is exactly the same
+        self.assertEqual(delta_bytes, ptrack_bytes)
+
+        # Clean after yourself
+        self.del_test_dir(module_name, fname)