DanS
4cc7e0491a
Follow-up to5e0a70683, which got start_node() working. Three more defects sat behind it: - initialize_chain() builds the 4-node cache with its own daemon invocation, which5e0a70683did not touch. It therefore still omitted -regtest (so those cache nodes ran on MAINNET) and -asmap (so they refused to start at all). Every test that uses the cache -- which is most of the wallet suite -- died there. - reindex.py and getblocktemplate_longpoll.py each carried a single python2 print statement, which is the whole reason they would not even parse under python3. Shebangs updated to match. The suite still does not pass: initialize_chain hits a remaining py2 str+int concatenation, and getblocktemplate.py reaches a real test assertion. Both are beyond this commit, but the harness now gets far enough to start nodes, answer RPC and begin building the shared chain, which it could not do before.
95 lines
3.7 KiB
Python
Executable File
95 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2016-2024 The Hush developers
|
|
# Copyright (c) 2014 The Bitcoin Core developers
|
|
# Distributed under the GPLv3 software license, see the accompanying
|
|
# file COPYING or https://www.gnu.org/licenses/gpl-3.0.en.html
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.authproxy import AuthServiceProxy
|
|
from test_framework.util import random_transaction
|
|
|
|
from decimal import Decimal
|
|
|
|
def check_array_result(object_array, to_match, expected):
|
|
"""
|
|
Pass in array of JSON objects, a dictionary with key/value pairs
|
|
to match against, and another dictionary with expected key/value
|
|
pairs.
|
|
"""
|
|
num_matched = 0
|
|
for item in object_array:
|
|
all_match = True
|
|
for key,value in to_match.items():
|
|
if item[key] != value:
|
|
all_match = False
|
|
if not all_match:
|
|
continue
|
|
for key,value in expected.items():
|
|
if item[key] != value:
|
|
raise AssertionError("%s : expected %s=%s"%(str(item), str(key), str(value)))
|
|
num_matched = num_matched+1
|
|
if num_matched == 0:
|
|
raise AssertionError("No objects matched %s"%(str(to_match)))
|
|
|
|
import threading
|
|
|
|
class LongpollThread(threading.Thread):
|
|
def __init__(self, node):
|
|
threading.Thread.__init__(self)
|
|
# query current longpollid
|
|
templat = node.getblocktemplate()
|
|
self.longpollid = templat['longpollid']
|
|
# create a new connection to the node, we can't use the same
|
|
# connection from two threads
|
|
self.node = AuthServiceProxy(node.url, timeout=600)
|
|
|
|
def run(self):
|
|
self.node.getblocktemplate({'longpollid':self.longpollid})
|
|
|
|
class GetBlockTemplateLPTest(BitcoinTestFramework):
|
|
'''
|
|
Test longpolling with getblocktemplate.
|
|
'''
|
|
|
|
def run_test(self):
|
|
print("Warning: this test will take about 70 seconds in the best case. Be patient.")
|
|
self.nodes[0].generate(10)
|
|
templat = self.nodes[0].getblocktemplate()
|
|
longpollid = templat['longpollid']
|
|
# longpollid should not change between successive invocations if nothing else happens
|
|
templat2 = self.nodes[0].getblocktemplate()
|
|
assert(templat2['longpollid'] == longpollid)
|
|
|
|
# Test 1: test that the longpolling wait if we do nothing
|
|
thr = LongpollThread(self.nodes[0])
|
|
thr.start()
|
|
# check that thread still lives
|
|
thr.join(5) # wait 5 seconds or until thread exits
|
|
assert(thr.is_alive())
|
|
|
|
# Test 2: test that longpoll will terminate if another node generates a block
|
|
self.nodes[1].generate(1) # generate a block on another node
|
|
# check that thread will exit now that new transaction entered mempool
|
|
thr.join(5) # wait 5 seconds or until thread exits
|
|
assert(not thr.is_alive())
|
|
|
|
# Test 3: test that longpoll will terminate if we generate a block ourselves
|
|
thr = LongpollThread(self.nodes[0])
|
|
thr.start()
|
|
self.nodes[0].generate(1) # generate a block on another node
|
|
thr.join(5) # wait 5 seconds or until thread exits
|
|
assert(not thr.is_alive())
|
|
|
|
# Test 4: test that introducing a new transaction into the mempool will terminate the longpoll
|
|
thr = LongpollThread(self.nodes[0])
|
|
thr.start()
|
|
# generate a random transaction and submit it
|
|
(txid, txhex, fee) = random_transaction(self.nodes, Decimal("1.1"), Decimal("0.0"), Decimal("0.001"), 20)
|
|
# after one minute, every 10 seconds the mempool is probed, so in 80 seconds it should have returned
|
|
thr.join(60 + 20)
|
|
assert(not thr.is_alive())
|
|
|
|
if __name__ == '__main__':
|
|
GetBlockTemplateLPTest().main()
|
|
|