async def test_transfer_request_finisher_run(config, mocker): """Test the TransferRequestFinisher does the work the transfer_request_finisher should do.""" logger_mock = mocker.MagicMock() p = TransferRequestFinisher(config, logger_mock) p._do_work = AsyncMock() await p.run() p._do_work.assert_called()
async def test_transfer_request_finisher_run_exception(config, mocker): """Test an error doesn't kill the TransferRequestFinisher.""" logger_mock = mocker.MagicMock() p = TransferRequestFinisher(config, logger_mock) p.last_work_end_timestamp = None p._do_work = AsyncMock() p._do_work.side_effect = [Exception("bad thing happen!")] await p.run() p._do_work.assert_called() assert p.last_work_end_timestamp