From dc0c825fd594c093a24543ff803d6c6564010e92 Mon Sep 17 00:00:00 2001
From: appel_c <christian.appel@psi.ch>
Date: Tue, 1 Oct 2024 21:28:49 +0200
Subject: [PATCH] test: add tests for scan_status_callback

---
 .../alignment/alignment_1d/alignment_1d.py    |  2 +-
 tests/unit_tests/test_alignment_gui.py        | 45 +++++++++++++++++++
 2 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 tests/unit_tests/test_alignment_gui.py

diff --git a/bec_widgets/applications/alignment/alignment_1d/alignment_1d.py b/bec_widgets/applications/alignment/alignment_1d/alignment_1d.py
index 6ed92c3c..81a87d41 100644
--- a/bec_widgets/applications/alignment/alignment_1d/alignment_1d.py
+++ b/bec_widgets/applications/alignment/alignment_1d/alignment_1d.py
@@ -90,7 +90,7 @@ class Alignment1D(BECWidget, QWidget):
     @Slot(dict, dict)
     def scan_status_callback(self, content: dict, _) -> None:
         """This slot allows to enable/disable the UI critical components when a scan is running"""
-        if content["status"] in ["running", "open"]:
+        if content["status"] in ["open"]:
             self.motion_is_active.emit(True)
             self.enable_ui(False)
         elif content["status"] in ["aborted", "halted", "closed"]:
diff --git a/tests/unit_tests/test_alignment_gui.py b/tests/unit_tests/test_alignment_gui.py
new file mode 100644
index 00000000..6fbdcf4a
--- /dev/null
+++ b/tests/unit_tests/test_alignment_gui.py
@@ -0,0 +1,45 @@
+from unittest import mock
+
+import pytest
+from bec_lib import messages
+
+from bec_widgets.applications.alignment.alignment_1d.alignment_1d import Alignment1D
+
+from .client_mocks import mocked_client
+from .conftest import create_widget
+
+
+@pytest.fixture(scope="function")
+def alignment_1d(qtbot, mocked_client):
+    """Fixture for Alignment1D widget"""
+    with mock.patch(
+        "bec_widgets.applications.alignment.alignment_1d.alignment_1d.Alignment1D.init_ui"
+    ):
+        widget = create_widget(qtbot, Alignment1D, client=mocked_client)
+        yield widget
+
+
+def test_scan_status_callback(qtbot, alignment_1d):
+    """Test the scan status callback."""
+    container = []
+
+    def callback(*args, **kwargs):
+        """Callback function to store signal calls."""
+        container.append(args)
+
+    alignment_1d.motion_is_active.connect(callback)
+    with mock.patch.object(alignment_1d, "enable_ui") as mock_enable_ui:
+        for status in ["open", "aborted", "halted", "closed"]:
+            msg = messages.ScanStatusMessage(scan_id="tmp_id", status=status, info={})
+            alignment_1d.scan_status_callback(msg.content, {})
+            qtbot.wait(100)
+            if status in ["open"]:
+                assert mock.call(False) == mock_enable_ui.call_args_list[-1]
+                assert container[-1] == (True,)
+            else:
+                assert mock.call(True) == mock_enable_ui.call_args_list[-1]
+                assert container[-1] == (False,)
+        container.clear()
+        alignment_1d.scan_status_callback({"status": "invalid_status"}, {})
+        qtbot.wait(100)
+        assert not container