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)