mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-09-11 21:20:39 +02:00
test: Check that RPCs do not time out, even under load
Also, modify send_cli, so that the test can be run under --usecli
This commit is contained in:
50
test/functional/rpc_echo_payload.py
Executable file
50
test/functional/rpc_echo_payload.py
Executable file
@@ -0,0 +1,50 @@
|
||||
#!/usr/bin/env python3
|
||||
# Copyright (c) The Bitcoin Core developers
|
||||
# Distributed under the MIT software license, see the accompanying
|
||||
# file COPYING or https://opensource.org/license/mit/.
|
||||
"""Ensure RPCs with a (possibly large) payload will either be rejected or handled, but will never time out."""
|
||||
|
||||
from concurrent.futures import ThreadPoolExecutor
|
||||
from test_framework.test_framework import BitcoinTestFramework
|
||||
from test_framework.util import assert_equal, JSONRPCException
|
||||
import random
|
||||
|
||||
|
||||
class RpcEchoPayloadTest(BitcoinTestFramework):
|
||||
def set_test_params(self):
|
||||
self.num_nodes = 1
|
||||
self.setup_clean_chain = True
|
||||
self.extra_args = [["-rpcworkqueue=2", "-rpcthreads=2"]]
|
||||
# enough to possibly fill the running threads as well as the queue:
|
||||
self.num_threads = 6
|
||||
|
||||
def run_test(self):
|
||||
node = self.nodes[0]
|
||||
# Use json-serializable, but non-hex data
|
||||
data = "z" + random.randbytes(1_999_000).hex()
|
||||
|
||||
def check_results(rpc):
|
||||
self.log.info("Starting thread ...")
|
||||
for i in range(200):
|
||||
payload = data[: random.randrange(0, len(data))]
|
||||
try:
|
||||
if random.getrandbits(1):
|
||||
assert_equal(payload, rpc.echo(payload)[0])
|
||||
else:
|
||||
rpc.sendrawtransaction(payload)
|
||||
except JSONRPCException as e:
|
||||
msg = e.error["message"]
|
||||
if msg not in [
|
||||
"TX decode failed. Make sure the tx has at least one input.",
|
||||
"non-JSON HTTP response with '503 Service Unavailable' from server: Work queue depth exceeded",
|
||||
]:
|
||||
raise AssertionError(f"Unexpected msg: {msg}")
|
||||
|
||||
rpcs = [node.create_new_rpc_connection() for _ in range(self.num_threads)]
|
||||
self.log.info("Starting threadpool ...")
|
||||
with ThreadPoolExecutor(max_workers=len(rpcs)) as threads:
|
||||
list(threads.map(check_results, rpcs))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
RpcEchoPayloadTest(__file__).main()
|
||||
@@ -153,6 +153,7 @@ BASE_SCRIPTS = [
|
||||
# vv Tests less than 30s vv
|
||||
'wallet_deprecated_rbf.py',
|
||||
'p2p_invalid_messages.py',
|
||||
'rpc_echo_payload.py',
|
||||
'rpc_createmultisig.py',
|
||||
'p2p_timeouts.py --v1transport',
|
||||
'p2p_timeouts.py --v2transport',
|
||||
|
||||
Reference in New Issue
Block a user