import pytest
from ophyd_devices.galil.galil_ophyd import GalilMotor

from utils import SocketMock


@pytest.mark.parametrize(
    "pos,msg,sign",
    [
        (1, b" -12800\n\r", 1),
        (-1, b" -12800\n\r", -1),
    ],
)
def test_axis_get(pos, msg, sign):
    leyey = GalilMotor(
        "H",
        name="leyey",
        host="mpc2680.psi.ch",
        port=8081,
        sign=sign,
        socket_cls=SocketMock,
    )
    leyey.controller.sock.flush_buffer()
    leyey.stage()
    leyey.controller.sock.buffer_recv = msg
    val = leyey.read()
    assert val["leyey"]["value"] == pos
    assert leyey.readback.get() == pos


@pytest.mark.parametrize(
    "target_pos,socket_put_messages,socket_get_messages",
    [
        (
            0,
            [
                b"MG allaxref\r",
                b"MG_XQ0\r",
                b"naxis=7\r",
                b"ntarget=0.000\r",
                b"movereq=1\r",
                b"XQ#NEWPAR\r",
            ],
            [
                b"1.00",
                b"-1",
                b":",
                b":",
                b":",
                b":",
            ],
        ),
    ],
)
def test_axis_put(target_pos, socket_put_messages, socket_get_messages):
    leyey = GalilMotor("H", name="leyey", host="mpc2680.psi.ch", port=8081, socket_cls=SocketMock)
    leyey.controller.sock.flush_buffer()
    leyey.controller.sock.buffer_recv = socket_get_messages
    leyey.user_setpoint.put(target_pos)
    assert leyey.controller.sock.buffer_put == socket_put_messages