Skip to content

Commit db81d9f

Browse files
committed
Only test leader_epoch w/ 2.1+ brokers
1 parent 2077b71 commit db81d9f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

test/integration/test_producer_integration.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,12 @@ def test_transactional_producer_messages(kafka_broker):
187187
@pytest.mark.skipif(env_kafka_version() < (0, 11), reason="Idempotent producer requires broker >=0.11")
188188
def test_transactional_producer_offsets(kafka_broker):
189189
connect_str = ':'.join([kafka_broker.host, str(kafka_broker.port)])
190-
offsets = {TopicPartition('transactional_test_topic', 0): OffsetAndMetadata(0, 'metadata', 0)}
190+
# Setting leader_epoch only supported in 2.1+
191+
if env_kafka_version >= (2, 1):
192+
leader_epoch = 0
193+
else:
194+
leader_epoch = -1
195+
offsets = {TopicPartition('transactional_test_topic', 0): OffsetAndMetadata(0, 'metadata', leader_epoch)}
191196
with producer_factory(bootstrap_servers=connect_str, transactional_id='testing') as producer:
192197
producer.init_transactions()
193198
producer.begin_transaction()

0 commit comments

Comments
 (0)