From 09186c5f308f3f5eb41002efc525130910ba6631 Mon Sep 17 00:00:00 2001 From: Piotr Maslanka <piotr.maslanka@henrietta.com.pl> Date: Sun, 4 Jun 2017 23:18:30 +0200 Subject: [PATCH] and tests --- tests/test_objects.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/test_objects.py b/tests/test_objects.py index 6a4c0c6..8ca24c5 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')) -- GitLab