mirror of
https://github.com/dashpay/dash.git
synced 2024-12-27 04:52:59 +01:00
2b941e594b
0fcaf731997c4989b869e42d8990f742637799c2 test: use explicit p2p objects where available (Oliver Gugger) Pull request description: This is a follow-up patch to #19804 as suggested by MarcoFalke (https://github.com/bitcoin/bitcoin/pull/19804#discussion_r494950062). To make the intent of the tests easier to understand, we reference the p2p connection objects by their explicit names instead of the p2ps array. ACKs for top commit: theStack: ACK 0fcaf731997c4989b869e42d8990f742637799c2 Tree-SHA512: 37db22185077beeadfa7245bb768b87d6b7a2cfb906c3c859ab92ec3d657122db7301777f0838e13dfc748f37303850144fb7553e6cb6c66903e304d6e10e659
50 lines
1.6 KiB
Python
Executable File
50 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 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 GETDATA processing behavior"""
|
|
from collections import defaultdict
|
|
|
|
from test_framework.messages import (
|
|
CInv,
|
|
msg_getdata,
|
|
)
|
|
from test_framework.p2p import P2PInterface
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
|
|
|
|
class P2PStoreBlock(P2PInterface):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.blocks = defaultdict(int)
|
|
|
|
def on_block(self, message):
|
|
message.block.calc_sha256()
|
|
self.blocks[message.block.sha256] += 1
|
|
|
|
|
|
class GetdataTest(BitcoinTestFramework):
|
|
def set_test_params(self):
|
|
self.num_nodes = 1
|
|
|
|
def run_test(self):
|
|
p2p_block_store = self.nodes[0].add_p2p_connection(P2PStoreBlock())
|
|
|
|
self.log.info("test that an invalid GETDATA doesn't prevent processing of future messages")
|
|
|
|
# Send invalid message and verify that node responds to later ping
|
|
invalid_getdata = msg_getdata()
|
|
invalid_getdata.inv.append(CInv(t=0, h=0)) # INV type 0 is invalid.
|
|
p2p_block_store.send_and_ping(invalid_getdata)
|
|
|
|
# Check getdata still works by fetching tip block
|
|
best_block = int(self.nodes[0].getbestblockhash(), 16)
|
|
good_getdata = msg_getdata()
|
|
good_getdata.inv.append(CInv(t=2, h=best_block))
|
|
p2p_block_store.send_and_ping(good_getdata)
|
|
p2p_block_store.wait_until(lambda: p2p_block_store.blocks[best_block] == 1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
GetdataTest().main()
|