Files
AareDAQ/tests/unit/utils/test_beamline_dispatch.py
T
perl_d 70c81abdd8
CI / lint (pull_request) Failing after 31s
CI / test (3.11) (pull_request) Skipped
CI / test (3.12) (pull_request) Skipped
CI / test (3.13) (pull_request) Skipped
CI / test-with-beamline-plugins (pxi_bec) (pull_request) Skipped
CI / test-with-beamline-plugins (pxii_bec) (pull_request) Skipped
CI / test-with-beamline-plugins (pxiii_bec) (pull_request) Skipped
CI / test-with-coverage (pull_request) Skipped
ci: add better linting
2026-07-31 15:48:53 +02:00

49 lines
1.7 KiB
Python

import importlib.util
from unittest.mock import patch
import pytest
from aare.beamline_dispatch.beamline_dispatch import get_beamline_dispatch
from aare.beamline_dispatch.simulated import SimulatedDispatch
from aare.beamline_dispatch.x10sa import X10saDispatch
def test_dispatch_cannot_be_instantiated_without_env():
with patch.dict("os.environ", {}, clear=True):
with pytest.raises(ValueError) as e:
_ = get_beamline_dispatch()
assert e.match("Please set the BEAMLINE environment variable")
def test_simulated_dispatch_can_be_instantiated():
with patch.dict("os.environ", {"BEAMLINE": "SIMULATED"}):
dispatch = get_beamline_dispatch()
assert isinstance(dispatch, SimulatedDispatch)
@pytest.mark.skipif(importlib.util.find_spec("pxi_bec") is None, reason="run only for pxi flavour")
def test_pxi_dispatch_can_be_instantiated():
with patch.dict("os.environ", {"BEAMLINE": "X06SA"}):
with pytest.raises(TypeError) as e:
_ = get_beamline_dispatch()
assert e.match("Can't instantiate abstract class")
@pytest.mark.skipif(
importlib.util.find_spec("pxii_bec") is None, reason="run only for pxii flavour"
)
def test_pxii_dispatch_can_be_instantiated():
with patch.dict("os.environ", {"BEAMLINE": "X10SA"}):
dispatch = get_beamline_dispatch()
assert isinstance(dispatch, X10saDispatch)
@pytest.mark.skipif(
importlib.util.find_spec("pxiii_bec") is None, reason="run only for pxiii flavour"
)
def test_pxiii_dispatch_can_be_instantiated():
with patch.dict("os.environ", {"BEAMLINE": "X06DA"}):
with pytest.raises(TypeError) as e:
_ = get_beamline_dispatch()
assert e.match("Can't instantiate abstract class")