highperfocused/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-05-23 18:32:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

test: use MiniWallet for feature_maxuploadtarget.py

Browse Source 
This test can now be run even with the Bitcoin Core wallet disabled.
This commit is contained in:
Sebastian Falbesoner 2022-03-11 16:09:54 +01:00
parent 23e8c702bc
commit aab552fa30
2 changed files with 23 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
test/functional/feature_maxuploadtarget.py
View File

@ -13,10 +13,19 @@ if uploadtarget has been reached.
from collections import defaultdict from collections import defaultdict
import time import time
from test_framework.messages import CInv, MSG_BLOCK, msg_getdata from test_framework.messages import (
CInv,
MSG_BLOCK,
msg_getdata,
)
from test_framework.p2p import P2PInterface from test_framework.p2p import P2PInterface
from test_framework.test_framework import BitcoinTestFramework from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal, mine_large_block from test_framework.util import (
assert_equal,
mine_large_block,
)
from test_framework.wallet import MiniWallet
class TestP2PConn(P2PInterface): class TestP2PConn(P2PInterface):
def __init__(self): def __init__(self):
@ -41,12 +50,6 @@ class MaxUploadTest(BitcoinTestFramework):
]] ]]
self.supports_cli = False self.supports_cli = False
# Cache for utxos, as the listunspent may take a long time later in the test
self.utxo_cache = []
def skip_test_if_missing_module(self):
self.skip_if_no_wallet()
def run_test(self): def run_test(self):
# Before we connect anything, we first set the time on the node # Before we connect anything, we first set the time on the node
# to be in the past, otherwise things break because the CNode # to be in the past, otherwise things break because the CNode
@ -55,7 +58,8 @@ class MaxUploadTest(BitcoinTestFramework):
self.nodes[0].setmocktime(old_time) self.nodes[0].setmocktime(old_time)
# Generate some old blocks # Generate some old blocks
self.generate(self.nodes[0], 130) self.wallet = MiniWallet(self.nodes[0])
self.generate(self.wallet, 130)
# p2p_conns[0] will only request old blocks # p2p_conns[0] will only request old blocks
# p2p_conns[1] will only request new blocks # p2p_conns[1] will only request new blocks
@ -66,7 +70,7 @@ class MaxUploadTest(BitcoinTestFramework):
p2p_conns.append(self.nodes[0].add_p2p_connection(TestP2PConn())) p2p_conns.append(self.nodes[0].add_p2p_connection(TestP2PConn()))
# Now mine a big block # Now mine a big block
mine_large_block(self, self.nodes[0], self.utxo_cache) mine_large_block(self, self.wallet, self.nodes[0])
# Store the hash; we'll request this later # Store the hash; we'll request this later
big_old_block = self.nodes[0].getbestblockhash() big_old_block = self.nodes[0].getbestblockhash()
@ -77,7 +81,7 @@ class MaxUploadTest(BitcoinTestFramework):
self.nodes[0].setmocktime(int(time.time()) - 2*60*60*24) self.nodes[0].setmocktime(int(time.time()) - 2*60*60*24)
# Mine one more block, so that the prior block looks old # Mine one more block, so that the prior block looks old
mine_large_block(self, self.nodes[0], self.utxo_cache) mine_large_block(self, self.wallet, self.nodes[0])
# We'll be requesting this new block too # We'll be requesting this new block too
big_new_block = self.nodes[0].getbestblockhash() big_new_block = self.nodes[0].getbestblockhash()

16
test/functional/test_framework/util.py
View File

@ -573,17 +573,17 @@ def create_lots_of_big_transactions(node, txouts, utxos, num, fee):
return txids return txids
def mine_large_block(test_framework, node, utxos=None): def mine_large_block(test_framework, mini_wallet, node):
# generate a 66k transaction, # generate a 66k transaction,
# and 14 of them is close to the 1MB block limit # and 14 of them is close to the 1MB block limit
num = 14
txouts = gen_return_txouts() txouts = gen_return_txouts()
utxos = utxos if utxos is not None else [] from .messages import COIN
if len(utxos) < num: fee = 100 * int(node.getnetworkinfo()["relayfee"] * COIN)
utxos.clear() for _ in range(14):
utxos.extend(node.listunspent()) tx = mini_wallet.create_self_transfer(from_node=node, fee_rate=0, mempool_valid=False)['tx']
fee = 100 * node.getnetworkinfo()["relayfee"] tx.vout[0].nValue -= fee
create_lots_of_big_transactions(node, txouts, utxos, num, fee=fee) tx.vout.extend(txouts)
mini_wallet.sendrawtransaction(from_node=node, tx_hex=tx.serialize().hex())
test_framework.generate(node, 1) test_framework.generate(node, 1)