From 3d3922c9651ec861f252bb48346d0b8b305b6579 Mon Sep 17 00:00:00 2001 From: Romuald Juchnowicz-Bierbasz Date: Mon, 18 Nov 2019 16:37:21 +0100 Subject: [PATCH] Add async_raise --- src/galaxy/unittest/mock.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/galaxy/unittest/mock.py b/src/galaxy/unittest/mock.py index b439671..da2e033 100644 --- a/src/galaxy/unittest/mock.py +++ b/src/galaxy/unittest/mock.py @@ -21,11 +21,19 @@ def coroutine_mock(): corofunc.coro = coro return corofunc + async def skip_loop(iterations=1): for _ in range(iterations): await asyncio.sleep(0) async def async_return_value(return_value, loop_iterations_delay=0): - await skip_loop(loop_iterations_delay) + if loop_iterations_delay > 0: + await skip_loop(loop_iterations_delay) return return_value + + +async def async_raise(error, loop_iterations_delay=0): + if loop_iterations_delay > 0: + await skip_loop(loop_iterations_delay) + raise error