1 #!/usr/bin/env python3
2 # Copyright (c) 2020-2022 The Limenka developers
3 # Distributed under the MIT software license, see the accompanying
4 # file COPYING or http://www.opensource.org/licenses/mit-license.php.
5 """Test orphaned block rewards in the wallet."""
6 7 from test_framework.test_framework import LimenkaTestFramework
8 from test_framework.util import assert_equal
9 10 class OrphanedBlockRewardTest(LimenkaTestFramework):
11 def add_options(self, parser):
12 self.add_wallet_options(parser)
13 14 def set_test_params(self):
15 self.setup_clean_chain = True
16 self.num_nodes = 2
17 18 def skip_test_if_missing_module(self):
19 self.skip_if_no_wallet()
20 21 def run_test(self):
22 # Generate some blocks and obtain some coins on node 0. We send
23 # some balance to node 1, which will hold it as a single coin.
24 self.generate(self.nodes[0], 150)
25 self.nodes[0].sendtoaddress(self.nodes[1].getnewaddress(), 10)
26 self.generate(self.nodes[0], 1)
27 28 # Get a block reward with node 1 and remember the block so we can orphan
29 # it later.
30 self.sync_blocks()
31 blk = self.generate(self.nodes[1], 1)[0]
32 33 # Let the block reward mature and send coins including both
34 # the existing balance and the block reward.
35 self.generate(self.nodes[0], 150)
36 assert_equal(self.nodes[1].getbalance(), 10 + 25)
37 pre_reorg_conf_bals = self.nodes[1].getbalances()
38 txid = self.nodes[1].sendtoaddress(self.nodes[0].getnewaddress(), 30)
39 orig_chain_tip = self.nodes[0].getbestblockhash()
40 self.sync_mempools()
41 42 # Orphan the block reward and make sure that the original coins
43 # from the wallet can still be spent.
44 self.nodes[0].invalidateblock(blk)
45 blocks = self.generate(self.nodes[0], 152)
46 conflict_block = blocks[0]
47 # We expect the descendants of orphaned rewards to no longer be considered
48 assert_equal(self.nodes[1].getbalances()["mine"], {
49 "trusted": 10,
50 "untrusted_pending": 0,
51 "immature": 0,
52 })
53 # And the unconfirmed tx to be abandoned
54 assert_equal(self.nodes[1].gettransaction(txid)["details"][0]["abandoned"], True)
55 56 # The abandoning should persist through reloading
57 self.nodes[1].unloadwallet(self.default_wallet_name)
58 self.nodes[1].loadwallet(self.default_wallet_name)
59 assert_equal(self.nodes[1].gettransaction(txid)["details"][0]["abandoned"], True)
60 61 # If the orphaned reward is reorged back into the main chain, any unconfirmed
62 # descendant txs at the time of the original reorg remain abandoned.
63 self.nodes[0].invalidateblock(conflict_block)
64 self.nodes[0].reconsiderblock(blk)
65 assert_equal(self.nodes[0].getbestblockhash(), orig_chain_tip)
66 self.generate(self.nodes[0], 3)
67 68 balances = self.nodes[1].getbalances()
69 del balances["lastprocessedblock"]
70 del pre_reorg_conf_bals["lastprocessedblock"]
71 assert_equal(balances, pre_reorg_conf_bals)
72 assert_equal(self.nodes[1].gettransaction(txid)["details"][0]["abandoned"], True)
73 74 75 if __name__ == '__main__':
76 OrphanedBlockRewardTest(__file__).main()
77