diff options
Diffstat (limited to 'test/test_client.py')
| -rw-r--r-- | test/test_client.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/test/test_client.py b/test/test_client.py index b7591e9..601e454 100644 --- a/test/test_client.py +++ b/test/test_client.py @@ -1,5 +1,6 @@ from . import unittest +import six from mock import MagicMock, patch from kafka import KafkaClient @@ -15,25 +16,25 @@ class TestKafkaClient(unittest.TestCase): with patch.object(KafkaClient, 'load_metadata_for_topics'): client = KafkaClient(hosts=['kafka01:9092', 'kafka02:9092', 'kafka03:9092']) - self.assertItemsEqual( - [('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)], - client.hosts) + self.assertEqual( + sorted([('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)]), + sorted(client.hosts)) def test_init_with_csv(self): with patch.object(KafkaClient, 'load_metadata_for_topics'): client = KafkaClient(hosts='kafka01:9092,kafka02:9092,kafka03:9092') - self.assertItemsEqual( - [('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)], - client.hosts) + self.assertEqual( + sorted([('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)]), + sorted(client.hosts)) def test_init_with_unicode_csv(self): with patch.object(KafkaClient, 'load_metadata_for_topics'): client = KafkaClient(hosts=u'kafka01:9092,kafka02:9092,kafka03:9092') - self.assertItemsEqual( - [('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)], - client.hosts) + self.assertEqual( + sorted([('kafka01', 9092), ('kafka02', 9092), ('kafka03', 9092)]), + sorted(client.hosts)) def test_send_broker_unaware_request_fail(self): 'Tests that call fails when all hosts are unavailable' @@ -58,7 +59,7 @@ class TestKafkaClient(unittest.TestCase): with self.assertRaises(KafkaUnavailableError): client._send_broker_unaware_request(1, 'fake request') - for key, conn in mocked_conns.iteritems(): + for key, conn in six.iteritems(mocked_conns): conn.send.assert_called_with(1, 'fake request') def test_send_broker_unaware_request(self): |