Add some tests for structure

minode/tests/test_structure.py

@@ -0,0 +1,69 @@
+import unittest
+import struct
+from binascii import unhexlify
+
+from minode import structure
+
+
+class TestStructure(unittest.TestCase):
+    """Testing structures serializing and deserializing"""
+
+    def test_varint(self):
+        """Test varint serializing and deserializing"""
+        s = structure.VarInt(0)
+        self.assertEqual(s.to_bytes(), b'\x00')
+        s = structure.VarInt.from_bytes(b'\x00')
+        self.assertEqual(s.n, 0)
+        s = structure.VarInt(42)
+        self.assertEqual(s.to_bytes(), b'*')
+        s = structure.VarInt.from_bytes(b'*')
+        self.assertEqual(s.n, 42)
+        s = structure.VarInt(252)
+        self.assertEqual(s.to_bytes(), unhexlify('fc'))
+        s = structure.VarInt.from_bytes(unhexlify('fc'))
+        self.assertEqual(s.n, 252)
+        s = structure.VarInt(253)
+        self.assertEqual(s.to_bytes(), unhexlify('fd00fd'))
+        s = structure.VarInt.from_bytes(unhexlify('fd00fd'))
+        self.assertEqual(s.n, 253)
+        s = structure.VarInt(100500)
+        self.assertEqual(s.to_bytes(), unhexlify('fe00018894'))
+        s = structure.VarInt.from_bytes(unhexlify('fe00018894'))
+        self.assertEqual(s.n, 100500)
+        s = structure.VarInt(65535)
+        self.assertEqual(s.to_bytes(), unhexlify('fdffff'))
+        s = structure.VarInt.from_bytes(unhexlify('fdffff'))
+        self.assertEqual(s.n, 65535)
+        s = structure.VarInt(4294967295)
+        self.assertEqual(s.to_bytes(), unhexlify('feffffffff'))
+        s = structure.VarInt.from_bytes(unhexlify('feffffffff'))
+        self.assertEqual(s.n, 4294967295)
+        s = structure.VarInt(4294967296)
+        self.assertEqual(s.to_bytes(), unhexlify('ff0000000100000000'))
+        s = structure.VarInt.from_bytes(unhexlify('ff0000000100000000'))
+        self.assertEqual(s.n, 4294967296)
+        s = structure.VarInt(18446744073709551615)
+        self.assertEqual(s.to_bytes(), unhexlify('ffffffffffffffffff'))
+        s = structure.VarInt.from_bytes(unhexlify('ffffffffffffffffff'))
+        self.assertEqual(s.n, 18446744073709551615)
+
+    def test_address(self):
+        """Check address encoding in structure.NetAddrNoPrefix()"""
+        addr = structure.NetAddrNoPrefix(1, '127.0.0.1', 8444)
+        self.assertEqual(
+            addr.to_bytes()[8:24],
+            b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xFF\xFF'
+            + struct.pack('>L', 2130706433))
+        addr = structure.NetAddrNoPrefix(1, '191.168.1.1', 8444)
+        self.assertEqual(
+            addr.to_bytes()[8:24],
+            unhexlify('00000000000000000000ffffbfa80101'))
+        addr = structure.NetAddrNoPrefix(1, '1.1.1.1', 8444)
+        self.assertEqual(
+            addr.to_bytes()[8:24],
+            unhexlify('00000000000000000000ffff01010101'))
+        addr = structure.NetAddrNoPrefix(
+            1, '0102:0304:0506:0708:090A:0B0C:0D0E:0F10', 8444)
+        self.assertEqual(
+            addr.to_bytes()[8:24],
+            unhexlify('0102030405060708090a0b0c0d0e0f10'))