mirror of
https://github.com/ivan-usov-org/bec.git
synced 2025-04-22 02:20:02 +02:00
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
import fakeredis
|
|
import pytest
|
|
|
|
from bec_lib import bec_logger, messages
|
|
from bec_lib.client import BECClient
|
|
from bec_lib.redis_connector import RedisConnector
|
|
|
|
# overwrite threads_check fixture from bec_lib,
|
|
# to have it in autouse
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def threads_check(threads_check):
|
|
yield
|
|
bec_logger.logger.remove()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def bec_client_singleton_reset():
|
|
"""Reset the BECClient singleton before and after each test."""
|
|
# pylint: disable=protected-access
|
|
BECClient._reset_singleton()
|
|
yield
|
|
BECClient._reset_singleton()
|
|
|
|
|
|
def fake_redis_server(host, port):
|
|
redis = fakeredis.FakeRedis()
|
|
return redis
|
|
|
|
|
|
@pytest.fixture
|
|
def connected_connector():
|
|
connector = RedisConnector("localhost:1", redis_cls=fake_redis_server)
|
|
connector._redis_conn.flushall()
|
|
try:
|
|
yield connector
|
|
finally:
|
|
connector.shutdown()
|
|
|
|
|
|
@pytest.fixture
|
|
def scan_queue_status_msg():
|
|
yield messages.ScanQueueStatusMessage(
|
|
queue={
|
|
"primary": {
|
|
"info": [
|
|
{
|
|
"queue_id": "7c15c9a2-71d4-4f2a-91a7-c4a63088fa38",
|
|
"scan_id": ["bfa582aa-f9cd-4258-ab5d-3e5d54d3dde5"],
|
|
"is_scan": [True],
|
|
"request_blocks": [
|
|
{
|
|
"msg": b"\x84\xa8msg_type\xa4scan\xa7content\x83\xa9scan_type\xabfermat_scan\xa9parameter\x82\xa4args\x82\xa4samx\x92\xfe\x02\xa4samy\x92\xfe\x02\xa6kwargs\x83\xa4step\xcb?\xf8\x00\x00\x00\x00\x00\x00\xa8exp_time\xcb?\x94z\xe1G\xae\x14{\xa8relative\xc3\xa5queue\xa7primary\xa8metadata\x81\xa3RID\xd9$cd8fc68f-fe65-4031-9a37-e0e7ba9df542\xa7version\xcb?\xf0\x00\x00\x00\x00\x00\x00",
|
|
"RID": "cd8fc68f-fe65-4031-9a37-e0e7ba9df542",
|
|
"scan_motors": ["samx", "samy"],
|
|
"is_scan": True,
|
|
"scan_number": 25,
|
|
"scan_id": "bfa582aa-f9cd-4258-ab5d-3e5d54d3dde5",
|
|
"metadata": {"RID": "cd8fc68f-fe65-4031-9a37-e0e7ba9df542"},
|
|
"content": {
|
|
"scan_type": "fermat_scan",
|
|
"parameter": {
|
|
"args": {"samx": [-2, 2], "samy": [-2, 2]},
|
|
"kwargs": {"step": 1.5, "exp_time": 0.02, "relative": True},
|
|
},
|
|
"queue": "primary",
|
|
},
|
|
}
|
|
],
|
|
"scan_number": [25],
|
|
"status": "PENDING",
|
|
"active_request_block": None,
|
|
}
|
|
],
|
|
"status": "RUNNING",
|
|
}
|
|
}
|
|
)
|