diff options
Diffstat (limited to 'test/test_producer.py')
| -rw-r--r-- | test/test_producer.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index e00f9af..1375525 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -2,25 +2,25 @@ import logging -import unittest2 from mock import MagicMock +from . import unittest from kafka.producer import Producer -class TestKafkaProducer(unittest2.TestCase): +class TestKafkaProducer(unittest.TestCase): def test_producer_message_types(self): producer = Producer(MagicMock()) - topic = "test-topic" + topic = b"test-topic" partition = 0 - bad_data_types = (u'你怎么样?', 12, ['a','list'], ('a','tuple'), {'a': 'dict'}) + bad_data_types = (u'你怎么样?', 12, ['a', 'list'], ('a', 'tuple'), {'a': 'dict'}) for m in bad_data_types: with self.assertRaises(TypeError): logging.debug("attempting to send message of type %s", type(m)) producer.send_messages(topic, partition, m) - good_data_types = ('a string!',) + good_data_types = (b'a string!',) for m in good_data_types: # This should not raise an exception producer.send_messages(topic, partition, m) |