summaryrefslogtreecommitdiff
path: root/test/test_protocol.py
diff options
context:
space:
mode:
authorDana Powers <dana.powers@rd.io>2014-09-01 17:03:46 -0700
committerDana Powers <dana.powers@rd.io>2014-09-01 18:04:10 -0700
commit945ecbcee7d2844ebbfa407b1542109fd8518cde (patch)
treeee996c536dd4458149f5c53257ed28b2fe5b1212 /test/test_protocol.py
parentb260b356b23802a595336c554d6ea044c9be0a79 (diff)
downloadkafka-python-945ecbcee7d2844ebbfa407b1542109fd8518cde.tar.gz
Use standard exceptions in client._get_leader_for_partition()
- drop custom PartitionUnavailable exception - raise UnknownTopicOrPartitionError or LeaderNotAvailableError - add tests for exception raises
Diffstat (limited to 'test/test_protocol.py')
-rw-r--r--test/test_protocol.py4
1 files changed, 2 insertions, 2 deletions
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,