diff --git a/bluesky_queueserver_api/tests/common.py b/bluesky_queueserver_api/tests/common.py index 745cd9b..d5bf571 100644 --- a/bluesky_queueserver_api/tests/common.py +++ b/bluesky_queueserver_api/tests/common.py @@ -8,6 +8,7 @@ ip_kernel_simple_client, re_manager, re_manager_cmd, + re_manager_factory, set_qserver_zmq_address, set_qserver_zmq_encoding, set_qserver_zmq_public_key, diff --git a/bluesky_queueserver_api/tests/test_api.py b/bluesky_queueserver_api/tests/test_api.py index bc1b69f..2d3b7c7 100644 --- a/bluesky_queueserver_api/tests/test_api.py +++ b/bluesky_queueserver_api/tests/test_api.py @@ -23,6 +23,7 @@ ip_kernel_simple_client, re_manager, re_manager_cmd, + re_manager_factory, ) _user, _user_group = "Test User", default_user_group diff --git a/bluesky_queueserver_api/tests/test_api_fs.py b/bluesky_queueserver_api/tests/test_api_fs.py index 9add09c..d5bde71 100644 --- a/bluesky_queueserver_api/tests/test_api_fs.py +++ b/bluesky_queueserver_api/tests/test_api_fs.py @@ -9,11 +9,12 @@ from ..comm_base import RequestParameterError from .common import fastapi_server_fs # noqa: F401 -from .common import re_manager_cmd # noqa: F401 -from .common import ( +from .common import ( # noqa: F401 _is_async, _select_re_manager_api, instantiate_re_api_class, + re_manager_cmd, + re_manager_factory, set_qserver_zmq_address, set_qserver_zmq_public_key, setup_server_with_config_file, diff --git a/bluesky_queueserver_api/tests/test_comm.py b/bluesky_queueserver_api/tests/test_comm.py index 0e8f380..abf5770 100644 --- a/bluesky_queueserver_api/tests/test_comm.py +++ b/bluesky_queueserver_api/tests/test_comm.py @@ -14,6 +14,7 @@ API_KEY_FOR_TESTS, re_manager, re_manager_cmd, + re_manager_factory, set_qserver_zmq_address, set_qserver_zmq_encoding, set_qserver_zmq_public_key, diff --git a/bluesky_queueserver_api/tests/test_security.py b/bluesky_queueserver_api/tests/test_security.py index e725285..a29f694 100644 --- a/bluesky_queueserver_api/tests/test_security.py +++ b/bluesky_queueserver_api/tests/test_security.py @@ -5,7 +5,13 @@ import pytest from .common import fastapi_server_fs # noqa: F401 -from .common import _is_async, _select_re_manager_api, re_manager, re_manager_cmd # noqa: F401 +from .common import ( # noqa: F401 + _is_async, + _select_re_manager_api, + re_manager, + re_manager_cmd, + re_manager_factory, +) # fmt: off