mirror of https://github.com/bitcoin/bitcoin.git
test: add functional test for `TestShell` (matching doc example)
This commit is contained in:
parent
5ea4bb112e
commit
9babfb9331
|
@ -0,0 +1,45 @@
|
|||
#!/usr/bin/env python3
|
||||
# Copyright (c) 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 for the `TestShell` submodule."""
|
||||
|
||||
from decimal import Decimal
|
||||
from pathlib import Path
|
||||
|
||||
# Note that we need to import from functional test framework modules
|
||||
# *after* extending the Python path via sys.path.insert(0, ...) below,
|
||||
# in order to work with the full symlinked (unresolved) path within the
|
||||
# build directory (usually ./build/test/functional).
|
||||
|
||||
|
||||
# Test matching the minimal example from the documentation. Should be kept
|
||||
# in sync with the interactive shell instructions ('>>> ') in test-shell.md.
|
||||
def run_testshell_doc_example(functional_tests_dir):
|
||||
import sys
|
||||
sys.path.insert(0, functional_tests_dir)
|
||||
from test_framework.test_shell import TestShell
|
||||
from test_framework.util import assert_equal
|
||||
|
||||
test = TestShell().setup(num_nodes=2, setup_clean_chain=True)
|
||||
try:
|
||||
assert test is not None
|
||||
test2 = TestShell().setup()
|
||||
assert test2 is None # TODO: check for "TestShell is already running!" output to stdout
|
||||
assert_equal(test.nodes[0].getblockchaininfo()["blocks"], 0)
|
||||
if test.is_wallet_compiled():
|
||||
res = test.nodes[0].createwallet('default')
|
||||
assert_equal(res, {'name': 'default'})
|
||||
address = test.nodes[0].getnewaddress()
|
||||
res = test.generatetoaddress(test.nodes[0], 101, address)
|
||||
assert_equal(len(res), 101)
|
||||
test.sync_blocks()
|
||||
assert_equal(test.nodes[1].getblockchaininfo()["blocks"], 101)
|
||||
assert_equal(test.nodes[0].getbalance(), Decimal('50.0'))
|
||||
test.nodes[0].log.info("Successfully mined regtest chain!")
|
||||
finally:
|
||||
test.shutdown()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
run_testshell_doc_example(str(Path(__file__).parent))
|
|
@ -360,6 +360,7 @@ BASE_SCRIPTS = [
|
|||
'rpc_getdescriptorinfo.py',
|
||||
'rpc_mempool_info.py',
|
||||
'rpc_help.py',
|
||||
'feature_framework_testshell.py',
|
||||
'tool_rpcauth.py',
|
||||
'p2p_handshake.py',
|
||||
'p2p_handshake.py --v2transport',
|
||||
|
|
Loading…
Reference in New Issue