commit dc6a98c30566184eb60cfccfca7fca5ca494869d Author: Illia Volochii illia.volochii@gmail.com Date: Mon Apr 27 19:18:24 2020 +0300
Commit a missing file --- test/unit/async_util.py | 26 ++++++++++++++++++++++++++ test/unit/connection/authentication.py | 2 +- test/unit/connection/connect.py | 2 +- test/unit/control/controller.py | 2 +- 4 files changed, 29 insertions(+), 3 deletions(-)
diff --git a/test/unit/async_util.py b/test/unit/async_util.py new file mode 100644 index 00000000..aa8a6dd0 --- /dev/null +++ b/test/unit/async_util.py @@ -0,0 +1,26 @@ +import asyncio +import functools + + +def async_test(func): + @functools.wraps(func) + def wrapper(*args, **kwargs): + loop = asyncio.new_event_loop() + try: + result = loop.run_until_complete(func(*args, **kwargs)) + finally: + loop.close() + return result + return wrapper + + +def coro_func_returning_value(return_value): + async def coroutine_func(*args, **kwargs): + return return_value + return coroutine_func + + +def coro_func_raising_exc(exc): + async def coroutine_func(*args, **kwargs): + raise exc + return coroutine_func diff --git a/test/unit/connection/authentication.py b/test/unit/connection/authentication.py index 596fa50c..8df38c8f 100644 --- a/test/unit/connection/authentication.py +++ b/test/unit/connection/authentication.py @@ -18,7 +18,7 @@ from unittest.mock import patch
from stem.response import ControlMessage from stem.util import log -from test.unit.util.asynchronous import ( +from test.unit.async_util import ( async_test, coro_func_raising_exc, coro_func_returning_value, diff --git a/test/unit/connection/connect.py b/test/unit/connection/connect.py index d2a22f18..aa91736b 100644 --- a/test/unit/connection/connect.py +++ b/test/unit/connection/connect.py @@ -11,7 +11,7 @@ import stem.socket
from unittest.mock import Mock, patch
-from test.unit.util.asynchronous import coro_func_raising_exc, coro_func_returning_value +from test.unit.async_util import coro_func_raising_exc, coro_func_returning_value
class TestConnect(unittest.TestCase): diff --git a/test/unit/control/controller.py b/test/unit/control/controller.py index d09b5ca8..99bd5f19 100644 --- a/test/unit/control/controller.py +++ b/test/unit/control/controller.py @@ -21,7 +21,7 @@ from stem import ControllerError, DescriptorUnavailable, InvalidArguments, Inval from stem.control import MALFORMED_EVENTS, _parse_circ_path, Listener, Controller, EventType from stem.response import ControlMessage from stem.exit_policy import ExitPolicy -from test.unit.util.asynchronous import ( +from test.unit.async_util import ( async_test, coro_func_raising_exc, coro_func_returning_value,