test(dock_area): tests extended

tests/unit_tests/test_bec_dock.py

@@ -1,4 +1,5 @@
 # pylint: disable=missing-function-docstring, missing-module-docstring, unused-import
+from unittest.mock import MagicMock, patch
 
 import pytest
 
@@ -24,7 +25,7 @@ def test_bec_dock_area_init(bec_dock_area):
 
 
 def test_bec_dock_area_add_remove_dock(bec_dock_area, qtbot):
-    initial_count = len(bec_dock_area.docks)
+    initial_count = len(bec_dock_area.dock_area.docks)
 
     # Adding 3 docks
     d0 = bec_dock_area.add_dock()
@@ -32,28 +33,28 @@ def test_bec_dock_area_add_remove_dock(bec_dock_area, qtbot):
     d2 = bec_dock_area.add_dock()
 
     # Check if the docks were added
-    assert len(bec_dock_area.docks) == initial_count + 3
-    assert d0.name() in dict(bec_dock_area.docks)
-    assert d1.name() in dict(bec_dock_area.docks)
-    assert d2.name() in dict(bec_dock_area.docks)
-    assert bec_dock_area.docks[d0.name()].config.widget_class == "BECDock"
-    assert bec_dock_area.docks[d1.name()].config.widget_class == "BECDock"
-    assert bec_dock_area.docks[d2.name()].config.widget_class == "BECDock"
+    assert len(bec_dock_area.dock_area.docks) == initial_count + 3
+    assert d0.name() in dict(bec_dock_area.dock_area.docks)
+    assert d1.name() in dict(bec_dock_area.dock_area.docks)
+    assert d2.name() in dict(bec_dock_area.dock_area.docks)
+    assert bec_dock_area.dock_area.docks[d0.name()].config.widget_class == "BECDock"
+    assert bec_dock_area.dock_area.docks[d1.name()].config.widget_class == "BECDock"
+    assert bec_dock_area.dock_area.docks[d2.name()].config.widget_class == "BECDock"
 
     # Check panels API for getting docks to CLI
-    assert bec_dock_area.panels == dict(bec_dock_area.docks)
+    assert bec_dock_area.panels == dict(bec_dock_area.dock_area.docks)
 
     # Remove docks
     d0_name = d0.name()
-    bec_dock_area.remove_dock(d0_name)  # TODO fix this, works in jupyter console
+    bec_dock_area.remove_dock(d0_name)
     qtbot.wait(200)
     d1.remove()
     qtbot.wait(200)
 
-    assert len(bec_dock_area.docks) == initial_count + 1
-    assert d0.name() not in dict(bec_dock_area.docks)
-    assert d1.name() not in dict(bec_dock_area.docks)
-    assert d2.name() in dict(bec_dock_area.docks)
+    assert len(bec_dock_area.dock_area.docks) == initial_count + 1
+    assert d0.name() not in dict(bec_dock_area.dock_area.docks)
+    assert d1.name() not in dict(bec_dock_area.dock_area.docks)
+    assert d2.name() in dict(bec_dock_area.dock_area.docks)
 
 
 def test_add_remove_bec_figure_to_dock(bec_dock_area):
@@ -63,7 +64,7 @@ def test_add_remove_bec_figure_to_dock(bec_dock_area):
     im = fig.image("eiger")
     mm = fig.motor_map("samx", "samy")
 
-    assert len(bec_dock_area.docks) == 1
+    assert len(bec_dock_area.dock_area.docks) == 1
     assert len(d0.widgets) == 1
     assert len(d0.widget_list) == 1
     assert len(fig.widgets) == 3
@@ -74,14 +75,6 @@ def test_add_remove_bec_figure_to_dock(bec_dock_area):
     assert mm.config.widget_class == "BECMotorMap"
 
 
-def test_dock_area_errors(bec_dock_area):
-    d0 = bec_dock_area.add_dock(name="dock_0")
-
-    with pytest.raises(ValueError) as excinfo:
-        bec_dock_area.add_dock(name="dock_0")
-        assert "Dock with name  dock_0 already exists." in str(excinfo.value)
-
-
 def test_close_docks(bec_dock_area, qtbot):
     d0 = bec_dock_area.add_dock(name="dock_0")
     d1 = bec_dock_area.add_dock(name="dock_1")
@@ -89,7 +82,7 @@ def test_close_docks(bec_dock_area, qtbot):
 
     bec_dock_area.clear_all()
     qtbot.wait(200)
-    assert len(bec_dock_area.docks) == 0
+    assert len(bec_dock_area.dock_area.docks) == 0
 
 
 def test_undock_and_dock_docks(bec_dock_area, qtbot):
@@ -102,13 +95,60 @@ def test_undock_and_dock_docks(bec_dock_area, qtbot):
     bec_dock_area.detach_dock("dock_1")
     d2.detach()
 
-    assert len(bec_dock_area.docks) == 4
-    assert len(bec_dock_area.tempAreas) == 3
+    assert len(bec_dock_area.dock_area.docks) == 4
+    assert len(bec_dock_area.dock_area.tempAreas) == 3
 
     d0.attach()
-    assert len(bec_dock_area.docks) == 4
-    assert len(bec_dock_area.tempAreas) == 2
+    assert len(bec_dock_area.dock_area.docks) == 4
+    assert len(bec_dock_area.dock_area.tempAreas) == 2
 
     bec_dock_area.attach_all()
-    assert len(bec_dock_area.docks) == 4
-    assert len(bec_dock_area.tempAreas) == 0
+    assert len(bec_dock_area.dock_area.docks) == 4
+    assert len(bec_dock_area.dock_area.tempAreas) == 0
+
+
+###################################
+# Toolbar Actions
+###################################
+def test_toolbar_add_plot_waveform(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_plots"].widgets["waveform"].trigger()
+    assert "waveform_1" in bec_dock_area.panels
+    assert bec_dock_area.panels["waveform_1"].widgets[0].config.widget_class == "BECWaveformWidget"
+
+
+def test_toolbar_add_plot_image(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_plots"].widgets["image"].trigger()
+    assert "image_1" in bec_dock_area.panels
+    assert bec_dock_area.panels["image_1"].widgets[0].config.widget_class == "BECImageWidget"
+
+
+def test_toolbar_add_plot_motor_map(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_plots"].widgets["motor_map"].trigger()
+    assert "motor_map_1" in bec_dock_area.panels
+    assert bec_dock_area.panels["motor_map_1"].widgets[0].config.widget_class == "BECMotorMapWidget"
+
+
+def test_toolbar_add_device_device_box(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_devices"].widgets["device_box"].trigger()
+    assert "device_box_1" in bec_dock_area.panels
+    assert bec_dock_area.panels["device_box_1"].widgets[0].config.widget_class == "DeviceBox"
+
+
+def test_toolbar_add_utils_queue(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_utils"].widgets["queue"].trigger()
+    assert "queue_1" in bec_dock_area.panels
+    assert bec_dock_area.panels["queue_1"].widgets[0].config.widget_class == "BECQueue"
+
+
+def test_toolbar_add_utils_status(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_utils"].widgets["status"].trigger()
+    assert "status_1" in bec_dock_area.panels
+    assert bec_dock_area.panels["status_1"].widgets[0].config.widget_class == "BECStatusBox"
+
+
+def test_toolbar_add_utils_progress_bar(bec_dock_area):
+    bec_dock_area.toolbar.widgets["menu_utils"].widgets["progress_bar"].trigger()
+    assert "progress_bar_1" in bec_dock_area.panels
+    assert (
+        bec_dock_area.panels["progress_bar_1"].widgets[0].config.widget_class == "RingProgressBar"
+    )