git / moonbite-core / test/functional/mweb_reorg.py · raw

#!/usr/bin/env python3# Copyright (c) 2014-2020 The Bitcoin Core developers# Distributed under the MIT software license, see the accompanying# file COPYING or http://www.opensource.org/licenses/mit-license.php."""Test mempool re-org scenarios for MWEB transactionsTest re-org scenarios with a mempool that contains transactionsthat create or spend (directly or indirectly) MWEB outputs."""import jsonfrom test_framework.blocktools import create_raw_transactionfrom test_framework.test_framework import BitcoinTestFrameworkfrom test_framework.util import assert_equal, assert_raises_rpc_errorfrom test_framework.ltc_util import setup_mweb_chainclass MWEBReorgTest(BitcoinTestFramework):    def set_test_params(self):        self.setup_clean_chain = True        self.num_nodes = 2        self.extra_args = [            [                '-whitelist=noban@127.0.0.1',  # immediate tx relay            ],            []        ]    def skip_test_if_missing_module(self):        self.skip_if_no_wallet()    def run_test(self):        self.basic_reorg_test()        self.multiple_blocks_reorg_test()    def basic_reorg_test(self):        self.log.info("Create all pre-MWEB blocks")        setup_mweb_chain(self.nodes[0])                self.log.info("Pegin some coins in pegin_tx1. pegin_tx1 should be in the mempool")        node0_mweb_addr = self.nodes[0].getnewaddress(address_type='mweb')        pegin_tx1_id = self.nodes[0].sendtoaddress(node0_mweb_addr, 100)        self.sync_all()        assert_equal(set(self.nodes[1].getrawmempool()), {pegin_tx1_id})        self.log.info("Mine pegin_tx1 in block0a, and mine a few blocks on top. mempool should be empty")        block0a = self.nodes[0].generate(4)[0]        self.sync_all()        assert_equal(len(self.nodes[1].getrawmempool()), 0)        self.log.info("Invalidate block0a. pegin_tx1 should be back in the mempool")        self.nodes[1].invalidateblock(block0a)        assert_equal(set(self.nodes[1].getrawmempool()), {pegin_tx1_id})                self.log.info("Generate block0b. pegin_tx1 should be included in the block")        block0b_hash = self.nodes[1].generate(1)[0]        block0b_txs = self.nodes[1].getblock(block0b_hash, 2)['tx']        assert_equal(len(block0b_txs), 3)        assert_equal(block0b_txs[1]['txid'], pegin_tx1_id)        self.nodes[1].generate(5)        self.sync_blocks()    def multiple_blocks_reorg_test(self):        node0 = self.nodes[0]        node1 = self.nodes[1]        # Make sure nodes are in sync        node0.generate(10)        self.sync_all()        # disconnect the nodes and mine separate forks        self.disconnect_nodes(0, 1)        node0.generate(5)        node1.generate(5)                # reconnect the nodes and have node0 mine 1 block, forcing a reorg on node1        self.connect_nodes(0, 1)        node0.generate(1)        self.sync_blocks()if __name__ == '__main__':    MWEBReorgTest().main()