diff --git a/tests/gossip-unittests/test_client_get_new_peers.py b/tests/gossip-unittests/test_client_get_new_peers.py index 7d369a93..878217af 100644 --- a/tests/gossip-unittests/test_client_get_new_peers.py +++ b/tests/gossip-unittests/test_client_get_new_peers.py @@ -36,7 +36,6 @@ TRANSPORT_SIZE_BYTES = 64 server_file = TEST_DIR + 'test_serv.sock' - class MockPeer: def __init__(self): self.transport_address = secrets.token_hex(16) @@ -49,6 +48,21 @@ class MockPeer: 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):