DiamondLightSource/hyperion#1192 fix some system tests

olliesilvester · Feb 29, 2024 · 32228c4 · 32228c4
1 parent 62da152
commit 32228c4
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/tests/system_tests/external_interaction/test_zocalo_system.py b/tests/system_tests/external_interaction/test_zocalo_system.py
@@ -55,11 +55,16 @@ async def test_when_running_start_stop_then_get_expected_returned_results(
 ):
     dcids = (1, 2)
     zc = ZocaloCallback()
-    zc.start({})  # TODO
-    for dcid in dcids:
-        zc.zocalo_interactor.run_start(dcid)
-    for dcid in dcids:
-        zc.zocalo_interactor.run_end(dcid)
+    zc.triggering_plan = "test"
+    zc.start(
+        {
+            "subplan_name": "test",
+            "uid": "123",
+            "zocalo_environment": "dev_artemis",
+            "ispyb_ids": dcids,
+        }  # type:ignore
+    )
+    zc.stop({"run_start": "123"})  # type:ignore
     RE(bps.trigger(zocalo_device, wait=True))
     result = await zocalo_device.read()
     assert result["zocalo-results"]["value"][0] == TEST_RESULT_LARGE[0]