@@ -206,8 +206,8 @@ def test_transaction_api():
206
206
# Any subsequent APIs will fail since init did not succeed.
207
207
with pytest .raises (KafkaException ) as ex :
208
208
p .begin_transaction ()
209
- assert ex .value .args [0 ].code () == KafkaError ._STATE
210
- assert ex .value .args [0 ].retriable () is False
209
+ assert ex .value .args [0 ].code () == KafkaError ._CONFLICT
210
+ assert ex .value .args [0 ].retriable () is True
211
211
assert ex .value .args [0 ].fatal () is False
212
212
assert ex .value .args [0 ].txn_requires_abort () is False
213
213
@@ -218,22 +218,22 @@ def test_transaction_api():
218
218
with pytest .raises (KafkaException ) as ex :
219
219
p .send_offsets_to_transaction ([TopicPartition ("topic" , 0 , 123 )],
220
220
group_metadata )
221
- assert ex .value .args [0 ].code () == KafkaError ._STATE
222
- assert ex .value .args [0 ].retriable () is False
221
+ assert ex .value .args [0 ].code () == KafkaError ._CONFLICT
222
+ assert ex .value .args [0 ].retriable () is True
223
223
assert ex .value .args [0 ].fatal () is False
224
224
assert ex .value .args [0 ].txn_requires_abort () is False
225
225
226
226
with pytest .raises (KafkaException ) as ex :
227
227
p .commit_transaction (0.5 )
228
- assert ex .value .args [0 ].code () == KafkaError ._STATE
229
- assert ex .value .args [0 ].retriable () is False
228
+ assert ex .value .args [0 ].code () == KafkaError ._CONFLICT
229
+ assert ex .value .args [0 ].retriable () is True
230
230
assert ex .value .args [0 ].fatal () is False
231
231
assert ex .value .args [0 ].txn_requires_abort () is False
232
232
233
233
with pytest .raises (KafkaException ) as ex :
234
234
p .abort_transaction (0.5 )
235
- assert ex .value .args [0 ].code () == KafkaError ._STATE
236
- assert ex .value .args [0 ].retriable () is False
235
+ assert ex .value .args [0 ].code () == KafkaError ._CONFLICT
236
+ assert ex .value .args [0 ].retriable () is True
237
237
assert ex .value .args [0 ].fatal () is False
238
238
assert ex .value .args [0 ].txn_requires_abort () is False
239
239
0 commit comments