From 6c0d9137eae9bd875a7a7ab70d3285dd74f6701b Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@rd.io>
Date: Mon, 1 Sep 2014 02:25:05 -0700
Subject: if error code is not recognized in check_error: raise UnknownError

---
 kafka/common.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'kafka/common.py')

diff --git a/kafka/common.py b/kafka/common.py
index e1713cf..6b73410 100644
--- a/kafka/common.py
+++ b/kafka/common.py
@@ -76,6 +76,9 @@ class KafkaError(RuntimeError):
 class BrokerResponseError(KafkaError):
     pass
 
+class NoError(BrokerResponseError):
+    errno = 0
+    message = 'SUCCESS'
 
 class UnknownError(BrokerResponseError):
     errno = -1
@@ -197,6 +200,7 @@ class UnsupportedCodecError(KafkaError):
 
 kafka_errors = {
     -1 : UnknownError,
+    0  : NoError,
     1  : OffsetOutOfRangeError,
     2  : InvalidMessageError,
     3  : UnknownTopicOrPartitionError,
@@ -214,7 +218,7 @@ kafka_errors = {
 
 
 def check_error(response):
-    error = kafka_errors.get(response.error)
-    if error:
+    error = kafka_errors.get(response.error, UnknownError)
+    if error is not NoError:
         raise error(response)
 
-- 
cgit v1.2.1