Changeset View
Changeset View
Standalone View
Standalone View
test/functional/wallet_resendwallettransactions.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2017 The Bitcoin Core developers | # Copyright (c) 2017-2018 The Bitcoin Core developers | ||||
| # Distributed under the MIT software license, see the accompanying | # Distributed under the MIT software license, see the accompanying | ||||
| # file COPYING or http://www.opensource.org/licenses/mit-license.php. | # file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
| """Test resendwallettransactions RPC.""" | """Test that the wallet resends transactions periodically.""" | ||||
| from collections import defaultdict | |||||
| import time | |||||
| from test_framework.blocktools import create_block, create_coinbase | |||||
| from test_framework.messages import ToHex | |||||
| from test_framework.mininode import P2PInterface, mininode_lock | |||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import assert_equal, assert_raises_rpc_error | from test_framework.util import assert_equal, wait_until | ||||
| class P2PStoreTxInvs(P2PInterface): | |||||
| def __init__(self): | |||||
| super().__init__() | |||||
| self.tx_invs_received = defaultdict(int) | |||||
| def on_inv(self, message): | |||||
| # Store how many times invs have been received for each tx. | |||||
| for i in message.inv: | |||||
| if i.type == 1: | |||||
| # save txid | |||||
| self.tx_invs_received[i.hash] += 1 | |||||
| class ResendWalletTransactionsTest(BitcoinTestFramework): | class ResendWalletTransactionsTest(BitcoinTestFramework): | ||||
| def set_test_params(self): | def set_test_params(self): | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| self.extra_args = [['--walletbroadcast=false']] | |||||
| def skip_test_if_missing_module(self): | def skip_test_if_missing_module(self): | ||||
| self.skip_if_no_wallet() | self.skip_if_no_wallet() | ||||
| def run_test(self): | def run_test(self): | ||||
| # Should raise RPC_WALLET_ERROR (-4) if walletbroadcast is disabled. | node = self.nodes[0] # alias | ||||
| assert_raises_rpc_error(-4, "Error: Wallet transaction broadcasting is disabled with -walletbroadcast", | |||||
| self.nodes[0].resendwallettransactions) | node.add_p2p_connection(P2PStoreTxInvs()) | ||||
| # Should return an empty array if there aren't unconfirmed wallet | self.log.info("Create a new transaction and wait until it's broadcast") | ||||
| # transactions. | txid = int(node.sendtoaddress(node.getnewaddress(), 1), 16) | ||||
| self.stop_node(0) | |||||
| self.start_node(0, extra_args=[]) | # Can take a few seconds due to transaction trickling | ||||
| assert_equal(self.nodes[0].resendwallettransactions(), []) | wait_until( | ||||
| lambda: node.p2p.tx_invs_received[txid] >= 1, | |||||
| # Should return an array with the unconfirmed wallet transaction. | lock=mininode_lock) | ||||
| txid = self.nodes[0].sendtoaddress(self.nodes[0].getnewaddress(), 1) | |||||
| assert_equal(self.nodes[0].resendwallettransactions(), [txid]) | # Add a second peer since txs aren't rebroadcast to the same peer (see | ||||
| # filterInventoryKnown) | |||||
| node.add_p2p_connection(P2PStoreTxInvs()) | |||||
| self.log.info("Create a block") | |||||
| # Create and submit a block without the transaction. | |||||
| # Transactions are only rebroadcast if there has been a block at least five minutes | |||||
| # after the last time we tried to broadcast. Use mocktime and give an | |||||
| # extra minute to be sure. | |||||
| block_time = int(time.time()) + 6 * 60 | |||||
| node.setmocktime(block_time) | |||||
| block = create_block(int(node.getbestblockhash(), 16), create_coinbase( | |||||
| node.getblockchaininfo()['blocks']), block_time) | |||||
| block.nVersion = 3 | |||||
| block.rehash() | |||||
| block.solve() | |||||
| node.submitblock(ToHex(block)) | |||||
| # Transaction should not be rebroadcast | |||||
| node.p2ps[1].sync_with_ping() | |||||
| assert_equal(node.p2ps[1].tx_invs_received[txid], 0) | |||||
| self.log.info("Transaction should be rebroadcast after 30 minutes") | |||||
| # Use mocktime and give an extra 5 minutes to be sure. | |||||
| rebroadcast_time = int(time.time()) + 41 * 60 | |||||
| node.setmocktime(rebroadcast_time) | |||||
| wait_until( | |||||
| lambda: node.p2ps[1].tx_invs_received[txid] >= 1, | |||||
| lock=mininode_lock) | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| ResendWalletTransactionsTest().main() | ResendWalletTransactionsTest().main() | ||||