Adjust consumer test error codes and timeouts

edenhill · edenhill · commit b520c8d92e8f · 2016-04-28T18:26:32.000-07:00
diff --git a/tests/test_Consumer.py b/tests/test_Consumer.py
@@ -16,6 +16,7 @@ def dummy_commit_cb (consumer, err, partitions):
         pass
 
     kc = Consumer({'group.id':'test', 'socket.timeout.ms':'100',
+                   'session.timeout.ms': 1000, # Avoid close() blocking too long
                    'on_commit': dummy_commit_cb})
 
     kc.subscribe(["test"])
@@ -45,7 +46,7 @@ def dummy_assign_revoke (consumer, partitions):
     try:
         kc.commit(async=False)
     except KafkaException as e:
-        assert e.args[0].code() in (KafkaError._TIMED_OUT, KafkaError._WAIT_COORD)
+        assert e.args[0].code() in (KafkaError._TIMED_OUT, KafkaError._NO_OFFSET)
 
     # Get current position, should all be invalid.
     kc.position(partitions)
