|
4 | 4 | import pytest |
5 | 5 | import requests |
6 | 6 | from bluesky_queueserver.manager.comms import zmq_single_request |
| 7 | +from bluesky_queueserver.manager.tests.common import re_manager_cmd # noqa: F401 |
7 | 8 | from bluesky_queueserver.manager.tests.common import set_qserver_zmq_encoding # noqa: F401 |
8 | 9 | from xprocess import ProcessStarter |
9 | 10 |
|
@@ -60,7 +61,11 @@ def fastapi_server_fs(xprocess): |
60 | 61 | to perform additional steps (such as setting environmental variables) before the server is started. |
61 | 62 | """ |
62 | 63 |
|
63 | | - def start(http_server_host=SERVER_ADDRESS, http_server_port=SERVER_PORT, api_key=API_KEY_FOR_TESTS): |
| 64 | + def start( |
| 65 | + http_server_host=SERVER_ADDRESS, |
| 66 | + http_server_port=SERVER_PORT, |
| 67 | + api_key=API_KEY_FOR_TESTS, |
| 68 | + ): |
64 | 69 | class Starter(ProcessStarter): |
65 | 70 | max_read_lines = 53 |
66 | 71 |
|
@@ -112,15 +117,27 @@ def add_plans_to_queue(): |
112 | 117 |
|
113 | 118 | user_group = _user_group |
114 | 119 | user = "HTTP unit test setup" |
115 | | - plan1 = {"name": "count", "args": [["det1", "det2"]], "kwargs": {"num": 10, "delay": 1}, "item_type": "plan"} |
| 120 | + plan1 = { |
| 121 | + "name": "count", |
| 122 | + "args": [["det1", "det2"]], |
| 123 | + "kwargs": {"num": 10, "delay": 1}, |
| 124 | + "item_type": "plan", |
| 125 | + } |
116 | 126 | plan2 = {"name": "count", "args": [["det1", "det2"]], "item_type": "plan"} |
117 | 127 | for plan in (plan1, plan2, plan2): |
118 | 128 | resp2, _ = zmq_single_request("queue_item_add", {"item": plan, "user": user, "user_group": user_group}) |
119 | 129 | assert resp2["success"] is True, str(resp2) |
120 | 130 |
|
121 | 131 |
|
122 | 132 | def request_to_json( |
123 | | - request_type, path, *, request_prefix="/api", api_key=API_KEY_FOR_TESTS, token=None, login=None, **kwargs |
| 133 | + request_type, |
| 134 | + path, |
| 135 | + *, |
| 136 | + request_prefix="/api", |
| 137 | + api_key=API_KEY_FOR_TESTS, |
| 138 | + token=None, |
| 139 | + login=None, |
| 140 | + **kwargs, |
124 | 141 | ): |
125 | 142 | if login: |
126 | 143 | auth = None |
|
0 commit comments