From bae7f745eed1721cc817788796911dbc9d01568b Mon Sep 17 00:00:00 2001 From: Kevin Froman Date: Fri, 5 Feb 2021 04:57:35 +0000 Subject: [PATCH] implemented peer announce in torgossip --- static-data/default-plugins/torgossip/runtest.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/static-data/default-plugins/torgossip/runtest.py b/static-data/default-plugins/torgossip/runtest.py index daf9bd46..825addb8 100644 --- a/static-data/default-plugins/torgossip/runtest.py +++ b/static-data/default-plugins/torgossip/runtest.py @@ -72,7 +72,8 @@ def torgossip_runtest(test_manager): assert s.recv(64) == bl_new.get_packed() s.sendall(b'40,tbt') - assert len(s.recv(100000)) == len(shared_state.get_by_string("SafeDB").get('bl-tbt')) + assert len(s.recv(100000)) == len( + shared_state.get_by_string("SafeDB").get('bl-tbt')) s.sendall(b'41,tbt') assert len(s.recv(100000)) == len( @@ -86,10 +87,11 @@ def torgossip_runtest(test_manager): fakes.append(fake) shared_state.get_by_string('TorGossipPeers').add_peer(fake) shared_state.get_by_string('TorGossipPeers').add_peer(fake_peer) - shared_state.get_by_string('TorGossipPeers').add_score(fake_peer, 100000) + shared_state.get_by_string( + 'TorGossipPeers').add_score(fake_peer, 100000) s.sendall(b'71') stored = s.recv(1000) - expected = _shrink_peer_address(fake_peer)#b32decode(fake_peer.replace('.onion', '')).replace(b'.onion', b'') + expected = _shrink_peer_address(fake_peer) try: assert stored == expected except AssertionError: @@ -102,11 +104,12 @@ def torgossip_runtest(test_manager): announce_raw = _shrink_peer_address(announce_peer) s.sendall(b'8' + announce_raw) assert s.recv(1) == b'1' - print(type(announce_raw), type(shared_state.get_by_string('TorGossipPeers').get_highest_score_peers(100)[0][0])) - assert announce_raw == shared_state.get_by_string('TorGossipPeers').get_highest_score_peers(100)[0][0] + print(type(announce_raw), type(shared_state.get_by_string( + 'TorGossipPeers').get_highest_score_peers(100)[0][0])) + assert announce_raw == shared_state.get_by_string( + 'TorGossipPeers').get_highest_score_peers(100)[0][0] shared_state.get_by_string('TorGossipPeers').remove_peer(announce_raw) - s.sendall(b'9') assert s.recv(64) == b"BYE"