From 945ecbcee7d2844ebbfa407b1542109fd8518cde Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Mon, 1 Sep 2014 17:03:46 -0700 Subject: Use standard exceptions in client._get_leader_for_partition() - drop custom PartitionUnavailable exception - raise UnknownTopicOrPartitionError or LeaderNotAvailableError - add tests for exception raises --- test/test_protocol.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'test/test_protocol.py') diff --git a/test/test_protocol.py b/test/test_protocol.py index a92d20e..dc2411d 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -13,8 +13,8 @@ from kafka.common import ( ProduceRequest, FetchRequest, Message, ChecksumError, ProduceResponse, FetchResponse, OffsetAndMessage, BrokerMetadata, TopicMetadata, PartitionMetadata, TopicAndPartition, - KafkaUnavailableError, PartitionUnavailableError, - UnsupportedCodecError, ConsumerFetchSizeTooSmall, ProtocolError, + KafkaUnavailableError, UnsupportedCodecError, ConsumerFetchSizeTooSmall, + ProtocolError ) from kafka.codec import ( has_snappy, gzip_encode, gzip_decode, -- cgit v1.2.1