Tests for processed tomo insert

DiamondLightSource · Aug 14, 2024 · ceac3ff · ceac3ff
1 parent f2ff3f1
commit ceac3ff
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/tests/services/test_denoise_slurm.py b/tests/services/test_denoise_slurm.py
@@ -240,11 +240,14 @@ def test_denoise_slurm_service(
     offline_transport.send.assert_any_call(
         destination="ispyb_connector",
         message={
-            "ispyb_command": "buffer",
-            "buffer_command": {"ispyb_command": "insert_processed_tomogram"},
-            "buffer_lookup": {"tomogram_id": 0},
-            "filePath": f"{tmp_path}/Denoise/job007/denoised/test_stack_aretomo.denoised.mrc",
-            "processingType": "Denoised",
+            "parameters": {
+                "ispyb_command": "buffer",
+                "buffer_command": {"ispyb_command": "insert_processed_tomogram"},
+                "buffer_lookup": {"tomogram_id": 0},
+                "filePath": f"{tmp_path}/Denoise/job007/denoised/test_stack_aretomo.denoised.mrc",
+                "processingType": "Denoised",
+            },
+            "content": {"dummy": "dummy"},
         },
     )
     offline_transport.send.assert_any_call(

diff --git a/tests/services/test_membrain_seg.py b/tests/services/test_membrain_seg.py
@@ -149,11 +149,14 @@ def test_membrain_seg_service(
     offline_transport.send.assert_any_call(
         destination="ispyb_connector",
         message={
-            "ispyb_command": "buffer",
-            "buffer_command": {"ispyb_command": "insert_processed_tomogram"},
-            "buffer_lookup": {"tomogram_id": 0},
-            "filePath": f"{tmp_path}/Segmentation/job008/test_stack_aretomo.denoised_segmented.mrc",
-            "processingType": "Segmented",
+            "parameters": {
+                "ispyb_command": "buffer",
+                "buffer_command": {"ispyb_command": "insert_processed_tomogram"},
+                "buffer_lookup": {"tomogram_id": 0},
+                "filePath": f"{tmp_path}/Segmentation/job008/test_stack_aretomo.denoised_segmented.mrc",
+                "processingType": "Segmented",
+            },
+            "content": {"dummy": "dummy"},
         },
     )