mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-04-29 14:59:39 -04:00
test: cover -rpcservertimeout
Testing this requires adding an option to TestNode to force the test framework to establish a new HTTP connection for every RPC. Otherwise, attempting to reuse a persistent connection would cause framework RPCs during startup and shutdown to fail.
This commit is contained in:
parent
00d18bbe09
commit
5ec06529bb
3 changed files with 58 additions and 1 deletions
|
@ -8,6 +8,7 @@ from test_framework.test_framework import BitcoinTestFramework
|
|||
from test_framework.util import assert_equal, str_to_b64str
|
||||
|
||||
import http.client
|
||||
import time
|
||||
import urllib.parse
|
||||
|
||||
class HTTPBasicsTest (BitcoinTestFramework):
|
||||
|
@ -124,7 +125,6 @@ class HTTPBasicsTest (BitcoinTestFramework):
|
|||
req2 = req
|
||||
req2 += f'Content-Length: {len(body2)}\r\n\r\n'
|
||||
req2 += body2
|
||||
|
||||
# Get the underlying socket from HTTP connection so we can send something unusual
|
||||
conn = http.client.HTTPConnection(urlNode2.hostname, urlNode2.port)
|
||||
conn.connect()
|
||||
|
@ -155,5 +155,55 @@ class HTTPBasicsTest (BitcoinTestFramework):
|
|||
assert chunks[1].startswith(b'{"hash":')
|
||||
assert chunks[2].startswith(bytes(f'{tip_height + 1}', 'utf8'))
|
||||
|
||||
|
||||
self.log.info("Check -rpcservertimeout")
|
||||
# The test framework typically reuses a single persistent HTTP connection
|
||||
# for all RPCs to a TestNode. Because we are setting -rpcservertimeout
|
||||
# so low on this one node, its connection will quickly timeout and get dropped by
|
||||
# the server. Negating this setting will force the AuthServiceProxy
|
||||
# for this node to create a fresh new HTTP connection for every command
|
||||
# called for the remainder of this test.
|
||||
self.nodes[2].reuse_http_connections = False
|
||||
|
||||
self.restart_node(2, extra_args=["-rpcservertimeout=1"])
|
||||
# This is the amount of time the server will wait for a client to
|
||||
# send a complete request. Test it by sending an incomplete but
|
||||
# so-far otherwise well-formed HTTP request, and never finishing it.
|
||||
|
||||
# Copied from http_incomplete_test_() in regress_http.c in libevent.
|
||||
# A complete request would have an additional "\r\n" at the end.
|
||||
http_request = "GET /test1 HTTP/1.1\r\nHost: somehost\r\n"
|
||||
|
||||
# Get the underlying socket from HTTP connection so we can send something unusual
|
||||
conn = http.client.HTTPConnection(urlNode2.hostname, urlNode2.port)
|
||||
conn.connect()
|
||||
sock = conn.sock
|
||||
sock.sendall(http_request.encode("utf-8"))
|
||||
# Wait for response, but expect a timeout disconnection after 1 second
|
||||
start = time.time()
|
||||
res = sock.recv(1024)
|
||||
stop = time.time()
|
||||
assert res == b""
|
||||
assert stop - start >= 1
|
||||
# definitely closed
|
||||
try:
|
||||
conn.request('GET', '/')
|
||||
conn.getresponse()
|
||||
# macos/linux windows
|
||||
except (ConnectionResetError, ConnectionAbortedError):
|
||||
pass
|
||||
|
||||
# Sanity check
|
||||
http_request = "GET /test2 HTTP/1.1\r\nHost: somehost\r\n\r\n"
|
||||
conn = http.client.HTTPConnection(urlNode2.hostname, urlNode2.port)
|
||||
conn.connect()
|
||||
sock = conn.sock
|
||||
sock.sendall(http_request.encode("utf-8"))
|
||||
res = sock.recv(1024)
|
||||
assert res.startswith(b"HTTP/1.1 404 Not Found")
|
||||
# still open
|
||||
conn.request('GET', '/')
|
||||
conn.getresponse()
|
||||
|
||||
if __name__ == '__main__':
|
||||
HTTPBasicsTest(__file__).main()
|
||||
|
|
|
@ -75,6 +75,7 @@ class AuthServiceProxy():
|
|||
self.__service_url = service_url
|
||||
self._service_name = service_name
|
||||
self.ensure_ascii = ensure_ascii # can be toggled on the fly by tests
|
||||
self.reuse_http_connections = True
|
||||
self.__url = urllib.parse.urlparse(service_url)
|
||||
user = None if self.__url.username is None else self.__url.username.encode('utf8')
|
||||
passwd = None if self.__url.password is None else self.__url.password.encode('utf8')
|
||||
|
@ -92,6 +93,8 @@ class AuthServiceProxy():
|
|||
raise AttributeError
|
||||
if self._service_name is not None:
|
||||
name = "%s.%s" % (self._service_name, name)
|
||||
if not self.reuse_http_connections:
|
||||
self._set_conn()
|
||||
return AuthServiceProxy(self.__service_url, name, connection=self.__conn)
|
||||
|
||||
def _request(self, method, path, postdata):
|
||||
|
@ -102,6 +105,8 @@ class AuthServiceProxy():
|
|||
'User-Agent': USER_AGENT,
|
||||
'Authorization': self.__auth_header,
|
||||
'Content-type': 'application/json'}
|
||||
if not self.reuse_http_connections:
|
||||
self._set_conn()
|
||||
self.__conn.request(method, path, postdata, headers)
|
||||
return self._get_response()
|
||||
|
||||
|
|
|
@ -157,6 +157,7 @@ class TestNode():
|
|||
self.process = None
|
||||
self.rpc_connected = False
|
||||
self.rpc = None
|
||||
self.reuse_http_connections = True # Must be set before calling get_rpc_proxy() i.e. before restarting node
|
||||
self.url = None
|
||||
self.log = logging.getLogger('TestFramework.node%d' % i)
|
||||
# Cache perf subprocesses here by their data output filename.
|
||||
|
@ -281,6 +282,7 @@ class TestNode():
|
|||
timeout=self.rpc_timeout // 2, # Shorter timeout to allow for one retry in case of ETIMEDOUT
|
||||
coveragedir=self.coverage_dir,
|
||||
)
|
||||
rpc.auth_service_proxy_instance.reuse_http_connections = self.reuse_http_connections
|
||||
rpc.getblockcount()
|
||||
# If the call to getblockcount() succeeds then the RPC connection is up
|
||||
if self.version_is_at_least(190000) and wait_for_import:
|
||||
|
|
Loading…
Add table
Reference in a new issue