mirror of
https://github.com/bec-project/ophyd_devices.git
synced 2026-02-20 09:18:42 +01:00
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import pytest
|
|
|
|
from ophyd_devices.devices.optics_shutter import Shutter, ShutterEnabled, ShutterOpenState
|
|
from ophyd_devices.tests.utils import patched_device
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def mock_shutter():
|
|
with patched_device(Shutter, name="shutter", prefix="X10SA-EH1-PSYS:SH-A-") as shutter:
|
|
yield shutter
|
|
|
|
|
|
def test_shutter_open(mock_shutter):
|
|
mock_shutter.is_enabled._read_pv.mock_data = ShutterEnabled.ENABLED.value
|
|
mock_shutter.is_open._read_pv.mock_data = ShutterOpenState.CLOSED.value
|
|
st = mock_shutter.open()
|
|
assert not st.done
|
|
mock_shutter.is_open._read_pv.mock_data = ShutterOpenState.OPEN.value
|
|
assert st.done
|
|
|
|
|
|
def test_shutter_close(mock_shutter):
|
|
mock_shutter.is_enabled._read_pv.mock_data = ShutterEnabled.ENABLED.value
|
|
mock_shutter.is_open._read_pv.mock_data = ShutterOpenState.OPEN.value
|
|
st = mock_shutter.close()
|
|
assert not st.done
|
|
mock_shutter.is_open._read_pv.mock_data = ShutterOpenState.CLOSED.value
|
|
assert st.done
|
|
|
|
|
|
def test_shutter_not_enabled(mock_shutter):
|
|
with pytest.raises(RuntimeError) as e:
|
|
mock_shutter.open()
|
|
assert e.match("The shutter is disabled!")
|
|
|
|
|
|
def test_shutter_status(mock_shutter):
|
|
mock_shutter.is_open._read_pv.mock_data = ShutterOpenState.OPEN.value
|
|
assert mock_shutter.status() == ShutterOpenState.OPEN.value
|
|
mock_shutter.is_open._read_pv.mock_data = ShutterOpenState.CLOSED.value
|
|
assert mock_shutter.status() == ShutterOpenState.CLOSED.value
|