You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.9 KiB
88 lines
2.9 KiB
#!/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 transactions
|
|
|
|
Test re-org scenarios with a mempool that contains transactions
|
|
that create or spend (directly or indirectly) MWEB outputs.
|
|
"""
|
|
|
|
import json
|
|
|
|
from test_framework.blocktools import create_raw_transaction
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import assert_equal, assert_raises_rpc_error
|
|
from test_framework.ltc_util import setup_mweb_chain
|
|
|
|
class 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()
|