Merge pull request #31 from golemfactory/init

Messages should be initialized even before creating first message

golem_messages/message.py

@@ -81,9 +81,6 @@ def __init__(self, timestamp=None, encrypted=False, sig=None,
         :param sig: signed message hash
         :param raw: original message bytes
         """
-        if not registered_message_types:
-            init_messages()
-
         # Child message slots
         self.load_slots(slots)
 
@@ -1313,3 +1310,6 @@ def init_messages():
                 .format(message_class.__name__, message_class.TYPE)
             )
         registered_message_types[message_class.TYPE] = message_class
+
+
+init_messages()

tests/test_basic.py

@@ -27,6 +27,21 @@ def test_total_basic(self):
                                    self.ecc.raw_pubkey)
         self.assertEqual(msg, msg2)
 
+    """ Deserialization should work even if we haven't created any message first
+    """
+    @mock.patch('golem_messages.message.verify_time')
+    def test_deserialization(self, verify_time):
+        verify_time.return_value = True
+        serialized_ping = b'\x03\xe9\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'\
+            b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'\
+            b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'\
+            b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'\
+            b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'\
+            b'\xd8\x1c\x80'
+        deserialized = message.Message.deserialize(serialized_ping, None)
+        assert deserialized is not None
+        assert deserialized.TYPE == message.MessagePing.TYPE
+
 
 testnow = datetime.datetime.utcnow().replace(microsecond=0)