From ca2344c72c603643dbc3c0671dd7ff2ea74a2371 Mon Sep 17 00:00:00 2001
From: Kevin F <kevin@voidnet>
Date: Sat, 23 Apr 2022 00:41:57 -0500
Subject: [PATCH] Added peer announce and exchange tests

---
 tests/gossip-unittests/test_peer_announce.py | 78 +++++++++++++++++++
 tests/gossip-unittests/test_peer_exchange.py | 79 ++++++++++++++++++++
 2 files changed, 157 insertions(+)
 create mode 100644 tests/gossip-unittests/test_peer_announce.py
 create mode 100644 tests/gossip-unittests/test_peer_exchange.py

diff --git a/tests/gossip-unittests/test_peer_announce.py b/tests/gossip-unittests/test_peer_announce.py
new file mode 100644
index 00000000..9c1964b6
--- /dev/null
+++ b/tests/gossip-unittests/test_peer_announce.py
@@ -0,0 +1,78 @@
+from ast import Assert
+import os, uuid
+from unittest.mock import Mock
+from time import sleep
+import secrets
+
+
+TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
+print("Test directory:", TEST_DIR)
+os.environ["ONIONR_HOME"] = TEST_DIR
+
+from threading import Thread
+import asyncio
+import unittest
+import sys
+sys.path.append(".")
+sys.path.append("src/")
+from unittest.mock import patch
+
+from ordered_set import OrderedSet
+from gossip import peerset
+
+
+import onionrblocks
+
+import blockdb
+from gossip.server import gossip_server
+from gossip.peer import Peer
+
+
+
+from filepaths import gossip_server_socket_file
+
+
+BLOCK_MAX_SIZE = 1024 * 2000
+BLOCK_MAX_SIZE_LEN = len(str(BLOCK_MAX_SIZE))
+BLOCK_ID_SIZE = 128
+BLOCK_STREAM_OFFSET_DIGITS = 8
+MAX_PEERS = 10
+TRANSPORT_SIZE_BYTES = 64
+
+class MockPeer(Peer):
+    def __init__(self):
+        self.transport_address = secrets.token_hex(16)
+    def __hash__(self):
+        return hash(self.transport_address)
+
+
+class OnionrServerPeerAnnounce(unittest.TestCase):
+
+
+    def test_peer_announce(self):
+
+        Thread(target=gossip_server, daemon=True).start()
+        sleep(0.03)
+
+        address = MockPeer().transport_address
+
+        async def announce_client():
+            reader, writer = await asyncio.open_unix_connection(
+                gossip_server_socket_file)
+            writer.write(int(2).to_bytes(1, 'big'))
+            await writer.drain()
+            writer.write(address.encode('utf-8') + b'\n')
+            await writer.drain()
+            self.assertEqual(await reader.readexactly(1), int(1).to_bytes(1,'big'))
+
+            try:
+                await reader.readexactly(1)
+            except asyncio.IncompleteReadError:
+                pass
+            else:
+                raise AssertionError("Read past expected, announce stream must close after announce submission")
+
+
+        asyncio.run(announce_client())
+
+unittest.main()
diff --git a/tests/gossip-unittests/test_peer_exchange.py b/tests/gossip-unittests/test_peer_exchange.py
new file mode 100644
index 00000000..2c65a9ce
--- /dev/null
+++ b/tests/gossip-unittests/test_peer_exchange.py
@@ -0,0 +1,79 @@
+from ast import Assert
+import os, uuid
+from unittest.mock import Mock
+from time import sleep
+import secrets
+
+
+TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
+print("Test directory:", TEST_DIR)
+os.environ["ONIONR_HOME"] = TEST_DIR
+
+from threading import Thread
+import asyncio
+import unittest
+import sys
+sys.path.append(".")
+sys.path.append("src/")
+from unittest.mock import patch
+
+from ordered_set import OrderedSet
+from gossip import peerset
+
+
+import onionrblocks
+
+import blockdb
+from gossip.server import gossip_server
+from gossip.peer import Peer
+
+
+
+from filepaths import gossip_server_socket_file
+
+
+BLOCK_MAX_SIZE = 1024 * 2000
+BLOCK_MAX_SIZE_LEN = len(str(BLOCK_MAX_SIZE))
+BLOCK_ID_SIZE = 128
+BLOCK_STREAM_OFFSET_DIGITS = 8
+MAX_PEERS = 10
+TRANSPORT_SIZE_BYTES = 64
+
+class MockPeer(Peer):
+    def __init__(self):
+        self.transport_address = secrets.token_hex(16)
+    def __hash__(self):
+        return hash(self.transport_address)
+
+
+class OnionrServerPeerExchange(unittest.TestCase):
+
+
+    def test_peer_exchange(self):
+
+        Thread(target=gossip_server, daemon=True).start()
+
+        peers = []
+        for _ in range(10):
+            p = MockPeer()
+            peers.append(p)
+            peerset.gossip_peer_set.add(p)
+
+
+        sleep(0.03)
+        async def exchange_client():
+            reader, writer = await asyncio.open_unix_connection(
+                gossip_server_socket_file)
+            writer.write(int(3).to_bytes(1, 'big'))
+            await writer.drain()
+            for i in range(MAX_PEERS):
+                peer = await reader.readline()
+                for p in peers:
+                    if p.transport_address == peer.decode('utf-8').strip():
+                        break
+                else:
+                    raise AssertionError("Peer received not in list " + peer.decode('utf-8'))
+
+        asyncio.run(exchange_client())
+
+unittest.main()