From 3f7250b328b8b2f5d63f323702445ac5c989b73d Mon Sep 17 00:00:00 2001 From: Amiti Uttarwar Date: Fri, 23 Jul 2021 18:19:32 -0700 Subject: [test] Use the new endpoint to improve tests --- test/functional/p2p_addr_relay.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'test/functional/p2p_addr_relay.py') diff --git a/test/functional/p2p_addr_relay.py b/test/functional/p2p_addr_relay.py index 4233d8d010..95743a1bf5 100755 --- a/test/functional/p2p_addr_relay.py +++ b/test/functional/p2p_addr_relay.py @@ -210,6 +210,11 @@ class AddrTest(BitcoinTestFramework): blackhole_peer = self.nodes[0].add_p2p_connection(AddrReceiver(send_getaddr=False)) initial_addrs_received = receiver_peer.num_ipv4_received + peerinfo = self.nodes[0].getpeerinfo() + assert_equal(peerinfo[0]['addr_relay_enabled'], True) # addr_source + assert_equal(peerinfo[1]['addr_relay_enabled'], True) # receiver_peer + assert_equal(peerinfo[2]['addr_relay_enabled'], False) # blackhole_peer + # addr_source sends 2 addresses to node0 msg = self.setup_addr_msg(2) addr_source.send_and_ping(msg) @@ -232,11 +237,14 @@ class AddrTest(BitcoinTestFramework): self.log.info("After blackhole peer sends addr message, it becomes eligible for addr gossip") blackhole_peer.send_and_ping(msg_addr()) - msg = self.setup_addr_msg(2) - self.send_addr_msg(addr_source, msg, [receiver_peer, blackhole_peer]) # Confirm node has now received addr-related messages from blackhole peer assert_greater_than(self.sum_addr_messages(peerinfo[1]['bytesrecv_per_msg']), 0) + assert_equal(self.nodes[0].getpeerinfo()[2]['addr_relay_enabled'], True) + + msg = self.setup_addr_msg(2) + self.send_addr_msg(addr_source, msg, [receiver_peer, blackhole_peer]) + # And that peer received addresses assert_equal(blackhole_peer.num_ipv4_received, 2) -- cgit v1.2.3