diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-12-10 10:57:27 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-12-10 11:25:15 -0800 |
| commit | 7470cade6bb8629d17541e136527369f9d2ec387 (patch) | |
| tree | ba424c4d0cc27ffb5ec77196a0e121f3075bc992 /test | |
| parent | c3d2fda3c368771cb93a09bb2f1edaa7a3cf9c2b (diff) | |
| download | kafka-python-7470cade6bb8629d17541e136527369f9d2ec387.tar.gz | |
Convert OffsetCommit and OffsetFetch protocol encode/decode
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_client_integration.py | 6 | ||||
| -rw-r--r-- | test/test_consumer.py | 4 | ||||
| -rw-r--r-- | test/test_protocol.py | 4 |
3 files changed, 7 insertions, 7 deletions
diff --git a/test/test_client_integration.py b/test/test_client_integration.py index 70da4a3..edd62da 100644 --- a/test/test_client_integration.py +++ b/test/test_client_integration.py @@ -1,7 +1,7 @@ import os from kafka.common import ( - FetchRequestPayload, OffsetCommitRequest, OffsetFetchRequest, + FetchRequestPayload, OffsetCommitRequestPayload, OffsetFetchRequestPayload, KafkaTimeoutError, ProduceRequestPayload ) from kafka.protocol import create_message @@ -85,11 +85,11 @@ class TestKafkaClientIntegration(KafkaIntegrationTestCase): @kafka_versions("0.8.1", "0.8.1.1", "0.8.2.1") def test_commit_fetch_offsets(self): - req = OffsetCommitRequest(self.bytes_topic, 0, 42, b"metadata") + req = OffsetCommitRequestPayload(self.bytes_topic, 0, 42, b"metadata") (resp,) = self.client.send_offset_commit_request(b"group", [req]) self.assertEqual(resp.error, 0) - req = OffsetFetchRequest(self.bytes_topic, 0) + req = OffsetFetchRequestPayload(self.bytes_topic, 0) (resp,) = self.client.send_offset_fetch_request(b"group", [req]) self.assertEqual(resp.error, 0) self.assertEqual(resp.offset, 42) diff --git a/test/test_consumer.py b/test/test_consumer.py index 31b7e72..ffce578 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -4,7 +4,7 @@ from . import unittest from kafka import SimpleConsumer, KafkaConsumer, MultiProcessConsumer from kafka.common import ( - KafkaConfigurationError, FetchResponsePayload, OffsetFetchResponse, + KafkaConfigurationError, FetchResponsePayload, OffsetFetchResponsePayload, FailedPayloadsError, OffsetAndMessage, NotLeaderForPartitionError, UnknownTopicOrPartitionError ) @@ -86,7 +86,7 @@ class TestSimpleConsumer(unittest.TestCase): client.get_partition_ids_for_topic.return_value = [0, 1] def mock_offset_fetch_request(group, payloads, **kwargs): - return [OffsetFetchResponse(p.topic, p.partition, 0, b'', 0) for p in payloads] + return [OffsetFetchResponsePayload(p.topic, p.partition, 0, b'', 0) for p in payloads] client.send_offset_fetch_request.side_effect = mock_offset_fetch_request diff --git a/test/test_protocol.py b/test/test_protocol.py index c5086b1..8cd4fee 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -7,8 +7,8 @@ from . import unittest from kafka.codec import has_snappy, gzip_decode, snappy_decode from kafka.common import ( - OffsetRequestPayload, OffsetCommitRequest, OffsetFetchRequest, - OffsetResponsePayload, OffsetCommitResponse, OffsetFetchResponse, + OffsetRequestPayload, OffsetCommitRequestPayload, OffsetFetchRequestPayload, + OffsetResponsePayload, OffsetCommitResponsePayload, OffsetFetchResponsePayload, ProduceRequestPayload, FetchRequestPayload, Message, ChecksumError, ProduceResponsePayload, FetchResponsePayload, OffsetAndMessage, BrokerMetadata, TopicMetadata, PartitionMetadata, TopicAndPartition, |