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