from pml.exceptions import PvException
import pml.device
import pytest
import mock


@pytest.fixture
def create_device(readback, setpoint):
    _rb = readback
    _sp = setpoint
    device = pml.device.Device(rb_pv=_rb, sp_pv=_sp, cs=mock.MagicMock())
    return device


def test_set_device_value():
    rb_pv = 'SR01A-PC-SQUAD-01:I'
    sp_pv = 'SR01A-PC-SQUAD-01:SETI'

    device1 = create_device(rb_pv, sp_pv)
    device1.put_value(40)
    device1._cs.put.assert_called_with(sp_pv, 40)

    device2 = create_device(rb_pv, None)
    with pytest.raises(PvException):
        device2.put_value(40)

def test_get_device_value():
    sp_pv = 'SR01A-PC-SQUAD-01:SETI'

    device = create_device(None, sp_pv)
    with pytest.raises(PvException):
        device.get_value('non_existant')

    with pytest.raises(PvException):
        create_device(None, None)

from pml.exceptions import PvException
import pml.device
import pytest
import mock


@pytest.fixture
def create_device(readback, setpoint):
    _rb = readback
    _sp = setpoint
    device = pml.device.Device(rb_pv=_rb, sp_pv=_sp, cs=mock.MagicMock())
    return device


def test_set_device_value():
    rb_pv = 'SR01A-PC-SQUAD-01:I'
    sp_pv = 'SR01A-PC-SQUAD-01:SETI'

    device1 = create_device(rb_pv, sp_pv)
    device1.put_value(40)
    device1._cs.put.assert_called_with(sp_pv, 40)

    device2 = create_device(rb_pv, None)
    with pytest.raises(PvException):
        device2.put_value(40)


def test_get_device_value():
    sp_pv = 'SR01A-PC-SQUAD-01:SETI'

    device = create_device(None, sp_pv)
    with pytest.raises(PvException):
        device.get_value('non_existant')

    with pytest.raises(PvException):
        create_device(None, None)

Add test to get device with non-existant key