Skip to content

Commit fac62e6

Browse files
author
MarcoFalke
committed
test: Delete generate* calls from TestNode
1 parent fac7f61 commit fac62e6

File tree

3 files changed

+20
-8
lines changed

3 files changed

+20
-8
lines changed

test/functional/test_framework/test_framework.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -628,19 +628,19 @@ def join_network(self):
628628
self.sync_all()
629629

630630
def generate(self, generator, *args, **kwargs):
631-
blocks = generator.generate(*args, **kwargs)
631+
blocks = generator.generate(*args, invalid_call=False, **kwargs)
632632
return blocks
633633

634634
def generateblock(self, generator, *args, **kwargs):
635-
blocks = generator.generateblock(*args, **kwargs)
635+
blocks = generator.generateblock(*args, invalid_call=False, **kwargs)
636636
return blocks
637637

638638
def generatetoaddress(self, generator, *args, **kwargs):
639-
blocks = generator.generatetoaddress(*args, **kwargs)
639+
blocks = generator.generatetoaddress(*args, invalid_call=False, **kwargs)
640640
return blocks
641641

642642
def generatetodescriptor(self, generator, *args, **kwargs):
643-
blocks = generator.generatetodescriptor(*args, **kwargs)
643+
blocks = generator.generatetodescriptor(*args, invalid_call=False, **kwargs)
644644
return blocks
645645

646646
def sync_blocks(self, nodes=None, wait=1, timeout=60):

test/functional/test_framework/test_node.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -297,9 +297,21 @@ def wait_for_cookie_credentials(self):
297297
time.sleep(1.0 / poll_per_s)
298298
self._raise_assertion_error("Unable to retrieve cookie credentials after {}s".format(self.rpc_timeout))
299299

300-
def generate(self, nblocks, maxtries=1000000):
300+
def generate(self, nblocks, maxtries=1000000, **kwargs):
301301
self.log.debug("TestNode.generate() dispatches `generate` call to `generatetoaddress`")
302-
return self.generatetoaddress(nblocks=nblocks, address=self.get_deterministic_priv_key().address, maxtries=maxtries)
302+
return self.generatetoaddress(nblocks=nblocks, address=self.get_deterministic_priv_key().address, maxtries=maxtries, **kwargs)
303+
304+
def generateblock(self, *args, invalid_call, **kwargs):
305+
assert not invalid_call
306+
return self.__getattr__('generateblock')(*args, **kwargs)
307+
308+
def generatetoaddress(self, *args, invalid_call, **kwargs):
309+
assert not invalid_call
310+
return self.__getattr__('generatetoaddress')(*args, **kwargs)
311+
312+
def generatetodescriptor(self, *args, invalid_call, **kwargs):
313+
assert not invalid_call
314+
return self.__getattr__('generatetodescriptor')(*args, **kwargs)
303315

304316
def get_wallet_rpc(self, wallet_name):
305317
if self.use_cli:

test/functional/test_framework/wallet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,9 +109,9 @@ def sign_tx(self, tx, fixed_length=True):
109109
break
110110
tx.vin[0].scriptSig = CScript([der_sig + bytes(bytearray([SIGHASH_ALL]))])
111111

112-
def generate(self, num_blocks):
112+
def generate(self, num_blocks, **kwargs):
113113
"""Generate blocks with coinbase outputs to the internal address, and append the outputs to the internal list"""
114-
blocks = self._test_node.generatetodescriptor(num_blocks, self.get_descriptor())
114+
blocks = self._test_node.generatetodescriptor(num_blocks, self.get_descriptor(), **kwargs)
115115
for b in blocks:
116116
cb_tx = self._test_node.getblock(blockhash=b, verbosity=2)['tx'][0]
117117
self._utxos.append({'txid': cb_tx['txid'], 'vout': 0, 'value': cb_tx['vout'][0]['value']})

0 commit comments

Comments
 (0)