import importlib.util
from unittest.mock import MagicMock, patch

import pytest
from aarecommon.models.models import SampleShortInfo

from aare.beamline_dispatch.beamline_dispatch import get_beamline_dispatch
from aare.daq.daq import AareDAQ


@pytest.fixture()
def daq(mock_daq):
    mock_daq.sync_current_sample_from_tell = MagicMock(
        return_value=SampleShortInfo(
            db_id=12345,
            puck_name="T1",
            pin=1,
            dewar_name="TEST",
            sample_name="test_sample",
            run_number=42,
        )
    )
    mock_daq.spreadsheet_params = MagicMock(return_value=(None, "/data/abcd/"))
    mock_daq._cfg = MagicMock(auto_params=None)
    return mock_daq


@pytest.mark.skipif(
    importlib.util.find_spec("pxii_bec") is None, reason="run only for pxii flavour"
)
def test_pxii_default_params(daq: AareDAQ):
    with patch.dict("os.environ", {"BEAMLINE": "X10SA"}):
        daq._dispatch = get_beamline_dispatch()
    params = daq.get_collection_params(prefer_smart=False)[0]
    assert params.transmission == 0.1


@pytest.mark.skipif(
    importlib.util.find_spec("pxiii_bec") is None, reason="run only for pxiii flavour"
)
def test_pxiii_default_params(daq: AareDAQ):
    with patch.dict("os.environ", {"BEAMLINE": "X06DA"}):
        daq._dispatch = get_beamline_dispatch()
    params = daq.get_collection_params(prefer_smart=False)[0]
    assert params.transmission == 1