From 24c38743976b72ba6dc3347e8ac62ac5b8f19450 Mon Sep 17 00:00:00 2001
From: Grigory Smolkin <g.smolkin@postgrespro.ru>
Date: Thu, 27 Aug 2020 17:34:35 +0300
Subject: [PATCH] tests: added
 tests.replica.ReplicaTest.test_replica_switchover

---
 tests/replica.py | 78 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)

diff --git a/tests/replica.py b/tests/replica.py
index de00c195..9a75a7aa 100644
--- a/tests/replica.py
+++ b/tests/replica.py
@@ -13,6 +13,84 @@ module_name = 'replica'
 
 class ReplicaTest(ProbackupTest, unittest.TestCase):
 
+    # @unittest.skip("skip")
+    # @unittest.expectedFailure
+    def test_replica_switchover(self):
+        """
+        check that archiving on replica works correctly
+        over the course of several switchovers
+        https://www.postgresql.org/message-id/54b059d4-2b48-13a4-6f43-95a087c92367%40postgrespro.ru
+        """
+
+        fname = self.id().split('.')[3]
+        backup_dir = os.path.join(self.tmp_path, module_name, fname, 'backup')
+        node1 = self.make_simple_node(
+            base_dir=os.path.join(module_name, fname, 'node1'),
+            set_replication=True,
+            initdb_params=['--data-checksums'])
+
+        self.init_pb(backup_dir)
+        self.add_instance(backup_dir, 'node1', node1)
+
+        node1.slow_start()
+
+        # take full backup and restore it
+        self.backup_node(backup_dir, 'node1', node1, options=['--stream'])
+        node2 = self.make_simple_node(
+            base_dir=os.path.join(module_name, fname, 'node2'))
+        node2.cleanup()
+
+        # create replica
+        self.restore_node(backup_dir, 'node1', node2)
+
+        # setup replica
+        self.add_instance(backup_dir, 'node2', node2)
+        self.set_archiving(backup_dir, 'node2', node2, replica=True)
+        self.set_replica(node1, node2, synchronous=False)
+        self.set_auto_conf(node2, {'port': node2.port})
+
+        node2.slow_start(replica=True)
+
+        # generate some data
+        node1.pgbench_init(scale=5)
+
+        # take full backup on replica
+        self.backup_node(backup_dir, 'node2', node2, options=['--stream'])
+
+        # first switchover
+        node1.stop()
+        node2.promote()
+
+        self.set_replica(node2, node1, synchronous=False)
+        node2.reload()
+        node1.slow_start(replica=True)
+
+        # take incremental backup from new master
+        self.backup_node(
+            backup_dir, 'node2', node2,
+            backup_type='delta', options=['--stream'])
+
+        # second switchover
+        node2.stop()
+        node1.promote()
+        self.set_replica(node1, node2, synchronous=False)
+        node1.reload()
+        node2.slow_start(replica=True)
+
+        # generate some more data
+        node1.pgbench_init(scale=5)
+
+        # take incremental backup from replica
+        self.backup_node(
+            backup_dir, 'node2', node2,
+            backup_type='delta', options=['--stream'])
+
+        # https://github.com/postgrespro/pg_probackup/issues/251
+        self.validate_pb(backup_dir)
+
+        # Clean after yourself
+        self.del_test_dir(module_name, fname)
+
     # @unittest.skip("skip")
     # @unittest.expectedFailure
     def test_replica_stream_ptrack_backup(self):