async def test_commit_tps__ProducerSendError(self, *, consumer):
consumer._handle_attached = Mock(name='_handle_attached')
exc = consumer._handle_attached.side_effect = ProducerSendError()
consumer.crash = AsyncMock(name='crash')
consumer._filter_committable_offsets = Mock(name='filt')
consumer._filter_committable_offsets.return_value = {
TP1: 4,
TP2: 30,
}
await consumer._commit_tps({TP1, TP2})
consumer.crash.assert_called_once_with(exc)
async def test_commit_tps__ProducerSendError(self, *, consumer):
consumer._handle_attached = Mock(name="_handle_attached")
exc = consumer._handle_attached.side_effect = ProducerSendError()
consumer.crash = AsyncMock(name="crash")
consumer._filter_committable_offsets = Mock(name="filt")
consumer._filter_committable_offsets.return_value = {
TP1: 4,
TP2: 30,
}
await consumer._commit_tps(
{TP1, TP2},
start_new_transaction=True,
)
consumer.crash.assert_called_once_with(exc)
