bec-server 2.9.3__tar.gz → 2.9.4__tar.gz

{bec_server-2.9.3 → bec_server-2.9.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bec-server
-Version: 2.9.3
+Version: 2.9.4
 Summary: BEC server
 Project-URL: Bug Tracker, https://gitlab.psi.ch/bec/bec/issues
 Project-URL: Homepage, https://gitlab.psi.ch/bec/bec

{bec_server-2.9.3 → bec_server-2.9.4}/bec_server/device_server/devices/config_update_handler.py

@@ -88,8 +88,8 @@ class ConfigUpdateHandler:
 
                 if "limits" in dev_config["deviceConfig"]:
                     limits = {
-                        "low": device.obj.low_limit_travel.get(),
-                        "high": device.obj.high_limit_travel.get(),
+                        "low": {"value": device.obj.low_limit_travel.get()},
+                        "high": {"value": device.obj.high_limit_travel.get()},
                     }
                     self.device_manager.connector.set_and_publish(
                         MessageEndpoints.device_limits(device.name),

{bec_server-2.9.3 → bec_server-2.9.4}/bec_server/device_server/devices/devicemanager.py

@@ -62,8 +62,8 @@ class DSDevice(DeviceBase):
         dev_config_msg = messages.DeviceMessage(signals=self.obj.read_configuration(), metadata={})
         if hasattr(self.obj, "low_limit_travel") and hasattr(self.obj, "high_limit_travel"):
             limits = {
-                "low": self.obj.low_limit_travel.get(),
-                "high": self.obj.high_limit_travel.get(),
+                "low": {"value": self.obj.low_limit_travel.get()},
+                "high": {"value": self.obj.high_limit_travel.get()},
             }
         else:
             limits = None
@@ -531,9 +531,7 @@ class DeviceManagerDS(DeviceManagerBase):
             for key, val in data.items():
                 signals[key] = {"value": val[ii], "timestamp": timestamp}
             bundle.append(
-                messages.DeviceMessage(
-                    signals={obj.name: signals}, metadata={"point_id": ii, **metadata}
-                )
+                messages.DeviceMessage(signals=signals, metadata={"point_id": ii, **metadata})
             )
         ds_obj.emitted_points[metadata["scan_id"]] = max_points
         pipe = self.connector.pipeline()

{bec_server-2.9.3 → bec_server-2.9.4}/bec_server/scan_server/scan_stubs.py

@@ -398,7 +398,7 @@ class ScanStubs:
         yield self._device_msg(
             device=device,
             action="publish_data_as_read",
-            parameter={"data": {device: data}},
+            parameter={"data": data},
             metadata=metadata,
         )
 

{bec_server-2.9.3 → bec_server-2.9.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "bec-server"
-version = "2.9.3"
+version = "2.9.4"
 description = "BEC server"
 requires-python = ">=3.10"
 classifiers = [
@@ -60,3 +60,10 @@ include = ["*"]
 [tool.black]
 line-length = 100
 skip-magic-trailing-comma = true
+
+[tool.isort]
+profile = "black"
+line_length = 100
+multi_line_output = 3
+include_trailing_comma = true
+known_first_party = ["bec_lib", "bec_server", "bec_ipython_client"]

{bec_server-2.9.3 → bec_server-2.9.4}/tests/tests_file_writer/test_file_writer_manager.py

@@ -1,16 +1,13 @@
 # pylint: skip-file
-import contextlib
 import os
 from unittest import mock
 
 import numpy as np
 import pytest
-import yaml
 
 import bec_lib
-from bec_lib import DeviceManagerBase, MessageEndpoints, ServiceConfig, messages
+from bec_lib import MessageEndpoints, ServiceConfig, messages
 from bec_lib.logger import bec_logger
-from bec_lib.messages import BECStatus
 from bec_lib.redis_connector import MessageObject
 from bec_lib.tests.utils import ConnectorMock
 from bec_server.file_writer import FileWriterManager
@@ -33,9 +30,10 @@ def file_writer_manager_mock():
             "log_writer": {"base_path": "./"},
         },
     )
-    with mock.patch.object(
-        FileWriterManager, "_start_device_manager", return_value=None
-    ), mock.patch.object(FileWriterManager, "wait_for_service"):
+    with (
+        mock.patch.object(FileWriterManager, "_start_device_manager", return_value=None),
+        mock.patch.object(FileWriterManager, "wait_for_service"),
+    ):
         file_writer_manager_mock = FileWriterManager(config=config, connector_cls=connector_cls)
         try:
             yield file_writer_manager_mock
@@ -195,10 +193,10 @@ def test_update_async_data(file_writer_manager_mock):
 def test_process_async_data_single_entry(file_writer_manager_mock):
     file_manager = file_writer_manager_mock
     file_manager.scan_storage["scan_id"] = ScanStorage(10, "scan_id")
-    data = [{"data": messages.DeviceMessage(signals={"data": np.zeros((10, 10))})}]
+    data = [{"data": messages.DeviceMessage(signals={"data": {"value": np.zeros((10, 10))}})}]
     file_manager._process_async_data(data, "scan_id", "dev1")
     assert np.isclose(
-        file_manager.scan_storage["scan_id"].async_data["dev1"]["data"], np.zeros((10, 10))
+        file_manager.scan_storage["scan_id"].async_data["dev1"]["data"]["value"], np.zeros((10, 10))
     ).all()
 
 

{bec_server-2.9.3 → bec_server-2.9.4}/tests/tests_scan_bundler/test_scan_bundler.py

@@ -9,10 +9,24 @@ from bec_lib.connector import MessageObject
 # pylint: disable=protected-access
 
 
-def test_device_read_callback(scan_bundler_mock):
+@pytest.fixture()
+def dummy_signal_data():
+    return {
+        "samx": {"value": 0.51, "timestamp": 1234.56},
+        "samx_setpoint": {"value": 0.51, "timestamp": 1234.56},
+    }
+
+
+@pytest.fixture()
+def dummy_device_data_message(dummy_signal_data):
+    return messages.DeviceMessage(
+        signals=dummy_signal_data, metadata={"scan_id": "scan_id", "readout_priority": "monitored"}
+    )
+
+
+def test_device_read_callback(scan_bundler_mock, dummy_signal_data):
     dev_msg = messages.DeviceMessage(
-        signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-        metadata={"scan_id": "laksjd", "readout_priority": "monitored"},
+        signals=dummy_signal_data, metadata={"scan_id": "laksjd", "readout_priority": "monitored"}
     )
     msg = MessageObject(MessageEndpoints.device_read("samx").endpoint, dev_msg)
 
@@ -68,10 +82,9 @@ def test_wait_for_scan_id(scan_bundler_mock, scan_id, storageID, scan_msg):
         sb._wait_for_scan_id(scan_id)
 
 
-def test_add_device_to_storage_returns_without_scan_id(scan_bundler_mock):
+def test_add_device_to_storage_returns_without_scan_id(scan_bundler_mock, dummy_signal_data):
     msg = messages.DeviceMessage(
-        signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-        metadata={"readout_priority": "monitored"},
+        signals=dummy_signal_data, metadata={"readout_priority": "monitored"}
     )
     sb = scan_bundler_mock
     sb._add_device_to_storage([msg], "samx", timeout_time=1)
@@ -87,10 +100,9 @@ def test_add_device_to_storage_returns_without_signal(scan_bundler_mock):
     assert "samx" not in sb.device_storage
 
 
-def test_add_device_to_storage_returns_on_timeout(scan_bundler_mock):
+def test_add_device_to_storage_returns_on_timeout(scan_bundler_mock, dummy_signal_data):
     msg = messages.DeviceMessage(
-        signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-        metadata={"scan_id": "scan_id", "readout_priority": "monitored"},
+        signals=dummy_signal_data, metadata={"scan_id": "scan_id", "readout_priority": "monitored"}
     )
     sb = scan_bundler_mock
     sb._add_device_to_storage([msg], "samx", timeout_time=1)
@@ -98,10 +110,11 @@ def test_add_device_to_storage_returns_on_timeout(scan_bundler_mock):
 
 
 @pytest.mark.parametrize("scan_status", ["aborted", "closed"])
-def test_add_device_to_storage_returns_without_scan_info(scan_bundler_mock, scan_status):
+def test_add_device_to_storage_returns_without_scan_info(
+    scan_bundler_mock, scan_status, dummy_signal_data
+):
     msg = messages.DeviceMessage(
-        signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-        metadata={"scan_id": "scan_id", "readout_priority": "monitored"},
+        signals=dummy_signal_data, metadata={"scan_id": "scan_id", "readout_priority": "monitored"}
     )
     sb = scan_bundler_mock
     sb.sync_storage["scan_id"] = {"info": {}}
@@ -113,30 +126,13 @@ def test_add_device_to_storage_returns_without_scan_info(scan_bundler_mock, scan
 @pytest.mark.parametrize(
     "msg,scan_type",
     [
-        (
-            messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-                metadata={"scan_id": "scan_id", "readout_priority": "monitored"},
-            ),
-            "step",
-        ),
-        (
-            messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-                metadata={"scan_id": "scan_id", "readout_priority": "monitored"},
-            ),
-            "fly",
-        ),
-        (
-            messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
-                metadata={"scan_id": "scan_id", "readout_priority": "monitored"},
-            ),
-            "wrong",
-        ),
+        ("dummy_device_data_message", "step"),
+        ("dummy_device_data_message", "fly"),
+        ("dummy_device_data_message", "wrong"),
     ],
 )
