bec/bec
0
0
Fork 0
mirror of https://github.com/bec-project/bec.git synced 2026-06-07 18:48:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor: make active workers private

Browse Source
This commit is contained in:
perl_d
2025-10-16 09:46:30 +02:00
committed by David Perl
parent f7353432d5
commit 4c17a5aa49
4 changed files with 24 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bec_ipython_client/tests/end-2-end/test_procedures_e2e.py
+6 -6
View File
@@ -62,7 +62,7 @@ def test_procedure_runner_spawns_worker(
client_logtool_and_manager: tuple[BECIPythonClient, "LogTestTool", ProcedureManager],
):
client, _, manager = client_logtool_and_manager
assert manager.active_workers == {}
assert manager._active_workers == {}
endpoint = MessageEndpoints.procedure_request()
msg = messages.ProcedureRequestMessage(
identifier="sleep", args_kwargs=((), {"time_s": 2}), queue="test"
@@ -77,8 +77,8 @@ def test_procedure_runner_spawns_worker(
manager.add_callback("test", cb)
client.connector.xadd(topic=endpoint, msg_dict=msg.model_dump())
_wait_while(lambda: manager.active_workers == {}, 5)
_wait_while(lambda: manager.active_workers != {}, 20)
_wait_while(lambda: manager._active_workers == {}, 5)
_wait_while(lambda: manager._active_workers != {}, 20)
assert logs != []
@@ -91,7 +91,7 @@ def test_happy_path_container_procedure_runner(
test_args = (1, 2, 3)
test_kwargs = {"a": "b", "c": "d"}
client, logtool, manager = client_logtool_and_manager
assert manager.active_workers == {}
assert manager._active_workers == {}
conn = client.connector
endpoint = MessageEndpoints.procedure_request()
msg = messages.ProcedureRequestMessage(
@@ -99,8 +99,8 @@ def test_happy_path_container_procedure_runner(
)
conn.xadd(topic=endpoint, msg_dict=msg.model_dump())
_wait_while(lambda: manager.active_workers == {}, 5)
_wait_while(lambda: manager.active_workers != {}, 20)
_wait_while(lambda: manager._active_workers == {}, 5)
_wait_while(lambda: manager._active_workers != {}, 20)
logtool.fetch()
assert logtool.is_present_in_any_message("procedure accepted: True, message:")