From 70a1de5456003438899c40a26a59d5e78e883732 Mon Sep 17 00:00:00 2001 From: appel_c Date: Thu, 13 Mar 2025 16:38:42 +0100 Subject: [PATCH] tests: fix e2e tests for namespace refactoring --- bec_widgets/cli/client_utils.py | 5 +++-- tests/end-2-end/test_bec_figure_rpc_e2e.py | 2 +- tests/end-2-end/test_rpc_register_e2e.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/bec_widgets/cli/client_utils.py b/bec_widgets/cli/client_utils.py index 42ca3ff9..974ab8ee 100644 --- a/bec_widgets/cli/client_utils.py +++ b/bec_widgets/cli/client_utils.py @@ -345,8 +345,9 @@ class BECGuiClient(RPCBase): rpc_client = RPCBase(gui_id=f"{self._gui_id}:window", parent=self) return rpc_client._run_rpc("_dump") - def start(self): - return self._start_server() + def start(self, wait: bool = True) -> None: + """Start the server and show the GUI window.""" + return self._start_server(wait=wait) def _do_show_all(self): rpc_client = RPCBase(gui_id=f"{self._gui_id}:window", parent=self) diff --git a/tests/end-2-end/test_bec_figure_rpc_e2e.py b/tests/end-2-end/test_bec_figure_rpc_e2e.py index dac1535a..da57862b 100644 --- a/tests/end-2-end/test_bec_figure_rpc_e2e.py +++ b/tests/end-2-end/test_bec_figure_rpc_e2e.py @@ -11,7 +11,7 @@ from bec_widgets.tests.utils import check_remote_data_size @pytest.fixture def connected_figure(connected_client_gui_obj): gui = connected_client_gui_obj - dock = gui.bec.new("dock") + dock = gui.window_list[0].new("dock") fig = dock.new(name="fig", widget="BECFigure") return fig diff --git a/tests/end-2-end/test_rpc_register_e2e.py b/tests/end-2-end/test_rpc_register_e2e.py index 28755d45..5db94d46 100644 --- a/tests/end-2-end/test_rpc_register_e2e.py +++ b/tests/end-2-end/test_rpc_register_e2e.py @@ -5,7 +5,7 @@ from bec_widgets.cli.client import BECFigure, BECImageShow, BECMotorMap, BECWave def test_rpc_register_list_connections(connected_client_gui_obj): gui = connected_client_gui_obj - fig = gui.bec.new("fig").new(name="fig", widget="BECFigure") + fig = gui.window_list[0].new("fig").new(name="fig", widget="BECFigure") plt = fig.plot(x_name="samx", y_name="bpm4i") im = fig.image("eiger")