bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-13 19:21:50 +02:00
Code Activity

test: fix tests for launcher close / hide behavior

Browse Source
This commit is contained in:
wakonig_k
2025-04-29 10:08:17 +02:00
parent 6e7920c119
commit 23fee22ef8
2 changed files with 60 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
tests/unit_tests/test_launch_window.py
View File

@ -1,6 +1,7 @@
# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
import os
from unittest import mock
import pytest
from bec_lib.endpoints import MessageEndpoints
@ -93,3 +94,62 @@ def test_launch_window_launch_plugin_auto_update(bec_launch_window):
# In real usage, the GUIServer would handle this in the sigint handler in case of a ctrl-c initiated shutdown.
res.close()
res.deleteLater()
@pytest.mark.parametrize(
"connections, hide",
[
({}, False),
({"launcher": mock.MagicMock()}, False),
({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, True),
],
)
def test_gui_server_turns_off_the_lights(bec_launch_window, connections, hide):
with (
mock.patch.object(bec_launch_window, "show") as mock_show,
mock.patch.object(bec_launch_window, "activateWindow") as mock_activate_window,
mock.patch.object(bec_launch_window, "raise_") as mock_raise,
mock.patch.object(bec_launch_window, "hide") as mock_hide,
mock.patch.object(
bec_launch_window.app, "setQuitOnLastWindowClosed"
) as mock_set_quit_on_last_window_closed,
):
bec_launch_window._turn_off_the_lights(connections)
if hide:
mock_hide.assert_called_once()
mock_set_quit_on_last_window_closed.assert_called_once_with(False)
else:
mock_show.assert_called_once()
mock_activate_window.assert_called_once()
mock_raise.assert_called_once()
mock_set_quit_on_last_window_closed.assert_called_once_with(True)
@pytest.mark.parametrize(
"connections, close_called",
[
({}, True),
({"launcher": mock.MagicMock()}, True),
({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, False),
],
)
def test_launch_window_closes(bec_launch_window, connections, close_called):
"""
Test that the close event is handled correctly based on the connections.
If there are no connections or only the launcher connection, the window should close.
If there are other connections, the window should hide instead of closing.
"""
close_event = mock.MagicMock()
with mock.patch.object(
bec_launch_window.register, "list_all_connections", return_value=connections
):
with mock.patch.object(bec_launch_window, "hide") as mock_hide:
bec_launch_window.closeEvent(close_event)
if close_called:
mock_hide.assert_not_called()
close_event.accept.assert_called_once()
else:
mock_hide.assert_called_once()
close_event.accept.assert_not_called()
close_event.ignore.assert_called_once()

24
tests/unit_tests/test_rpc_server.py
View File

@ -1,5 +1,4 @@
import argparse
from unittest import mock
import pytest
from bec_lib.service_config import ServiceConfig
@ -31,26 +30,3 @@ def test_gui_server_get_service_config(gui_server):
Test that the server is started with the correct arguments.
"""
assert gui_server._get_service_config().config is ServiceConfig().config
@pytest.mark.parametrize(
"connections, hide",
[
({}, False),
({"launcher": mock.MagicMock()}, False),
({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, True),
],
)
def test_gui_server_turns_off_the_lights(gui_server, connections, hide):
with mock.patch.object(gui_server, "launcher_window") as mock_launcher_window:
with mock.patch.object(gui_server, "app") as mock_app:
gui_server._turn_off_the_lights(connections)
if not hide:
mock_launcher_window.show.assert_called_once()
mock_launcher_window.activateWindow.assert_called_once()
mock_launcher_window.raise_.assert_called_once()
mock_app.setQuitOnLastWindowClosed.assert_called_once_with(True)
else:
mock_launcher_window.hide.assert_called_once()
mock_app.setQuitOnLastWindowClosed.assert_called_once_with(False)