mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-12-13 20:36:21 +01:00
53 lines
2.1 KiB
Python
Executable File
53 lines
2.1 KiB
Python
Executable File
#!/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 contextlib import redirect_stdout
|
|
from decimal import Decimal
|
|
from io import StringIO
|
|
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
|
|
stdout_buf = StringIO()
|
|
with redirect_stdout(stdout_buf):
|
|
test2 = TestShell().setup()
|
|
assert test2 is None
|
|
assert_equal(stdout_buf.getvalue().rstrip(), "TestShell is already running!")
|
|
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()
|
|
test.reset()
|
|
assert test.num_nodes is None
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_testshell_doc_example(str(Path(__file__).parent))
|