You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
261 lines
14 KiB
261 lines
14 KiB
#!/usr/bin/env python3 |
|
# Copyright (c) 2017-2020 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 bitcoin-cli""" |
|
|
|
from decimal import Decimal |
|
|
|
from test_framework.blocktools import COINBASE_MATURITY |
|
from test_framework.test_framework import BitcoinTestFramework |
|
from test_framework.util import ( |
|
assert_equal, |
|
assert_greater_than_or_equal, |
|
assert_raises_process_error, |
|
assert_raises_rpc_error, |
|
get_auth_cookie, |
|
) |
|
import time |
|
|
|
# The block reward of coinbaseoutput.nValue (50) BTC/block matures after |
|
# COINBASE_MATURITY (100) blocks. Therefore, after mining 101 blocks we expect |
|
# node 0 to have a balance of (BLOCKS - COINBASE_MATURITY) * 50 BTC/block. |
|
BLOCKS = COINBASE_MATURITY + 1 |
|
BALANCE = (BLOCKS - 100) * 50 |
|
|
|
JSON_PARSING_ERROR = 'error: Error parsing JSON: foo' |
|
BLOCKS_VALUE_OF_ZERO = 'error: the first argument (number of blocks to generate, default: 1) must be an integer value greater than zero' |
|
TOO_MANY_ARGS = 'error: too many arguments (maximum 2 for nblocks and maxtries)' |
|
WALLET_NOT_LOADED = 'Requested wallet does not exist or is not loaded' |
|
WALLET_NOT_SPECIFIED = 'Wallet file not specified' |
|
|
|
class TestBitcoinCli(BitcoinTestFramework): |
|
def set_test_params(self): |
|
self.setup_clean_chain = True |
|
self.num_nodes = 1 |
|
if self.is_wallet_compiled(): |
|
self.requires_wallet = True |
|
|
|
def skip_test_if_missing_module(self): |
|
self.skip_if_no_cli() |
|
|
|
def run_test(self): |
|
"""Main test logic""" |
|
self.nodes[0].generate(BLOCKS) |
|
|
|
self.log.info("Compare responses from getblockchaininfo RPC and `bitcoin-cli getblockchaininfo`") |
|
cli_response = self.nodes[0].cli.getblockchaininfo() |
|
rpc_response = self.nodes[0].getblockchaininfo() |
|
assert_equal(cli_response, rpc_response) |
|
|
|
user, password = get_auth_cookie(self.nodes[0].datadir, self.chain) |
|
|
|
self.log.info("Test -stdinrpcpass option") |
|
assert_equal(BLOCKS, self.nodes[0].cli('-rpcuser={}'.format(user), '-stdinrpcpass', input=password).getblockcount()) |
|
assert_raises_process_error(1, 'Incorrect rpcuser or rpcpassword', self.nodes[0].cli('-rpcuser={}'.format(user), '-stdinrpcpass', input='foo').echo) |
|
|
|
self.log.info("Test -stdin and -stdinrpcpass") |
|
assert_equal(['foo', 'bar'], self.nodes[0].cli('-rpcuser={}'.format(user), '-stdin', '-stdinrpcpass', input=password + '\nfoo\nbar').echo()) |
|
assert_raises_process_error(1, 'Incorrect rpcuser or rpcpassword', self.nodes[0].cli('-rpcuser={}'.format(user), '-stdin', '-stdinrpcpass', input='foo').echo) |
|
|
|
self.log.info("Test connecting to a non-existing server") |
|
assert_raises_process_error(1, "Could not connect to the server", self.nodes[0].cli('-rpcport=1').echo) |
|
|
|
self.log.info("Test connecting with non-existing RPC cookie file") |
|
assert_raises_process_error(1, "Could not locate RPC credentials", self.nodes[0].cli('-rpccookiefile=does-not-exist', '-rpcpassword=').echo) |
|
|
|
self.log.info("Test -getinfo with arguments fails") |
|
assert_raises_process_error(1, "-getinfo takes no arguments", self.nodes[0].cli('-getinfo').help) |
|
|
|
self.log.info("Test -getinfo returns expected network and blockchain info") |
|
if self.is_wallet_compiled(): |
|
self.nodes[0].encryptwallet(password) |
|
cli_get_info = self.nodes[0].cli('-getinfo').send_cli() |
|
network_info = self.nodes[0].getnetworkinfo() |
|
blockchain_info = self.nodes[0].getblockchaininfo() |
|
assert_equal(cli_get_info['version'], network_info['version']) |
|
assert_equal(cli_get_info['blocks'], blockchain_info['blocks']) |
|
assert_equal(cli_get_info['headers'], blockchain_info['headers']) |
|
assert_equal(cli_get_info['timeoffset'], network_info['timeoffset']) |
|
assert_equal( |
|
cli_get_info['connections'], |
|
{ |
|
'in': network_info['connections_in'], |
|
'out': network_info['connections_out'], |
|
'total': network_info['connections'] |
|
} |
|
) |
|
assert_equal(cli_get_info['proxy'], network_info['networks'][0]['proxy']) |
|
assert_equal(cli_get_info['difficulty'], blockchain_info['difficulty']) |
|
assert_equal(cli_get_info['chain'], blockchain_info['chain']) |
|
|
|
if self.is_wallet_compiled(): |
|
self.log.info("Test -getinfo and bitcoin-cli getwalletinfo return expected wallet info") |
|
assert_equal(cli_get_info['balance'], BALANCE) |
|
assert 'balances' not in cli_get_info.keys() |
|
wallet_info = self.nodes[0].getwalletinfo() |
|
assert_equal(cli_get_info['keypoolsize'], wallet_info['keypoolsize']) |
|
assert_equal(cli_get_info['unlocked_until'], wallet_info['unlocked_until']) |
|
assert_equal(cli_get_info['paytxfee'], wallet_info['paytxfee']) |
|
assert_equal(cli_get_info['relayfee'], network_info['relayfee']) |
|
assert_equal(self.nodes[0].cli.getwalletinfo(), wallet_info) |
|
|
|
# Setup to test -getinfo, -generate, and -rpcwallet= with multiple wallets. |
|
wallets = [self.default_wallet_name, 'Encrypted', 'secret'] |
|
amounts = [BALANCE + Decimal('9.999928'), Decimal(9), Decimal(31)] |
|
self.nodes[0].createwallet(wallet_name=wallets[1]) |
|
self.nodes[0].createwallet(wallet_name=wallets[2]) |
|
w1 = self.nodes[0].get_wallet_rpc(wallets[0]) |
|
w2 = self.nodes[0].get_wallet_rpc(wallets[1]) |
|
w3 = self.nodes[0].get_wallet_rpc(wallets[2]) |
|
rpcwallet2 = '-rpcwallet={}'.format(wallets[1]) |
|
rpcwallet3 = '-rpcwallet={}'.format(wallets[2]) |
|
w1.walletpassphrase(password, self.rpc_timeout) |
|
w2.encryptwallet(password) |
|
w1.sendtoaddress(w2.getnewaddress(), amounts[1]) |
|
w1.sendtoaddress(w3.getnewaddress(), amounts[2]) |
|
|
|
# Mine a block to confirm; adds a block reward (50 BTC) to the default wallet. |
|
self.nodes[0].generate(1) |
|
|
|
self.log.info("Test -getinfo with multiple wallets and -rpcwallet returns specified wallet balance") |
|
for i in range(len(wallets)): |
|
cli_get_info = self.nodes[0].cli('-getinfo', '-rpcwallet={}'.format(wallets[i])).send_cli() |
|
assert 'balances' not in cli_get_info.keys() |
|
assert_equal(cli_get_info['balance'], amounts[i]) |
|
|
|
self.log.info("Test -getinfo with multiple wallets and -rpcwallet=non-existing-wallet returns no balances") |
|
cli_get_info_keys = self.nodes[0].cli('-getinfo', '-rpcwallet=does-not-exist').send_cli().keys() |
|
assert 'balance' not in cli_get_info_keys |
|
assert 'balances' not in cli_get_info_keys |
|
|
|
self.log.info("Test -getinfo with multiple wallets returns all loaded wallet names and balances") |
|
assert_equal(set(self.nodes[0].listwallets()), set(wallets)) |
|
cli_get_info = self.nodes[0].cli('-getinfo').send_cli() |
|
assert 'balance' not in cli_get_info.keys() |
|
assert_equal(cli_get_info['balances'], {k: v for k, v in zip(wallets, amounts)}) |
|
|
|
# Unload the default wallet and re-verify. |
|
self.nodes[0].unloadwallet(wallets[0]) |
|
assert wallets[0] not in self.nodes[0].listwallets() |
|
cli_get_info = self.nodes[0].cli('-getinfo').send_cli() |
|
assert 'balance' not in cli_get_info.keys() |
|
assert_equal(cli_get_info['balances'], {k: v for k, v in zip(wallets[1:], amounts[1:])}) |
|
|
|
self.log.info("Test -getinfo after unloading all wallets except a non-default one returns its balance") |
|
self.nodes[0].unloadwallet(wallets[2]) |
|
assert_equal(self.nodes[0].listwallets(), [wallets[1]]) |
|
cli_get_info = self.nodes[0].cli('-getinfo').send_cli() |
|
assert 'balances' not in cli_get_info.keys() |
|
assert_equal(cli_get_info['balance'], amounts[1]) |
|
|
|
self.log.info("Test -getinfo with -rpcwallet=remaining-non-default-wallet returns only its balance") |
|
cli_get_info = self.nodes[0].cli('-getinfo', rpcwallet2).send_cli() |
|
assert 'balances' not in cli_get_info.keys() |
|
assert_equal(cli_get_info['balance'], amounts[1]) |
|
|
|
self.log.info("Test -getinfo with -rpcwallet=unloaded wallet returns no balances") |
|
cli_get_info_keys = self.nodes[0].cli('-getinfo', rpcwallet3).send_cli().keys() |
|
assert 'balance' not in cli_get_info_keys |
|
assert 'balances' not in cli_get_info_keys |
|
|
|
# Test bitcoin-cli -generate. |
|
n1 = 3 |
|
n2 = 4 |
|
w2.walletpassphrase(password, self.rpc_timeout) |
|
blocks = self.nodes[0].getblockcount() |
|
|
|
self.log.info('Test -generate with no args') |
|
generate = self.nodes[0].cli('-generate').send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), 1) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 1) |
|
|
|
self.log.info('Test -generate with bad args') |
|
assert_raises_process_error(1, JSON_PARSING_ERROR, self.nodes[0].cli('-generate', 'foo').echo) |
|
assert_raises_process_error(1, BLOCKS_VALUE_OF_ZERO, self.nodes[0].cli('-generate', 0).echo) |
|
assert_raises_process_error(1, TOO_MANY_ARGS, self.nodes[0].cli('-generate', 1, 2, 3).echo) |
|
|
|
self.log.info('Test -generate with nblocks') |
|
generate = self.nodes[0].cli('-generate', n1).send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), n1) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 1 + n1) |
|
|
|
self.log.info('Test -generate with nblocks and maxtries') |
|
generate = self.nodes[0].cli('-generate', n2, 1000000).send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), n2) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 1 + n1 + n2) |
|
|
|
self.log.info('Test -generate -rpcwallet in single-wallet mode') |
|
generate = self.nodes[0].cli(rpcwallet2, '-generate').send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), 1) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 2 + n1 + n2) |
|
|
|
self.log.info('Test -generate -rpcwallet=unloaded wallet raises RPC error') |
|
assert_raises_rpc_error(-18, WALLET_NOT_LOADED, self.nodes[0].cli(rpcwallet3, '-generate').echo) |
|
assert_raises_rpc_error(-18, WALLET_NOT_LOADED, self.nodes[0].cli(rpcwallet3, '-generate', 'foo').echo) |
|
assert_raises_rpc_error(-18, WALLET_NOT_LOADED, self.nodes[0].cli(rpcwallet3, '-generate', 0).echo) |
|
assert_raises_rpc_error(-18, WALLET_NOT_LOADED, self.nodes[0].cli(rpcwallet3, '-generate', 1, 2, 3).echo) |
|
|
|
# Test bitcoin-cli -generate with -rpcwallet in multiwallet mode. |
|
self.nodes[0].loadwallet(wallets[2]) |
|
n3 = 4 |
|
n4 = 10 |
|
blocks = self.nodes[0].getblockcount() |
|
|
|
self.log.info('Test -generate -rpcwallet with no args') |
|
generate = self.nodes[0].cli(rpcwallet2, '-generate').send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), 1) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 1) |
|
|
|
self.log.info('Test -generate -rpcwallet with bad args') |
|
assert_raises_process_error(1, JSON_PARSING_ERROR, self.nodes[0].cli(rpcwallet2, '-generate', 'foo').echo) |
|
assert_raises_process_error(1, BLOCKS_VALUE_OF_ZERO, self.nodes[0].cli(rpcwallet2, '-generate', 0).echo) |
|
assert_raises_process_error(1, TOO_MANY_ARGS, self.nodes[0].cli(rpcwallet2, '-generate', 1, 2, 3).echo) |
|
|
|
self.log.info('Test -generate -rpcwallet with nblocks') |
|
generate = self.nodes[0].cli(rpcwallet2, '-generate', n3).send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), n3) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 1 + n3) |
|
|
|
self.log.info('Test -generate -rpcwallet with nblocks and maxtries') |
|
generate = self.nodes[0].cli(rpcwallet2, '-generate', n4, 1000000).send_cli() |
|
assert_equal(set(generate.keys()), {'address', 'blocks'}) |
|
assert_equal(len(generate["blocks"]), n4) |
|
assert_equal(self.nodes[0].getblockcount(), blocks + 1 + n3 + n4) |
|
|
|
self.log.info('Test -generate without -rpcwallet in multiwallet mode raises RPC error') |
|
assert_raises_rpc_error(-19, WALLET_NOT_SPECIFIED, self.nodes[0].cli('-generate').echo) |
|
assert_raises_rpc_error(-19, WALLET_NOT_SPECIFIED, self.nodes[0].cli('-generate', 'foo').echo) |
|
assert_raises_rpc_error(-19, WALLET_NOT_SPECIFIED, self.nodes[0].cli('-generate', 0).echo) |
|
assert_raises_rpc_error(-19, WALLET_NOT_SPECIFIED, self.nodes[0].cli('-generate', 1, 2, 3).echo) |
|
else: |
|
self.log.info("*** Wallet not compiled; cli getwalletinfo and -getinfo wallet tests skipped") |
|
self.nodes[0].generate(25) # maintain block parity with the wallet_compiled conditional branch |
|
|
|
self.log.info("Test -version with node stopped") |
|
self.stop_node(0) |
|
cli_response = self.nodes[0].cli('-version').send_cli() |
|
assert "{} RPC client version".format(self.config['environment']['PACKAGE_NAME']) in cli_response |
|
|
|
self.log.info("Test -rpcwait option successfully waits for RPC connection") |
|
self.nodes[0].start() # start node without RPC connection |
|
self.nodes[0].wait_for_cookie_credentials() # ensure cookie file is available to avoid race condition |
|
blocks = self.nodes[0].cli('-rpcwait').send_cli('getblockcount') |
|
self.nodes[0].wait_for_rpc_connection() |
|
assert_equal(blocks, BLOCKS + 25) |
|
|
|
self.log.info("Test -rpcwait option waits at most -rpcwaittimeout seconds for startup") |
|
self.stop_node(0) # stop the node so we time out |
|
start_time = time.time() |
|
assert_raises_process_error(1, "Could not connect to the server", self.nodes[0].cli('-rpcwait', '-rpcwaittimeout=5').echo) |
|
assert_greater_than_or_equal(time.time(), start_time + 5) |
|
|
|
|
|
if __name__ == '__main__': |
|
TestBitcoinCli().main()
|
|
|