From 7470cade6bb8629d17541e136527369f9d2ec387 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Thu, 10 Dec 2015 10:57:27 -0800 Subject: Convert OffsetCommit and OffsetFetch protocol encode/decode --- test/test_client_integration.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test/test_client_integration.py') 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) -- cgit v1.2.1