From 51535239d020b4c4d9c7e232386bcdea06c908cc Mon Sep 17 00:00:00 2001
From: GotthardG <51994228+GotthardG@users.noreply.github.com>
Date: Tue, 30 Jun 2026 15:16:22 +0200
Subject: [PATCH] adding beam center to raster shots, correcting test logic

---
 tests/unit/daq/test_rotation_screening_diffraction.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/unit/daq/test_rotation_screening_diffraction.py b/tests/unit/daq/test_rotation_screening_diffraction.py
index d777958d..3b4f1be2 100644
--- a/tests/unit/daq/test_rotation_screening_diffraction.py
+++ b/tests/unit/daq/test_rotation_screening_diffraction.py
@@ -22,11 +22,13 @@ def test_screening_diffraction_uploads_one_image_per_wedge():
     scan_result = SimpleNamespace(images=[_img(0, 0.0, 245, 2.1), _img(1, 90.0, 12, 8.5)])
     svc._ingest_screening_diffraction(SimpleNamespace(db_id=123), scan_result)
 
-    # one fetch + one upload per wedge, with spots + resolution-estimate ring on
+    # one fetch + one upload per wedge, with spots + resolution-estimate ring
+    # + beam center overlays on
     assert jfjoch.get_diffraction_image.call_count == 2
     for call in jfjoch.get_diffraction_image.call_args_list:
         assert call.kwargs["show_spots"] is True
         assert call.kwargs["show_res_est"] is True
+        assert call.kwargs["show_beam_center"] is True
 
     assert aare.upload_jpg.call_count == 2
     first = aare.upload_jpg.call_args_list[0]
@@ -46,7 +48,9 @@ def test_screening_diffraction_skips_images_without_id():
         SimpleNamespace(images=[SimpleNamespace(number=None), _img(3)]),
     )
 
-    jfjoch.get_diffraction_image.assert_called_once_with(3, show_spots=True, show_res_est=True)
+    jfjoch.get_diffraction_image.assert_called_once_with(
+        3, show_spots=True, show_res_est=True, show_beam_center=True
+    )
     assert aare.upload_jpg.call_count == 1