diff --git a/tests/test_objects.py b/tests/test_objects.py index 6a4c0c69e5700e5fdd3907494aaadef6eac5bc08..8ca24c54d69172626a1faf703fc049020efbaa3e 100644 --- a/tests/test_objects.py +++ b/tests/test_objects.py @@ -5,9 +5,7 @@ It sounds like a melody from __future__ import print_function, absolute_import, division import six import unittest - - -from coolamqp.objects import NodeDefinition +from coolamqp.objects import NodeDefinition, Message class TestObjects(unittest.TestCase): @@ -23,3 +21,10 @@ class TestObjects(unittest.TestCase): n1 = NodeDefinition(u'amqp://ala:ma@kota/') self.assertEquals(n1.virtual_host, u'/') + + def test_get_message_properties(self): + empty_p_msg = Message(b'') + ce_p_msg = Message(b'', content_encoding=b'wtf') + + self.assertIsNone(empty_p_msg.properties.get('content_encoding')) + self.assertEquals(ce_p_msg.properties.get('content_encoding', b'wtf'))