diff --git a/bec_widgets/applications/launch_window.py b/bec_widgets/applications/launch_window.py index 4a62dc2f..56c13b8e 100644 --- a/bec_widgets/applications/launch_window.py +++ b/bec_widgets/applications/launch_window.py @@ -542,7 +542,7 @@ class LaunchWindow(BECMainWindow): remaining_connections = [ connection for connection in connections.values() if connection.parent_id != self.gui_id ] - return len(remaining_connections) <= 1 + return len(remaining_connections) <= 2 def _turn_off_the_lights(self, connections: dict): """ diff --git a/tests/end-2-end/test_rpc_widgets_e2e.py b/tests/end-2-end/test_rpc_widgets_e2e.py index b6dc0b8c..175e642d 100644 --- a/tests/end-2-end/test_rpc_widgets_e2e.py +++ b/tests/end-2-end/test_rpc_widgets_e2e.py @@ -79,7 +79,7 @@ def test_available_widgets(qtbot, connected_client_gui_obj): gui = connected_client_gui_obj dock_area = gui.bec # Number of top level widgets, should be 4 - top_level_widgets_count = 4 + top_level_widgets_count = 8 assert len(gui._server_registry) == top_level_widgets_count # Number of widgets with parent_id == None, should be 2 widgets = [ diff --git a/tests/unit_tests/test_launch_window.py b/tests/unit_tests/test_launch_window.py index defc6663..4cad21e5 100644 --- a/tests/unit_tests/test_launch_window.py +++ b/tests/unit_tests/test_launch_window.py @@ -102,7 +102,15 @@ def test_launch_window_launch_plugin_auto_update(bec_launch_window): [ ({}, False), ({"launcher": mock.MagicMock()}, False), - ({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, True), + ({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, False), + ( + { + "launcher": mock.MagicMock(), + "dock_area": mock.MagicMock(), + "scan_progress": mock.MagicMock(), + }, + True, + ), ], ) def test_gui_server_turns_off_the_lights(bec_launch_window, connections, hide): @@ -132,7 +140,15 @@ def test_gui_server_turns_off_the_lights(bec_launch_window, connections, hide): [ ({}, True), ({"launcher": mock.MagicMock()}, True), - ({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, False), + ({"launcher": mock.MagicMock(), "dock_area": mock.MagicMock()}, True), + ( + { + "launcher": mock.MagicMock(), + "dock_area": mock.MagicMock(), + "scan_progress": mock.MagicMock(), + }, + False, + ), ], ) def test_launch_window_closes(bec_launch_window, connections, close_called):