diff --git a/minode/tests/test_structure.py b/minode/tests/test_structure.py index 16e41fe..6a0fa2f 100644 --- a/minode/tests/test_structure.py +++ b/minode/tests/test_structure.py @@ -23,6 +23,9 @@ sample_addr_data = unhexlify( sample_object_data = unhexlify( '000000000000000000000000652724030000002a010248454c4c4f') +sample_onion_host = \ + 'bmtestlmgmvpbsg7kzmrxu47chs3cdou2tj4t5iloocgujzsf3e7rbqd.onion' + logging.basicConfig( level=shared.log_level, format='[%(asctime)s] [%(levelname)s] %(message)s') @@ -171,3 +174,21 @@ class TestStructure(unittest.TestCase): nonce, obj.expires_time, obj.object_type, obj.version, obj.stream_number, obj.object_payload) self.assertTrue(obj.is_valid()) + + def test_onion_peer(self): + """Make an onion peer object and decode it back""" + with self.assertRaises(ValueError): + onion_peer = structure.OnionPeer('testing2') + with self.assertRaises(ValueError): + onion_peer = structure.OnionPeer('testing.onion') + onion_peer = structure.OnionPeer(sample_onion_host) + self.assertEqual(onion_peer.stream, shared.stream) + obj = onion_peer.to_object() + self.assertEqual(obj.object_type, shared.onion_obj_type) + self.assertEqual(obj.version, shared.onion_obj_version) + decoded = structure.OnionPeer.from_object(obj) + self.assertEqual(decoded.dest_pub, onion_peer.dest_pub) + self.assertEqual(decoded.port, onion_peer.port) + obj.object_payload = obj.object_payload[0:1] + obj.object_payload[2:] + with self.assertRaises(ValueError): + structure.OnionPeer.from_object(obj)