From 83001a0d8267e1320549b07032857dcf46ecd293 Mon Sep 17 00:00:00 2001
From: wyzula-jan <jan.wyzula@psi.ch>
Date: Fri, 21 Jun 2024 13:28:14 +0200
Subject: [PATCH] test(scan_control):e2e tests added

---
 .../device_combobox/device_combobox.py        |  2 +-
 .../device_line_edit/device_line_edit.py      |  2 +-
 tests/end-2-end/test_scan_control_e2e.py      | 71 +++++++++++++++++++
 3 files changed, 73 insertions(+), 2 deletions(-)
 create mode 100644 tests/end-2-end/test_scan_control_e2e.py

diff --git a/bec_widgets/widgets/device_inputs/device_combobox/device_combobox.py b/bec_widgets/widgets/device_inputs/device_combobox/device_combobox.py
index f5e4e451..64887007 100644
--- a/bec_widgets/widgets/device_inputs/device_combobox/device_combobox.py
+++ b/bec_widgets/widgets/device_inputs/device_combobox/device_combobox.py
@@ -89,4 +89,4 @@ class DeviceComboBox(DeviceInputBase, QComboBox):
 
     def closeEvent(self, event):
         super().cleanup()
-        QComboBox().closeEvent(event)
\ No newline at end of file
+        QComboBox().closeEvent(event)
diff --git a/bec_widgets/widgets/device_inputs/device_line_edit/device_line_edit.py b/bec_widgets/widgets/device_inputs/device_line_edit/device_line_edit.py
index a3756df7..7d346bf4 100644
--- a/bec_widgets/widgets/device_inputs/device_line_edit/device_line_edit.py
+++ b/bec_widgets/widgets/device_inputs/device_line_edit/device_line_edit.py
@@ -101,4 +101,4 @@ class DeviceLineEdit(DeviceInputBase, QLineEdit):
 
     def closeEvent(self, event):
         super().cleanup()
-        QLineEdit().closeEvent(event)
\ No newline at end of file
+        QLineEdit().closeEvent(event)
diff --git a/tests/end-2-end/test_scan_control_e2e.py b/tests/end-2-end/test_scan_control_e2e.py
new file mode 100644
index 00000000..c0519f67
--- /dev/null
+++ b/tests/end-2-end/test_scan_control_e2e.py
@@ -0,0 +1,71 @@
+import time
+
+import pytest
+
+from bec_widgets.utils.widget_io import WidgetIO
+from bec_widgets.widgets.scan_control import ScanControl
+
+
+@pytest.fixture(scope="function")
+def scan_control(qtbot, bec_client_lib):  # , mock_dev):
+    widget = ScanControl(client=bec_client_lib)
+    qtbot.addWidget(widget)
+    qtbot.waitExposed(widget)
+    yield widget
+
+
+def test_scan_control_populate_scans_e2e(scan_control):
+    expected_scans = [
+        "grid_scan",
+        "fermat_scan",
+        "round_scan",
+        "cont_line_scan",
+        "cont_line_fly_scan",
+        "round_scan_fly",
+        "round_roi_scan",
+        "time_scan",
+        "monitor_scan",
+        "acquire",
+        "line_scan",
+    ]
+    items = [
+        scan_control.comboBox_scan_selection.itemText(i)
+        for i in range(scan_control.comboBox_scan_selection.count())
+    ]
+    assert scan_control.comboBox_scan_selection.count() == len(expected_scans)
+    assert sorted(items) == sorted(expected_scans)
+
+
+def test_run_line_scan_with_parameters_e2e(scan_control, bec_client_lib, qtbot):
+    client = bec_client_lib
+    queue = client.queue
+
+    scan_name = "line_scan"
+    kwargs = {"exp_time": 0.01, "steps": 10, "relative": True, "burst_at_each_point": 1}
+    args = {"device": "samx", "start": -5, "stop": 5}
+
+    scan_control.comboBox_scan_selection.setCurrentText(scan_name)
+
+    # Set kwargs in the UI
+    for kwarg_box in scan_control.kwarg_boxes:
+        for widget in kwarg_box.widgets:
+            for key, value in kwargs.items():
+                if widget.arg_name == key:
+                    WidgetIO.set_value(widget, value)
+                    break
+    # Set args in the UI
+    for widget in scan_control.arg_box.widgets:
+        for key, value in args.items():
+            if widget.arg_name == key:
+                WidgetIO.set_value(widget, value)
+                break
+
+    # Run the scan
+    scan_control.button_run_scan.click()
+    time.sleep(2)
+
+    last_scan = queue.scan_storage.storage[-1]
+    assert last_scan.status_message.info["scan_name"] == scan_name
+    assert last_scan.status_message.info["exp_time"] == kwargs["exp_time"]
+    assert last_scan.status_message.info["scan_motors"] == [args["device"]]
+    assert last_scan.status_message.info["num_points"] == kwargs["steps"]