mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-12-29 18:31:17 +01:00
96 lines
3.7 KiB
Python
96 lines
3.7 KiB
Python
# pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
|
|
|
|
import os
|
|
|
|
import pytest
|
|
from bec_lib.endpoints import MessageEndpoints
|
|
|
|
import bec_widgets
|
|
from bec_widgets.applications.launch_window import LaunchWindow
|
|
from bec_widgets.widgets.containers.auto_update.auto_updates import AutoUpdates
|
|
from bec_widgets.widgets.containers.main_window.main_window import BECMainWindow, UILaunchWindow
|
|
|
|
from .client_mocks import mocked_client
|
|
|
|
base_path = os.path.dirname(bec_widgets.__file__)
|
|
|
|
|
|
@pytest.fixture
|
|
def bec_launch_window(qtbot, mocked_client):
|
|
widget = LaunchWindow(client=mocked_client)
|
|
qtbot.addWidget(widget)
|
|
qtbot.waitExposed(widget)
|
|
yield widget
|
|
|
|
|
|
def test_launch_window_initialization(bec_launch_window):
|
|
assert isinstance(bec_launch_window, BECMainWindow)
|
|
|
|
|
|
def test_launch_window_launch_ui_file(bec_launch_window):
|
|
# Mock the file dialog to return a specific UI file path
|
|
ui_file_path = os.path.join(
|
|
base_path, "widgets/control/device_control/positioner_box/positioner_box/positioner_box.ui"
|
|
)
|
|
bec_launch_window._open_custom_ui_file = lambda: ui_file_path
|
|
|
|
# Call the method to launch the custom UI file
|
|
res = bec_launch_window.launch("custom_ui_file", ui_file=ui_file_path)
|
|
|
|
assert isinstance(res, UILaunchWindow)
|
|
|
|
# Check if the custom UI file was launched correctly
|
|
assert res.object_name == "positioner_box"
|
|
assert res.windowTitle() == "BEC - positioner_box"
|
|
|
|
# We need to manually close the launched window as it is not registered with qtbot.
|
|
# In real usage, the GUIServer would handle this in the sigint handler in case of a ctrl-c initiated shutdown.
|
|
res.close()
|
|
res.deleteLater()
|
|
|
|
|
|
def test_launch_window_launch_ui_file_raises_for_qmainwindow(bec_launch_window):
|
|
# Mock the file dialog to return a specific UI file path
|
|
# the selected file must contain a QMainWindow widget but can be any file
|
|
ui_file_path = os.path.join(base_path, "applications/alignment/alignment_1d/alignment_1d.ui")
|
|
|
|
# Call the method to launch the custom UI file
|
|
with pytest.raises(ValueError) as excinfo:
|
|
bec_launch_window.launch("custom_ui_file", ui_file=ui_file_path)
|
|
|
|
assert "Loading a QMainWindow from a UI file is currently not supported." in str(excinfo.value)
|
|
|
|
|
|
def test_launch_window_launch_default_auto_update(bec_launch_window):
|
|
# Mock the auto update selection
|
|
bec_launch_window.tile_auto_update.selector.setCurrentText("Default")
|
|
|
|
# Call the method to launch the auto update
|
|
res = bec_launch_window._open_auto_update()
|
|
|
|
assert isinstance(res, AutoUpdates)
|
|
assert res.windowTitle() == "BEC - AutoUpdates"
|
|
|
|
# We need to manually close the launched window as it is not registered with qtbot.
|
|
# In real usage, the GUIServer would handle this in the sigint handler in case of a ctrl-c initiated shutdown.
|
|
res.close()
|
|
res.deleteLater()
|
|
|
|
|
|
def test_launch_window_launch_plugin_auto_update(bec_launch_window):
|
|
class PluginAutoUpdate(AutoUpdates): ...
|
|
|
|
bec_launch_window.available_auto_updates = {"PluginAutoUpdate": PluginAutoUpdate}
|
|
bec_launch_window.tile_auto_update.selector.clear()
|
|
bec_launch_window.tile_auto_update.selector.addItems(
|
|
list(bec_launch_window.available_auto_updates.keys()) + ["Default"]
|
|
)
|
|
bec_launch_window.tile_auto_update.selector.setCurrentText("PluginAutoUpdate")
|
|
res = bec_launch_window._open_auto_update()
|
|
assert isinstance(res, PluginAutoUpdate)
|
|
assert res.windowTitle() == "BEC - PluginAutoUpdate"
|
|
# We need to manually close the launched window as it is not registered with qtbot.
|
|
# In real usage, the GUIServer would handle this in the sigint handler in case of a ctrl-c initiated shutdown.
|
|
res.close()
|
|
res.deleteLater()
|