diff options
Diffstat (limited to 'test/test_protocol.py')
| -rw-r--r-- | test/test_protocol.py | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/test/test_protocol.py b/test/test_protocol.py index 125169f..8bd2f5e 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -1,5 +1,5 @@ import struct -import unittest +import unittest2 from kafka import KafkaClient from kafka.common import ( @@ -19,7 +19,7 @@ from kafka.protocol import ( create_gzip_message, create_message, create_snappy_message, KafkaProtocol ) -class TestProtocol(unittest.TestCase): +class TestProtocol(unittest2.TestCase): def test_create_message(self): payload = "test" key = "key" @@ -58,7 +58,7 @@ class TestProtocol(unittest.TestCase): self.assertEqual(decoded, expect) - @unittest.skipUnless(has_snappy(), "Snappy not available") + @unittest2.skipUnless(has_snappy(), "Snappy not available") def test_create_snappy(self): payloads = ["v1", "v2"] msg = create_snappy_message(payloads) @@ -216,7 +216,7 @@ class TestProtocol(unittest.TestCase): self.assertEqual(returned_offset2, 0) self.assertEqual(decoded_message2, create_message("v2")) - @unittest.skipUnless(has_snappy(), "Snappy not available") + @unittest2.skipUnless(has_snappy(), "Snappy not available") def test_decode_message_snappy(self): snappy_encoded = ('\xec\x80\xa1\x95\x00\x02\xff\xff\xff\xff\x00\x00' '\x00,8\x00\x00\x19\x01@\x10L\x9f[\xc2\x00\x00\xff' @@ -567,10 +567,10 @@ class TestProtocol(unittest.TestCase): ]) results = KafkaProtocol.decode_offset_response(encoded) - self.assertEqual(set(results), { + self.assertEqual(set(results), set([ OffsetResponse(topic = 'topic1', partition = 2, error = 0, offsets=(4,)), OffsetResponse(topic = 'topic1', partition = 4, error = 0, offsets=(8,)), - }) + ])) def test_encode_offset_commit_request(self): header = "".join([ @@ -629,10 +629,10 @@ class TestProtocol(unittest.TestCase): ]) results = KafkaProtocol.decode_offset_commit_response(encoded) - self.assertEqual(set(results), { + self.assertEqual(set(results), set([ OffsetCommitResponse(topic = 'topic1', partition = 2, error = 0), OffsetCommitResponse(topic = 'topic1', partition = 4, error = 0), - }) + ])) def test_encode_offset_fetch_request(self): header = "".join([ @@ -688,7 +688,7 @@ class TestProtocol(unittest.TestCase): ]) results = KafkaProtocol.decode_offset_fetch_response(encoded) - self.assertEqual(set(results), { + self.assertEqual(set(results), set([ OffsetFetchResponse(topic = 'topic1', partition = 2, offset = 4, error = 0, metadata = "meta"), OffsetFetchResponse(topic = 'topic1', partition = 4, offset = 8, error = 0, metadata = "meta"), - }) + ])) |