mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-04-29 14:59:39 -04:00
0d2eefca8b
Add new TestNode.binaries object to manage paths to bitcoin binaries. Having this object makes it possible for the test framework to exercise the bitcoin wrapper executable introduced in https://github.com/bitcoin/bitcoin/pull/31375 and also makes it easier to add new binaries and options and environment variables controlling how they are invoked, because logic for invoking them that was previously spread out is now consolidated in one place. Co-authored-by: Sjors Provoost <sjors@sprovoost.nl>
69 lines
2.4 KiB
Python
Executable file
69 lines
2.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright (c) 2022 The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
"""Test signet miner tool"""
|
|
|
|
import os.path
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
from test_framework.blocktools import DIFF_1_N_BITS
|
|
from test_framework.key import ECKey
|
|
from test_framework.script_util import key_to_p2wpkh_script
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import assert_equal
|
|
from test_framework.wallet_util import bytes_to_wif
|
|
|
|
|
|
CHALLENGE_PRIVATE_KEY = (42).to_bytes(32, 'big')
|
|
|
|
|
|
class SignetMinerTest(BitcoinTestFramework):
|
|
def add_options(self, parser):
|
|
self.add_wallet_options(parser)
|
|
|
|
def set_test_params(self):
|
|
self.chain = "signet"
|
|
self.setup_clean_chain = True
|
|
self.num_nodes = 1
|
|
|
|
# generate and specify signet challenge (simple p2wpkh script)
|
|
privkey = ECKey()
|
|
privkey.set(CHALLENGE_PRIVATE_KEY, True)
|
|
pubkey = privkey.get_pubkey().get_bytes()
|
|
challenge = key_to_p2wpkh_script(pubkey)
|
|
self.extra_args = [[f'-signetchallenge={challenge.hex()}']]
|
|
|
|
def skip_test_if_missing_module(self):
|
|
self.skip_if_no_cli()
|
|
self.skip_if_no_wallet()
|
|
self.skip_if_no_bitcoin_util()
|
|
|
|
def run_test(self):
|
|
node = self.nodes[0]
|
|
# import private key needed for signing block
|
|
node.importprivkey(bytes_to_wif(CHALLENGE_PRIVATE_KEY))
|
|
|
|
# generate block with signet miner tool
|
|
base_dir = self.config["environment"]["SRCDIR"]
|
|
signet_miner_path = os.path.join(base_dir, "contrib", "signet", "miner")
|
|
rpc_argv = node.binaries.rpc_argv() + [f"-datadir={node.cli.datadir}"]
|
|
util_argv = node.binaries.util_argv() + ["grind"]
|
|
subprocess.run([
|
|
sys.executable,
|
|
signet_miner_path,
|
|
f'--cli={" ".join(rpc_argv)}',
|
|
'generate',
|
|
f'--address={node.getnewaddress()}',
|
|
f'--grind-cmd={" ".join(util_argv)}',
|
|
f'--nbits={DIFF_1_N_BITS:08x}',
|
|
f'--set-block-time={int(time.time())}',
|
|
'--poolnum=99',
|
|
], check=True, stderr=subprocess.STDOUT)
|
|
assert_equal(node.getblockcount(), 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
SignetMinerTest(__file__).main()
|