From b2722acdf7d41a13d0a3413bb5d695c7b066ed2b Mon Sep 17 00:00:00 2001
From: Dmitri Bogomolov <4glitch@gmail.com>
Date: Sun, 1 Aug 2021 19:58:06 +0300
Subject: [PATCH] Add some tests for message

---
 minode/tests/test_message.py | 42 ++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 minode/tests/test_message.py

diff --git a/minode/tests/test_message.py b/minode/tests/test_message.py
new file mode 100644
index 0000000..8ba2925
--- /dev/null
+++ b/minode/tests/test_message.py
@@ -0,0 +1,42 @@
+import unittest
+from binascii import unhexlify
+
+from minode import message
+
+
+magic = 0xE9BEB4D9
+
+# 500 identical peers:
+# import ipaddress
+# from hyperbit import net, packet
+# [packet.Address(
+#     1626611891, 1, 1, net.ipv6(ipaddress.ip_address('127.0.0.1')).packed,
+#     8444
+# ) for _ in range(1000)]
+sample_data = unhexlify(
+    'fd01f4' + (
+        '0000000060f420b30000000'
+        '1000000000000000100000000000000000000ffff7f00000120fc'
+        ) * 500
+)
+
+
+class TestMessage(unittest.TestCase):
+    """Test assembling and disassembling of network mesages"""
+
+    def test_packet(self):
+        """Check the packet created by message.Message()"""
+        head = unhexlify(b'%x' % magic)
+        self.assertEqual(
+            message.Message(b'ping', b'').to_bytes()[:len(head)], head)
+
+    def test_addr(self):
+        """Test addr messages"""
+        msg = message.Message(b'addr', sample_data)
+        addr_packet = message.Addr.from_message(msg)
+        self.assertEqual(len(addr_packet.addresses), 500)
+        address = addr_packet.addresses.pop()
+        self.assertEqual(address.stream, 1)
+        self.assertEqual(address.services, 1)
+        self.assertEqual(address.port, 8444)
+        self.assertEqual(address.host, '127.0.0.1')