[Tests] Introduce tiertwo_reorg_mempool test

random-zebra · random-zebra · commit 57fa99f950c6 · 2021-05-29T12:50:58.000+02:00
diff --git a/test/functional/test_framework/test_framework.py b/test/functional/test_framework/test_framework.py
@@ -1312,6 +1312,7 @@ def register_new_dmn(self, idx, miner_idx, controller_idx, strType,
         return dmn
 
     def check_mn_list_on_node(self, idx, mns):
+        self.nodes[idx].syncwithvalidationinterfacequeue()
         mnlist = self.nodes[idx].listmasternodes()
         if len(mnlist) != len(mns):
             raise Exception("Invalid mn list on node %d:\n%s\nExpected:%s" % (idx, str(mnlist), str(mns)))
diff --git a/test/functional/test_runner.py b/test/functional/test_runner.py
@@ -141,11 +141,12 @@
 
 TIERTWO_SCRIPTS = [
     # Longest test should go first, to favor running tests in parallel
-    'tiertwo_deterministicmns.py',
-    'tiertwo_governance_sync_basic.py',
-    'tiertwo_mn_compatibility.py',
-    'tiertwo_masternode_activation.py',
-    'tiertwo_masternode_ping.py',
+    'tiertwo_governance_sync_basic.py',         # ~ 445 sec
+    'tiertwo_mn_compatibility.py',              # ~ 413 sec
+    'tiertwo_deterministicmns.py',              # ~ 366 sec
+    'tiertwo_masternode_activation.py',         # ~ 352 sec
+    'tiertwo_masternode_ping.py',               # ~ 293 sec
+    'tiertwo_reorg_mempool.py',                 # ~ 107 sec
 ]
 
 SAPLING_SCRIPTS = [
diff --git a/test/functional/tiertwo_reorg_mempool.py b/test/functional/tiertwo_reorg_mempool.py
@@ -0,0 +1,244 @@
+#!/usr/bin/env python3
+# Copyright (c) 2021 The PIVX Core developers
+# Distributed under the MIT software license, see the accompanying
+# file COPYING or http://www.opensource.org/licenses/mit-license.php.
+
+"""
+Test deterministic masternodes conflicts and reorgs.
+- Check that in-mempool reuse of mn unique-properties is invalid
+- Check mempool eviction after conflict with newly connected block / reorg
+- Check deterministic list consensus after reorg
+"""
+
+import random
+import time
+
+from test_framework.test_framework import PivxTestFramework
+
+from test_framework.util import (
+    assert_equal,
+    assert_raises_rpc_error,
+    create_new_dmn,
+    connect_nodes,
+    disconnect_nodes,
+)
+
+class TiertwoReorgMempoolTest(PivxTestFramework):
+
+    def set_test_params(self):
+        # two nodes mining on separate chains
+        self.num_nodes = 2
+        self.setup_clean_chain = True
+        self.extra_args = [["-nuparams=v5_shield:1", "-nuparams=v6_evo:160"]] * self.num_nodes
+        self.extra_args[0].append("-sporkkey=932HEevBSujW2ud7RfB1YF91AFygbBRQj3de3LyaCRqNzKKgWXi")
+
+    def setup_network(self):
+        self.setup_nodes()
+        self.connect_all()
+
+    def connect_all(self):
+        connect_nodes(self.nodes[0], 1)
+        connect_nodes(self.nodes[1], 0)
+
+    def disconnect_all(self):
+        self.log.info("Disconnecting nodes...")
+        disconnect_nodes(self.nodes[0], 1)
+        disconnect_nodes(self.nodes[1], 0)
+        self.log.info("Nodes disconnected")
+
+    def register_masternode(self, from_node, dmn, collateral_addr):
+        dmn.proTx = from_node.protx_register_fund(collateral_addr, dmn.ipport, dmn.owner,
+                                                  dmn.operator, dmn.voting, dmn.payee)
+
+    def run_test(self):
+        self.disable_mocktime()
+        nodeA = self.nodes[0]
+        nodeB = self.nodes[1]
+        free_idx = 1  # unique id for masternodes. first available.
+
+        # Enforce mn payments and reject legacy mns at block 202
+        self.activate_spork(0, "SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT")
+        assert_equal("success", self.set_spork(0, "SPORK_21_LEGACY_MNS_MAX_HEIGHT", 201))
+        time.sleep(1)
+        assert_equal([201] * self.num_nodes, [self.get_spork(x, "SPORK_21_LEGACY_MNS_MAX_HEIGHT")
+                                              for x in range(self.num_nodes)])
+
+        # Mine 201 blocks
+        self.log.info("Mining...")
+        nodeA.generate(25)
+        self.sync_blocks()
+        nodeB.generate(25)
+        self.sync_blocks()
+        nodeA.generate(50)
+        self.sync_blocks()
+        nodeB.generate(101)
+        self.sync_blocks()
+        self.assert_equal_for_all(201, "getblockcount")
+
+        # Register one masternode before the split
+        collateral_addr = nodeA.getnewaddress() # for both collateral and payouts
+        pre_split_mn = create_new_dmn(100, nodeA, nodeA.getnewaddress(), None)
+        self.register_masternode(nodeA, pre_split_mn, collateral_addr)
+        nodeA.generate(1)
+        self.sync_blocks()
+        mnsA = [pre_split_mn]
+        mnsB = [pre_split_mn]
+        self.check_mn_list_on_node(0, mnsA)
+        self.check_mn_list_on_node(1, mnsB)
+        self.log.info("Pre-split masternode registered.")
+
+        # Disconnect the nodes
+        self.disconnect_all()   # network splits at block 203
+
+        #
+        # -- CHAIN A --
+        #
+
+        # Register 5 masternodes, then mine 5 blocks
+        self.log.info("Registering masternodes on chain A...")
+        for _ in range(5):
+            dmn = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+            free_idx += 1
+            self.register_masternode(nodeA, dmn, collateral_addr)
+            mnsA.append(dmn)
+        nodeA.generate(5)
+        self.check_mn_list_on_node(0, mnsA)
+        self.log.info("Masternodes registered on chain A.")
+
+        # Lock any utxo with less than 101 confs (e.g. change), so we can resurrect everything
+        for x in nodeA.listunspent(0, 101):
+            nodeA.lockunspent(False, [{"txid": x["txid"], "vout": x["vout"]}])
+
+        # Now send a valid proReg tx to the mempool, without mining it
+        mempool_dmn1 = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+        free_idx += 1
+        self.register_masternode(nodeA, mempool_dmn1, collateral_addr)
+        assert mempool_dmn1.proTx in nodeA.getrawmempool()
+
+        # Try sending a proReg tx with same owner
+        self.log.info("Testing in-mempool duplicate-owner rejection...")
+        dmn_A1 = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+        free_idx += 1
+        dmn_A1.owner = mempool_dmn1.owner
+        assert_raises_rpc_error(-26, "protx-dup",
+                                self.register_masternode, nodeA, dmn_A1, collateral_addr)
+        assert dmn_A1.proTx not in nodeA.getrawmempool()
+
+        # Try sending a proReg tx with same operator
+        self.log.info("Testing in-mempool duplicate-operator rejection...")
+        dmn_A2 = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+        free_idx += 1
+        dmn_A2.operator = mempool_dmn1.operator
+        assert_raises_rpc_error(-26, "protx-dup",
+                                self.register_masternode, nodeA, dmn_A2, collateral_addr)
+        assert dmn_A2.proTx not in nodeA.getrawmempool()
+
+        # Try sending a proReg tx with same IP
+        self.log.info("Testing proReg in-mempool duplicate-IP rejection...")
+        dmn_A3 = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+        free_idx += 1
+        dmn_A3.ipport = mempool_dmn1.ipport
+        assert_raises_rpc_error(-26, "protx-dup",
+                                self.register_masternode, nodeA, dmn_A3, collateral_addr)
+        assert dmn_A3.proTx not in nodeA.getrawmempool()
+
+        # Now send other 2 valid proReg tx to the mempool, without mining it
+        mempool_dmn2 = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+        free_idx += 1
+        mempool_dmn3 = create_new_dmn(free_idx, nodeA, collateral_addr, None)
+        free_idx += 1
+        self.register_masternode(nodeA, mempool_dmn2, collateral_addr)
+        self.register_masternode(nodeA, mempool_dmn3, collateral_addr)
+
+        # Now nodeA has 3 proReg txes in its mempool
+        mempoolA = nodeA.getrawmempool()
+        assert mempool_dmn1.proTx in mempoolA
+        assert mempool_dmn2.proTx in mempoolA
+        assert mempool_dmn3.proTx in mempoolA
+
+        assert_equal(nodeA.getblockcount(), 207)
+
+        #
+        # -- CHAIN B --
+        #
+        collateral_addr = nodeB.getnewaddress()
+        self.log.info("Registering masternodes on chain B...")
+
+        # Register first the 3 nodes that conflict with the mempool of nodes[0]
+        # mine one block after each registration
+        for dmn in [dmn_A1, dmn_A2, dmn_A3]:
+            self.register_masternode(nodeB, dmn, collateral_addr)
+            mnsB.append(dmn)
+            nodeB.generate(1)
+        self.check_mn_list_on_node(1, mnsB)
+
+        # Pick the proReg for the first MN registered on chain A, and replay it on chain B
+        self.log.info("Replaying a masternode on a different chain...")
+        mnsA.remove(pre_split_mn)
+        replay_mn = mnsA.pop(0)
+        mnsB.append(replay_mn)  # same proTx hash
+        nodeB.sendrawtransaction(nodeA.getrawtransaction(replay_mn.proTx, False))
+        nodeB.generate(1)
+        self.check_mn_list_on_node(1, mnsB)
+
+        # Now pick a proReg for another MN registered on chain A, and re-register it on chain B
+        self.log.info("Re-registering a masternode on a different chain...")
+        rereg_mn = random.choice(mnsA)
+        mnsA.remove(rereg_mn)
+        self.register_masternode(nodeB, rereg_mn, collateral_addr)
+        mnsB.append(rereg_mn)   # changed proTx hash
+        nodeB.generate(1)
+        self.check_mn_list_on_node(1, mnsB)
+
+        # Register 5 more masternodes. One per block.
+        for _ in range(5):
+            dmn = create_new_dmn(free_idx, nodeB, collateral_addr, None)
+            free_idx += 1
+            self.register_masternode(nodeB, dmn, collateral_addr)
+            mnsB.append(dmn)
+            nodeB.generate(1)
+
+        # Then mine 10 more blocks on chain B
+        nodeB.generate(10)
+        self.check_mn_list_on_node(1, mnsB)
+        self.log.info("Masternodes registered on chain B.")
+
+        assert_equal(nodeB.getblockcount(), 222)
+
+        #
+        # -- RECONNECT --
+        #
+
+        # Reconnect and sync (give it some more time)
+        self.log.info("Reconnecting nodes...")
+        self.connect_all()
+        # !TODO: FIXME - failing because we check budget/mn payment in CheckBlock
+        # during a reorg, the previous block hasn't been connected yet, so the dmn list is empty.
+        self.sync_blocks(wait=3, timeout=180)
+
+        # Both nodes have the same list (mnB)
+        self.log.info("Checking masternode list...")
+        self.check_mn_list_on_node(0, mnsB)
+        self.check_mn_list_on_node(1, mnsB)
+        self.log.info("Both nodes have %d registered masternodes." % len(mnsB))
+
+        # The first mempool proReg tx has been removed from nodeA's mempool due to
+        # conflicts with the masternodes of chain B, now connected.
+        self.log.info("Checking mempool...")
+        mempoolA = nodeA.getrawmempool()
+        assert mempool_dmn1.proTx not in mempoolA
+        assert mempool_dmn2.proTx in mempoolA
+        assert mempool_dmn3.proTx in mempoolA
+        # The mempool contains also all the ProReg from the disconnected blocks,
+        # except the ones re-registered and replayed on chain B.
+        for mn in mnsA:
+            assert mn.proTx in mempoolA
+        assert rereg_mn.proTx not in mempoolA
+        assert replay_mn.proTx not in mempoolA
+        assert pre_split_mn.proTx not in mempoolA
+
+        self.log.info("All good.")
+
+
+if __name__ == '__main__':
+    TiertwoReorgMempoolTest().main()
