|
|
|
@ -41,40 +41,36 @@ class TestManager(NodeConnCB):
|
|
|
|
|
self.disconnectOkay = False
|
|
|
|
|
|
|
|
|
|
def run(self):
|
|
|
|
|
try:
|
|
|
|
|
fail = False
|
|
|
|
|
self.connection.rpc.generate(1) # Leave IBD
|
|
|
|
|
|
|
|
|
|
numBlocksToGenerate = [ 8, 16, 128, 1024 ]
|
|
|
|
|
for count in range(len(numBlocksToGenerate)):
|
|
|
|
|
current_invs = []
|
|
|
|
|
for i in range(numBlocksToGenerate[count]):
|
|
|
|
|
current_invs.append(CInv(2, random.randrange(0, 1<<256)))
|
|
|
|
|
if len(current_invs) >= 50000:
|
|
|
|
|
self.connection.send_message(msg_inv(current_invs))
|
|
|
|
|
current_invs = []
|
|
|
|
|
if len(current_invs) > 0:
|
|
|
|
|
self.connection.rpc.generate(1) # Leave IBD
|
|
|
|
|
|
|
|
|
|
numBlocksToGenerate = [8, 16, 128, 1024]
|
|
|
|
|
for count in range(len(numBlocksToGenerate)):
|
|
|
|
|
current_invs = []
|
|
|
|
|
for i in range(numBlocksToGenerate[count]):
|
|
|
|
|
current_invs.append(CInv(2, random.randrange(0, 1 << 256)))
|
|
|
|
|
if len(current_invs) >= 50000:
|
|
|
|
|
self.connection.send_message(msg_inv(current_invs))
|
|
|
|
|
|
|
|
|
|
# Wait and see how many blocks were requested
|
|
|
|
|
time.sleep(2)
|
|
|
|
|
|
|
|
|
|
total_requests = 0
|
|
|
|
|
with mininode_lock:
|
|
|
|
|
for key in self.blockReqCounts:
|
|
|
|
|
total_requests += self.blockReqCounts[key]
|
|
|
|
|
if self.blockReqCounts[key] > 1:
|
|
|
|
|
raise AssertionError("Error, test failed: block %064x requested more than once" % key)
|
|
|
|
|
if total_requests > MAX_REQUESTS:
|
|
|
|
|
raise AssertionError("Error, too many blocks (%d) requested" % total_requests)
|
|
|
|
|
print "Round %d: success (total requests: %d)" % (count, total_requests)
|
|
|
|
|
except AssertionError as e:
|
|
|
|
|
print "TEST FAILED: ", e.args
|
|
|
|
|
current_invs = []
|
|
|
|
|
if len(current_invs) > 0:
|
|
|
|
|
self.connection.send_message(msg_inv(current_invs))
|
|
|
|
|
|
|
|
|
|
# Wait and see how many blocks were requested
|
|
|
|
|
time.sleep(2)
|
|
|
|
|
|
|
|
|
|
total_requests = 0
|
|
|
|
|
with mininode_lock:
|
|
|
|
|
for key in self.blockReqCounts:
|
|
|
|
|
total_requests += self.blockReqCounts[key]
|
|
|
|
|
if self.blockReqCounts[key] > 1:
|
|
|
|
|
raise AssertionError("Error, test failed: block %064x requested more than once" % key)
|
|
|
|
|
if total_requests > MAX_REQUESTS:
|
|
|
|
|
raise AssertionError("Error, too many blocks (%d) requested" % total_requests)
|
|
|
|
|
print "Round %d: success (total requests: %d)" % (count, total_requests)
|
|
|
|
|
|
|
|
|
|
self.disconnectOkay = True
|
|
|
|
|
self.connection.disconnect_node()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class MaxBlocksInFlightTest(BitcoinTestFramework):
|
|
|
|
|
def add_options(self, parser):
|
|
|
|
|
parser.add_option("--testbinary", dest="testbinary",
|
|
|
|
@ -86,7 +82,7 @@ class MaxBlocksInFlightTest(BitcoinTestFramework):
|
|
|
|
|
initialize_chain_clean(self.options.tmpdir, 1)
|
|
|
|
|
|
|
|
|
|
def setup_network(self):
|
|
|
|
|
self.nodes = start_nodes(1, self.options.tmpdir,
|
|
|
|
|
self.nodes = start_nodes(1, self.options.tmpdir,
|
|
|
|
|
extra_args=[['-debug', '-whitelist=127.0.0.1']],
|
|
|
|
|
binary=[self.options.testbinary])
|
|
|
|
|
|
|
|
|
|