6
6
7
7
import time
8
8
9
- from test_framework .messages import msg_getaddr
10
9
from test_framework .p2p import (
11
10
P2PInterface ,
12
11
p2p_lock
21
20
MAX_ADDR_TO_SEND = 1000
22
21
MAX_PCT_ADDR_TO_SEND = 23
23
22
23
+
24
24
class AddrReceiver (P2PInterface ):
25
25
26
26
def __init__ (self ):
@@ -70,11 +70,8 @@ def run_test(self):
70
70
cur_mock_time = int (time .time ())
71
71
for i in range (N ):
72
72
addr_receiver_local = self .nodes [0 ].add_p2p_connection (AddrReceiver ())
73
- addr_receiver_local .send_and_ping (msg_getaddr ())
74
73
addr_receiver_onion1 = self .nodes [0 ].add_p2p_connection (AddrReceiver (), dstport = self .onion_port1 )
75
- addr_receiver_onion1 .send_and_ping (msg_getaddr ())
76
74
addr_receiver_onion2 = self .nodes [0 ].add_p2p_connection (AddrReceiver (), dstport = self .onion_port2 )
77
- addr_receiver_onion2 .send_and_ping (msg_getaddr ())
78
75
79
76
# Trigger response
80
77
cur_mock_time += 5 * 60
@@ -105,11 +102,8 @@ def run_test(self):
105
102
106
103
self .log .info ('After time passed, see a new response to addr request' )
107
104
addr_receiver_local = self .nodes [0 ].add_p2p_connection (AddrReceiver ())
108
- addr_receiver_local .send_and_ping (msg_getaddr ())
109
105
addr_receiver_onion1 = self .nodes [0 ].add_p2p_connection (AddrReceiver (), dstport = self .onion_port1 )
110
- addr_receiver_onion1 .send_and_ping (msg_getaddr ())
111
106
addr_receiver_onion2 = self .nodes [0 ].add_p2p_connection (AddrReceiver (), dstport = self .onion_port2 )
112
- addr_receiver_onion2 .send_and_ping (msg_getaddr ())
113
107
114
108
# Trigger response
115
109
cur_mock_time += 5 * 60
@@ -123,5 +117,6 @@ def run_test(self):
123
117
assert set (last_response_on_onion_bind1 ) != set (addr_receiver_onion1 .get_received_addrs ())
124
118
assert set (last_response_on_onion_bind2 ) != set (addr_receiver_onion2 .get_received_addrs ())
125
119
120
+
126
121
if __name__ == '__main__' :
127
122
AddrTest ().main ()
0 commit comments