Skip to content

Commit e039108

Browse files
committed
fix tests
1 parent a9fdfb0 commit e039108

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

test/test_consumer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def test_non_integer_partitions(self):
1818

1919
def test_session_timeout_larger_than_request_timeout_raises(self):
2020
with self.assertRaises(KafkaConfigurationError):
21-
KafkaConsumer(bootstrap_servers='localhost:9092', group_id='foo', session_timeout_ms=60000, request_timeout_ms=40000)
21+
KafkaConsumer(bootstrap_servers='localhost:9092', api_version=(0,9), group_id='foo', session_timeout_ms=60000, request_timeout_ms=40000)
2222

2323
def test_fetch_max_wait_larger_than_request_timeout_raises(self):
2424
with self.assertRaises(KafkaConfigurationError):

test/test_coordinator.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,8 @@ def test_autocommit_enable_api_version(client, api_version):
4444
coordinator = ConsumerCoordinator(client, SubscriptionState(),
4545
Metrics(),
4646
enable_auto_commit=True,
47+
session_timeout_ms=30000, # session_timeout_ms and max_poll_interval_ms
48+
max_poll_interval_ms=30000, # should be the same to avoid KafkaConfigurationError
4749
group_id='foobar',
4850
api_version=api_version)
4951
if api_version < (0, 8, 1):
@@ -362,6 +364,8 @@ def test_maybe_auto_commit_offsets_sync(mocker, api_version, group_id, enable,
362364
coordinator = ConsumerCoordinator(client, SubscriptionState(),
363365
Metrics(),
364366
api_version=api_version,
367+
session_timeout_ms=30000,
368+
max_poll_interval_ms=30000,
365369
enable_auto_commit=enable,
366370
group_id=group_id)
367371
commit_sync = mocker.patch.object(coordinator, 'commit_offsets_sync',

0 commit comments

Comments
 (0)