From 9016280e0c443b38eff5b5bb6bb15d7dc136ca02 Mon Sep 17 00:00:00 2001
From: wyzula-jan <jan.wyzula@psi.ch>
Date: Tue, 9 Jun 2026 11:04:51 +0200
Subject: [PATCH] feat(dock-area): expose beamline state manager

---
 bec_widgets/widgets/containers/dock_area/dock_area.py |  5 +++++
 tests/unit_tests/test_dock_area.py                    | 11 ++++++++++-
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/bec_widgets/widgets/containers/dock_area/dock_area.py b/bec_widgets/widgets/containers/dock_area/dock_area.py
index 01e0656e..acd03392 100644
--- a/bec_widgets/widgets/containers/dock_area/dock_area.py
+++ b/bec_widgets/widgets/containers/dock_area/dock_area.py
@@ -385,6 +385,11 @@ class BECDockArea(DockAreaWidget):
             "bec_shell": (widget_icons["BECShell"], "Add BEC Shell", "BECShell"),
             "sbb_monitor": (widget_icons["SBBMonitor"], "Add SBB Monitor", "SBBMonitor"),
             "log_panel": (widget_icons["LogPanel"], "Add LogPanel", "LogPanel"),
+            "beamline_state_manager": (
+                widget_icons["BeamlineStateManager"],
+                "Add Beamline State Manager",
+                "BeamlineStateManager",
+            ),
         }
 
         # Create expandable menu actions (original behavior)
diff --git a/tests/unit_tests/test_dock_area.py b/tests/unit_tests/test_dock_area.py
index 35c1900b..10334ede 100644
--- a/tests/unit_tests/test_dock_area.py
+++ b/tests/unit_tests/test_dock_area.py
@@ -869,7 +869,14 @@ class TestToolbarFunctionality:
 
     def test_toolbar_utils_actions(self, advanced_dock_area):
         """Test utils toolbar actions trigger widget creation."""
-        utils_actions = ["queue", "terminal", "status", "progress_bar", "sbb_monitor"]
+        utils_actions = [
+            "queue",
+            "terminal",
+            "status",
+            "progress_bar",
+            "sbb_monitor",
+            "beamline_state_manager",
+        ]
 
         for action_name in utils_actions:
             with patch.object(advanced_dock_area, "new") as mock_new:
@@ -2428,6 +2435,7 @@ class TestFlatToolbarActions:
             "flat_terminal",
             "flat_bec_shell",
             "flat_sbb_monitor",
+            "flat_beamline_state_manager",
         ]
 
         for action_name in utils_actions:
@@ -2472,6 +2480,7 @@ class TestFlatToolbarActions:
             "flat_terminal": "BecConsole",
             "flat_bec_shell": "BECShell",
             "flat_sbb_monitor": "SBBMonitor",
+            "flat_beamline_state_manager": "BeamlineStateManager",
         }
 
         for action_name, widget_type in utils_action_mapping.items():