From 5a14bd8c947251d1a8f848175cc3cf2b07af3411 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Tue, 5 Apr 2016 09:34:48 -0700 Subject: Update imports from kafka.common -> kafka.errors / kafka.structs --- test/test_protocol.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) (limited to 'test/test_protocol.py') diff --git a/test/test_protocol.py b/test/test_protocol.py index 1d91e7d..d705e3a 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -7,21 +7,21 @@ from mock import patch, sentinel from . import unittest from kafka.codec import has_snappy, gzip_decode, snappy_decode -from kafka.common import ( +from kafka.errors import ( + ChecksumError, KafkaUnavailableError, UnsupportedCodecError, + ConsumerFetchSizeTooSmall, ProtocolError) +from kafka.protocol import ( + ATTRIBUTE_CODEC_MASK, CODEC_NONE, CODEC_GZIP, CODEC_SNAPPY, KafkaProtocol, + create_message, create_gzip_message, create_snappy_message, + create_message_set) +from kafka.structs import ( OffsetRequestPayload, OffsetResponsePayload, OffsetCommitRequestPayload, OffsetCommitResponsePayload, OffsetFetchRequestPayload, OffsetFetchResponsePayload, ProduceRequestPayload, ProduceResponsePayload, FetchRequestPayload, FetchResponsePayload, - Message, ChecksumError, OffsetAndMessage, BrokerMetadata, - KafkaUnavailableError, UnsupportedCodecError, ConsumerFetchSizeTooSmall, - ProtocolError, ConsumerMetadataResponse -) -from kafka.protocol import ( - ATTRIBUTE_CODEC_MASK, CODEC_NONE, CODEC_GZIP, CODEC_SNAPPY, KafkaProtocol, - create_message, create_gzip_message, create_snappy_message, - create_message_set -) + Message, OffsetAndMessage, BrokerMetadata, ConsumerMetadataResponse) + class TestProtocol(unittest.TestCase): def test_create_message(self): -- cgit v1.2.1