kev/Onionr
1
0
Fork 0
Code Issues 24 Pull Requests Projects Releases 2 Wiki Activity

started get new peers test

Browse Source
This commit is contained in:
Kevin F 2022-05-14 11:28:12 -05:00
parent 2be0175326
commit 12d4ed7638
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/gossip-unittests/test_client_get_new_peers.py
View File

@ -36,7 +36,6 @@ TRANSPORT_SIZE_BYTES = 64
server_file = TEST_DIR + 'test_serv.sock' server_file = TEST_DIR + 'test_serv.sock'
class MockPeer: class MockPeer:
def __init__(self): def __init__(self):
self.transport_address = secrets.token_hex(16) self.transport_address = secrets.token_hex(16)
@ -49,6 +48,21 @@ class MockPeer:
return s return s
def _server():
fake_peer_addresses = [MockPeer().transport_address for i in range(10)]
with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s:
s.bind(server_file)
s.listen(1)
conn, _ = s.accept()
with conn:
conn.recv(1)
for address in fake_peer_addresses:
conn.sendall(address.zfill(TRANSPORT_SIZE_BYTES).encode('utf-8'))
Thread(target=_server, daemon=True).start()
class OnionrGossipClientGetNewPeers(unittest.TestCase): class OnionrGossipClientGetNewPeers(unittest.TestCase):