diff --git a/tests/test_objects.py b/tests/test_objects.py index 8ca24c54d69172626a1faf703fc049020efbaa3e..aaeb6ebcf9804b7047545fdfd8d6f92b722d16e9 100644 --- a/tests/test_objects.py +++ b/tests/test_objects.py @@ -5,7 +5,7 @@ It sounds like a melody from __future__ import print_function, absolute_import, division import six import unittest -from coolamqp.objects import NodeDefinition, Message +from coolamqp.objects import NodeDefinition, MessageProperties class TestObjects(unittest.TestCase): @@ -23,8 +23,8 @@ class TestObjects(unittest.TestCase): 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') + empty_p_msg = MessageProperties() + ce_p_msg = MessageProperties(content_encoding=b'wtf') - self.assertIsNone(empty_p_msg.properties.get('content_encoding')) - self.assertEquals(ce_p_msg.properties.get('content_encoding', b'wtf')) + self.assertIsNone(empty_p_msg.get('content_encoding')) + self.assertEquals(ce_p_msg.get('content_encoding', b'wtf'))