示例#1
0
async def test_dispatch_providers_with_error(route, event_loop):
    dispatcher = LoaferDispatcher([route])
    dispatcher.process_route = CoroutineMock(side_effect=ValueError)
    with pytest.raises(ValueError):
        await dispatcher.dispatch_providers(event_loop, forever=False)

    assert dispatcher.process_route.called
    dispatcher.process_route.assert_called_once_with(route)
示例#2
0
async def test_dispatch_providers(route, event_loop):
    dispatcher = LoaferDispatcher([route])
    dispatcher.process_route = CoroutineMock()
    dispatcher.stop_providers = Mock()
    await dispatcher.dispatch_providers(event_loop, forever=False)

    assert dispatcher.process_route.called
    dispatcher.process_route.assert_called_once_with(route)