mirror of
https://github.com/bec-project/bec_widgets.git
synced 2026-03-04 16:02:51 +01:00
130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
from contextlib import contextmanager
|
|
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
from bec_widgets.cli.client import BECDockArea
|
|
from bec_widgets.cli.client_utils import BECGuiClient, _start_plot_process
|
|
|
|
|
|
@pytest.fixture
|
|
def cli_dock_area():
|
|
dock_area = BECDockArea(gui_id="test")
|
|
with mock.patch.object(dock_area, "_run_rpc") as mock_rpc_call:
|
|
with mock.patch.object(dock_area, "_gui_is_alive", return_value=True):
|
|
yield dock_area, mock_rpc_call
|
|
|
|
|
|
def test_rpc_call_new_dock(cli_dock_area):
|
|
dock_area, mock_rpc_call = cli_dock_area
|
|
dock_area.new(name="test")
|
|
mock_rpc_call.assert_called_with("new", name="test")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"config, call_config",
|
|
[
|
|
(None, None),
|
|
("/path/to/config.yml", "/path/to/config.yml"),
|
|
({"key": "value"}, '{"key": "value"}'),
|
|
],
|
|
)
|
|
def test_client_utils_start_plot_process(config, call_config):
|
|
with mock.patch("bec_widgets.cli.client_utils.subprocess.Popen") as mock_popen:
|
|
_start_plot_process("gui_id", "bec", config, gui_class="AdvancedDockArea")
|
|
command = [
|
|
"bec-gui-server",
|
|
"--id",
|
|
"gui_id",
|
|
"--gui_class",
|
|
"AdvancedDockArea",
|
|
"--gui_class_id",
|
|
"bec",
|
|
"--hide",
|
|
]
|
|
if call_config:
|
|
command.extend(["--config", call_config])
|
|
mock_popen.assert_called_once_with(
|
|
command,
|
|
text=True,
|
|
start_new_session=True,
|
|
stdout=mock.ANY,
|
|
stderr=mock.ANY,
|
|
env=mock.ANY,
|
|
)
|
|
|
|
|
|
def test_client_utils_passes_client_config_to_server(bec_dispatcher):
|
|
"""
|
|
Test that the client config is passed to the server. This ensures that
|
|
changes to the client config (either through config files or plugins) are
|
|
reflected in the server.
|
|
"""
|
|
|
|
@contextmanager
|
|
def bec_client_mixin():
|
|
mixin = BECGuiClient()
|
|
mixin._client = bec_dispatcher.client
|
|
mixin._gui_id = "gui_id"
|
|
mixin._gui_is_alive = mock.MagicMock()
|
|
mixin._gui_is_alive.side_effect = [False, False, True]
|
|
|
|
try:
|
|
yield mixin
|
|
finally:
|
|
mixin.kill_server()
|
|
|
|
with bec_client_mixin() as mixin:
|
|
with mock.patch("bec_widgets.cli.client_utils._start_plot_process") as mock_start_plot:
|
|
mock_start_plot.return_value = [mock.MagicMock(), mock.MagicMock()]
|
|
mixin._start_server(
|
|
wait=False
|
|
) # the started event will not be set, wait=True would block forever
|
|
mock_start_plot.assert_called_once_with(
|
|
"gui_id",
|
|
gui_class_id="bec",
|
|
config=mixin._client._service_config.config,
|
|
logger=mock.ANY,
|
|
)
|
|
|
|
|
|
@contextmanager
|
|
def _no_wait_for_server(_client):
|
|
yield
|
|
|
|
|
|
@pytest.mark.parametrize("theme", ["light", "dark"])
|
|
def test_client_utils_apply_theme_explicit(theme):
|
|
gui = BECGuiClient()
|
|
launcher = mock.MagicMock()
|
|
|
|
with mock.patch.object(
|
|
BECGuiClient, "launcher", new_callable=mock.PropertyMock
|
|
) as launcher_prop:
|
|
launcher_prop.return_value = launcher
|
|
with mock.patch("bec_widgets.cli.client_utils.wait_for_server", _no_wait_for_server):
|
|
with mock.patch.object(gui, "_check_if_server_is_alive", return_value=True):
|
|
gui.change_theme(theme)
|
|
|
|
launcher._run_rpc.assert_called_once_with("change_theme", theme=theme)
|
|
|
|
|
|
@pytest.mark.parametrize("current_theme, expected_theme", [("light", "dark"), ("dark", "light")])
|
|
def test_client_utils_apply_theme_toggles_when_none(current_theme, expected_theme):
|
|
gui = BECGuiClient()
|
|
launcher = mock.MagicMock()
|
|
launcher._run_rpc.side_effect = [current_theme, None]
|
|
|
|
with mock.patch.object(
|
|
BECGuiClient, "launcher", new_callable=mock.PropertyMock
|
|
) as launcher_prop:
|
|
launcher_prop.return_value = launcher
|
|
with mock.patch("bec_widgets.cli.client_utils.wait_for_server", _no_wait_for_server):
|
|
with mock.patch.object(gui, "_check_if_server_is_alive", return_value=True):
|
|
gui.change_theme(None)
|
|
|
|
assert launcher._run_rpc.call_args_list == [
|
|
mock.call("fetch_theme"),
|
|
mock.call("change_theme", theme=expected_theme),
|
|
]
|