-def test_add_device_to_storage_primary(scan_bundler_mock, msg, scan_type):
+def test_add_device_to_storage_primary(scan_bundler_mock, msg, scan_type, request):
+    msg = request.getfixturevalue(msg)
     sb = scan_bundler_mock
     sb.sync_storage["scan_id"] = {"info": {"scan_type": scan_type, "monitor_sync": "bec"}}
     sb.sync_storage["scan_id"]["status"] = "open"
@@ -165,7 +161,11 @@ def test_add_device_to_storage_primary(scan_bundler_mock, msg, scan_type):
     [
         (
             messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
+                signals={
+                    "samx": {"value": 0.51},
+                    "setpoint": {"value": 0.5},
+                    "motor_is_moving": {"value": 0},
+                },
                 metadata={"scan_id": "scan_id"},
             ),
             "fly",
@@ -173,7 +173,9 @@ def test_add_device_to_storage_primary(scan_bundler_mock, msg, scan_type):
         (
             messages.DeviceMessage(
                 signals={
-                    "flyer": {"flyer": 0.51, "flyer_setpoint": 0.5, "flyer_motor_is_moving": 0}
+                    "flyer": {"value": 0.51},
+                    "flyer_setpoint": {"value": 0.5},
+                    "flyer_motor_is_moving": {"value": 0},
                 },
                 metadata={"scan_id": "scan_id"},
             ),
@@ -204,7 +206,11 @@ def test_add_device_to_storage_primary_flyer(scan_bundler_mock, msg, scan_type):
     [
         (
             messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
+                signals={
+                    "samx": {"value": 0.51},
+                    "setpoint": {"value": 0.5},
+                    "motor_is_moving": {"value": 0},
+                },
                 metadata={"scan_id": "scan_id", "readout_priority": "baseline"},
             ),
             "step",
@@ -402,7 +408,6 @@ def test_status_modification(scan_bundler_mock):
             scan_id="6ff7a89a-79e5-43ad-828b-c1e1aeed5803",
             status="closed",
             info={
-                "readout_priority": "monitored",
                 "DIID": 4,
                 "RID": "a53538b4-79f3-4132-91b5-d044e438f460",
                 "scan_id": "3ea07f69-b0ee-44fa-8451-b85824a37397",
@@ -421,7 +426,6 @@ def test_status_modification(scan_bundler_mock):
             scan_id="6ff7a89a-79e5-43ad-828b-c1e1aeed5803",
             status="open",
             info={
-                "readout_priority": "monitored",
                 "DIID": 4,
                 "RID": "a53538b4-79f3-4132-91b5-d044e438f460",
                 "scan_id": "3ea07f69-b0ee-44fa-8451-b85824a37397",
@@ -476,7 +480,7 @@ def test_initialize_scan_container(scan_bundler_mock, scan_msg):
     [
         [
             messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
+                signals={"samx": {"value": 0.51, "timestamp": 1234.56}},
                 metadata={
                     "scan_id": "adlk-jalskdja",
                     "readout_priority": "monitored",
@@ -488,7 +492,7 @@ def test_initialize_scan_container(scan_bundler_mock, scan_msg):
         ],
         [
             messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
+                signals={"samx": {"value": 0.51, "timestamp": 1234.56}},
                 metadata={
                     "scan_id": "adlk-jalskdjb",
                     "readout_priority": "monitored",
@@ -500,7 +504,7 @@ def test_initialize_scan_container(scan_bundler_mock, scan_msg):
         ],
         [
             messages.DeviceMessage(
-                signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
+                signals={"samx": {"value": 0.51, "timestamp": 1234.56}},
                 metadata={"scan_id": "adlk-jalskdjc", "readout_priority": "monitored"},
             ),
             23,
@@ -652,7 +656,11 @@ def test_update_monitor_signals(scan_bundler_mock):
 def test_get_last_device_readback(scan_bundler_mock):
     sb = scan_bundler_mock
     dev_msg = messages.DeviceMessage(
-        signals={"samx": {"samx": 0.51, "setpoint": 0.5, "motor_is_moving": 0}},
+        signals={
+            "samx": {"value": 0.51},
+            "setpoint": {"value": 0.5},
+            "motor_is_moving": {"value": 0},
+        },
         metadata={"scan_id": "laksjd", "readout_priority": "monitored"},
     )
     with mock.patch.object(sb, "connector") as connector_mock:

{bec_server-2.9.3 → bec_server-2.9.4}/tests/tests_scan_server/test_scans.py

@@ -1953,7 +1953,7 @@ def test_monitor_scan_run():
                     messages.DeviceInstructionMessage(
                         device="samx",
                         action="publish_data_as_read",
-                        parameter={"data": {"samx": {"rb1": {"value": 1}}}},
+                        parameter={"data": {"rb1": {"value": 1}}},
                         metadata={"readout_priority": "monitored", "DIID": 11, "point_id": 0},
                     ),
                     messages.DeviceInstructionMessage(