Exemplo n.º 1
0
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()
Exemplo n.º 2
0
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