fix: fix data format in DAQJobN1081B and fix test

ENRG-tr · Oct 12, 2024 · 995b825 · 995b825
1 parent 656a8cc
commit 995b825
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/daq/caen/n1081b.py b/src/daq/caen/n1081b.py
@@ -81,13 +81,16 @@ def _poll_sections(self):
 
     def _send_store_message(self, data: dict, section):
         keys = ["timestamp", *[f"lemo_{x['lemo']}" for x in data["counters"]]]
-        values = [datetime.now().timestamp(), *[x["value"] for x in data["counters"]]]
+        values = [
+            int(datetime.now().timestamp()),
+            *[x["value"] for x in data["counters"]],
+        ]
         self.message_out.put(
             DAQJobMessageStore(
                 store_config=self.config.store_config,
                 daq_job=self,
                 prefix=section,
                 keys=keys,
-                data=values,
+                data=[values],
             )
         )
diff --git a/src/tests/test_n1081b.py b/src/tests/test_n1081b.py
@@ -90,7 +90,7 @@ def test_send_store_message(self, mock_get_function_results):
         self.assertEqual(message.prefix, "SEC_A")
         self.assertIn("timestamp", message.keys)
         self.assertIn("lemo_1", message.keys)
-        self.assertIn(100, message.data)
+        self.assertIn(100, message.data[0])
 
     def test_invalid_section_in_config(self):
         invalid_config = DAQJobN1081BConfig(