diff --git a/tests/tests_devices/test_eiger.py b/tests/tests_devices/test_eiger.py index 73ca78e..3e94b42 100644 --- a/tests/tests_devices/test_eiger.py +++ b/tests/tests_devices/test_eiger.py @@ -398,8 +398,13 @@ def test_eiger_on_stage(eiger_1_5m): with ( mock.patch.object(eiger.jfj_client, "wait_for_idle", return_value=True), mock.patch.object(eiger.jfj_client, "start") as mock_start, + mock.patch.object(eiger.jfj_client, "stop") as mock_stop, + mock.patch.object( + eiger.jfj_client.api, "status_get", return_value=BrokerStatus(state="Idle") + ), ): eiger.stage() + mock_stop.assert_not_called() assert ( eiger._full_path == f"{scan_msg.info['file_components'][0]}_{eiger.name}_master.{scan_msg.info['file_components'][1]}"