Skip to content

Commit 3d3922c

Browse files
committed
Add async_raise
1 parent b695cdf commit 3d3922c

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/galaxy/unittest/mock.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -21,11 +21,19 @@ def coroutine_mock():
2121
corofunc.coro = coro
2222
return corofunc
2323

24+
2425
async def skip_loop(iterations=1):
2526
for _ in range(iterations):
2627
await asyncio.sleep(0)
2728

2829

2930
async def async_return_value(return_value, loop_iterations_delay=0):
30-
await skip_loop(loop_iterations_delay)
31+
if loop_iterations_delay > 0:
32+
await skip_loop(loop_iterations_delay)
3133
return return_value
34+
35+
36+
async def async_raise(error, loop_iterations_delay=0):
37+
if loop_iterations_delay > 0:
38+
await skip_loop(loop_iterations_delay)
39+
raise error

0 commit comments

Comments
 (0)