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.
111 lines
5.1 KiB
111 lines
5.1 KiB
#!/usr/bin/env python3 |
|
# Copyright (c) 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. |
|
"""Tests that a mempool transaction expires after a given timeout and that its |
|
children are removed as well. |
|
|
|
Both the default expiry timeout defined by DEFAULT_MEMPOOL_EXPIRY and a user |
|
definable expiry timeout via the '-mempoolexpiry=<n>' command line argument |
|
(<n> is the timeout in hours) are tested. |
|
""" |
|
|
|
from datetime import timedelta |
|
|
|
from test_framework.blocktools import COINBASE_MATURITY |
|
from test_framework.test_framework import BitcoinTestFramework |
|
from test_framework.util import ( |
|
assert_equal, |
|
assert_raises_rpc_error, |
|
) |
|
from test_framework.wallet import MiniWallet |
|
|
|
DEFAULT_MEMPOOL_EXPIRY = 336 # hours |
|
CUSTOM_MEMPOOL_EXPIRY = 10 # hours |
|
|
|
|
|
class MempoolExpiryTest(BitcoinTestFramework): |
|
def set_test_params(self): |
|
self.num_nodes = 1 |
|
self.setup_clean_chain = True |
|
|
|
def test_transaction_expiry(self, timeout): |
|
"""Tests that a transaction expires after the expiry timeout and its |
|
children are removed as well.""" |
|
node = self.nodes[0] |
|
self.wallet = MiniWallet(node) |
|
|
|
# Add enough mature utxos to the wallet so that all txs spend confirmed coins. |
|
self.wallet.generate(4) |
|
node.generate(COINBASE_MATURITY) |
|
|
|
# Send a parent transaction that will expire. |
|
parent_txid = self.wallet.send_self_transfer(from_node=node)['txid'] |
|
parent_utxo = self.wallet.get_utxo(txid=parent_txid) |
|
independent_utxo = self.wallet.get_utxo() |
|
|
|
# Ensure the transactions we send to trigger the mempool check spend utxos that are independent of |
|
# the transactions being tested for expiration. |
|
trigger_utxo1 = self.wallet.get_utxo() |
|
trigger_utxo2 = self.wallet.get_utxo() |
|
|
|
# Set the mocktime to the arrival time of the parent transaction. |
|
entry_time = node.getmempoolentry(parent_txid)['time'] |
|
node.setmocktime(entry_time) |
|
|
|
# Let half of the timeout elapse and broadcast the child transaction spending the parent transaction. |
|
half_expiry_time = entry_time + int(60 * 60 * timeout/2) |
|
node.setmocktime(half_expiry_time) |
|
child_txid = self.wallet.send_self_transfer(from_node=node, utxo_to_spend=parent_utxo)['txid'] |
|
assert_equal(parent_txid, node.getmempoolentry(child_txid)['depends'][0]) |
|
self.log.info('Broadcast child transaction after {} hours.'.format( |
|
timedelta(seconds=(half_expiry_time-entry_time)))) |
|
|
|
# Broadcast another (independent) transaction. |
|
independent_txid = self.wallet.send_self_transfer(from_node=node, utxo_to_spend=independent_utxo)['txid'] |
|
|
|
# Let most of the timeout elapse and check that the parent tx is still |
|
# in the mempool. |
|
nearly_expiry_time = entry_time + 60 * 60 * timeout - 5 |
|
node.setmocktime(nearly_expiry_time) |
|
# Broadcast a transaction as the expiry of transactions in the mempool is only checked |
|
# when a new transaction is added to the mempool. |
|
self.wallet.send_self_transfer(from_node=node, utxo_to_spend=trigger_utxo1) |
|
self.log.info('Test parent tx not expired after {} hours.'.format( |
|
timedelta(seconds=(nearly_expiry_time-entry_time)))) |
|
assert_equal(entry_time, node.getmempoolentry(parent_txid)['time']) |
|
|
|
# Transaction should be evicted from the mempool after the expiry time |
|
# has passed. |
|
expiry_time = entry_time + 60 * 60 * timeout + 5 |
|
node.setmocktime(expiry_time) |
|
# Again, broadcast a transaction so the expiry of transactions in the mempool is checked. |
|
self.wallet.send_self_transfer(from_node=node, utxo_to_spend=trigger_utxo2) |
|
self.log.info('Test parent tx expiry after {} hours.'.format( |
|
timedelta(seconds=(expiry_time-entry_time)))) |
|
assert_raises_rpc_error(-5, 'Transaction not in mempool', |
|
node.getmempoolentry, parent_txid) |
|
|
|
# The child transaction should be removed from the mempool as well. |
|
self.log.info('Test child tx is evicted as well.') |
|
assert_raises_rpc_error(-5, 'Transaction not in mempool', |
|
node.getmempoolentry, child_txid) |
|
|
|
# Check that the independent tx is still in the mempool. |
|
self.log.info('Test the independent tx not expired after {} hours.'.format( |
|
timedelta(seconds=(expiry_time-half_expiry_time)))) |
|
assert_equal(half_expiry_time, node.getmempoolentry(independent_txid)['time']) |
|
|
|
def run_test(self): |
|
self.log.info('Test default mempool expiry timeout of %d hours.' % |
|
DEFAULT_MEMPOOL_EXPIRY) |
|
self.test_transaction_expiry(DEFAULT_MEMPOOL_EXPIRY) |
|
|
|
self.log.info('Test custom mempool expiry timeout of %d hours.' % |
|
CUSTOM_MEMPOOL_EXPIRY) |
|
self.restart_node(0, ['-mempoolexpiry=%d' % CUSTOM_MEMPOOL_EXPIRY]) |
|
self.test_transaction_expiry(CUSTOM_MEMPOOL_EXPIRY) |
|
|
|
|
|
if __name__ == '__main__': |
|
MempoolExpiryTest().main()
|
|
|