54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
import sys
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from PySide6.QtWidgets import QApplication
|
|
|
|
from aare.common.models import SampleShortInfo, DewarAddress
|
|
|
|
# external service mocks
|
|
from unittest.mock import MagicMock
|
|
import sys
|
|
|
|
def mock_module(name):
|
|
m = MagicMock()
|
|
m.__path__ = []
|
|
sys.modules[name] = m
|
|
return m
|
|
|
|
mock_module("aarelcinfer_client")
|
|
mock_module("aarelcinfer_client.api")
|
|
mock_module("aarelcinfer_client.models")
|
|
|
|
mock_module("aarescan_client")
|
|
mock_module("aarescan_client.models")
|
|
mock_module("aarescan_client.models.grid_request")
|
|
mock_module("aarescan_client.models.screen_request")
|
|
|
|
sys.modules.setdefault("pxii_bec", MagicMock())
|
|
sys.modules.setdefault("pxii_bec.macros", MagicMock())
|
|
sys.modules.setdefault("pxii_bec.macros.planner", MagicMock())
|
|
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_info():
|
|
return SampleShortInfo(
|
|
db_id=1,
|
|
puck_name="puck1",
|
|
dewar_name="dew1",
|
|
sample_name="sample1",
|
|
run_number=1,
|
|
user="group1",
|
|
pin=3,
|
|
location=DewarAddress(segment="A", pos=2),
|
|
)
|
|
|
|
@pytest.fixture(scope="session")
|
|
def qapp():
|
|
app = QApplication.instance()
|
|
if app is None:
|
|
app = QApplication(sys.argv)
|
|
return app
|