0
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-02-13 11:25:02 -05:00

test: check disconnection when sending sendaddrv2 after verack

This commit is contained in:
brunoerg 2024-03-21 17:51:40 -03:00
parent c8e3978114
commit b4c9ace6ff

View file

@ -11,6 +11,7 @@ import time
from test_framework.messages import ( from test_framework.messages import (
CAddress, CAddress,
msg_addrv2, msg_addrv2,
msg_sendaddrv2,
) )
from test_framework.p2p import ( from test_framework.p2p import (
P2PInterface, P2PInterface,
@ -75,6 +76,12 @@ class AddrTest(BitcoinTestFramework):
self.extra_args = [["-whitelist=addr@127.0.0.1"]] self.extra_args = [["-whitelist=addr@127.0.0.1"]]
def run_test(self): def run_test(self):
self.log.info('Check disconnection when sending sendaddrv2 after verack')
conn = self.nodes[0].add_p2p_connection(P2PInterface())
with self.nodes[0].assert_debug_log(['sendaddrv2 received after verack from peer=0; disconnecting']):
conn.send_message(msg_sendaddrv2())
conn.wait_for_disconnect()
self.log.info('Create connection that sends addrv2 messages') self.log.info('Create connection that sends addrv2 messages')
addr_source = self.nodes[0].add_p2p_connection(P2PInterface()) addr_source = self.nodes[0].add_p2p_connection(P2PInterface())
msg = msg_addrv2() msg = msg_addrv2()
@ -89,8 +96,8 @@ class AddrTest(BitcoinTestFramework):
msg.addrs = ADDRS msg.addrs = ADDRS
msg_size = calc_addrv2_msg_size(ADDRS) msg_size = calc_addrv2_msg_size(ADDRS)
with self.nodes[0].assert_debug_log([ with self.nodes[0].assert_debug_log([
f'received: addrv2 ({msg_size} bytes) peer=0', f'received: addrv2 ({msg_size} bytes) peer=1',
f'sending addrv2 ({msg_size} bytes) peer=1', f'sending addrv2 ({msg_size} bytes) peer=2',
]): ]):
addr_source.send_and_ping(msg) addr_source.send_and_ping(msg)
self.nodes[0].setmocktime(int(time.time()) + 30 * 60) self.nodes[0].setmocktime(int(time.time()) + 30 * 60)