diff options
Diffstat (limited to 'test/test_consumer.py')
| -rw-r--r-- | test/test_consumer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index e5dd946..013529f 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -14,11 +14,11 @@ from kafka.structs import ( class TestKafkaConsumer(unittest.TestCase): def test_non_integer_partitions(self): with self.assertRaises(AssertionError): - SimpleConsumer(MagicMock(), 'group', 'topic', partitions = [ '0' ]) + SimpleConsumer(MagicMock(), 'group', 'topic', partitions=['0']) def test_session_timeout_larger_than_request_timeout_raises(self): with self.assertRaises(KafkaConfigurationError): - KafkaConsumer(bootstrap_servers='localhost:9092', session_timeout_ms=60000, request_timeout_ms=40000) + KafkaConsumer(bootstrap_servers='localhost:9092', api_version=(0,9), group_id='foo', session_timeout_ms=60000, request_timeout_ms=40000) def test_fetch_max_wait_larger_than_request_timeout_raises(self): with self.assertRaises(KafkaConfigurationError): |