import os

import bec_utils
import yaml
from bec_utils import DeviceManagerBase as DeviceManager
from bec_utils.tests.utils import ConnectorMock, create_session_from_config

from scan_server.scan_server import ScanServer
from scan_server.scan_worker import InstructionQueueStatus

# pylint: disable=missing-function-docstring
# pylint: disable=protected-access

# dir_path = os.path.dirname(os.path.realpath(__file__))
dir_path = os.path.dirname(bec_utils.__file__)


def load_ScanServerMock():
    connector = ConnectorMock("")
    device_manager = DeviceManager(connector, "")
    device_manager.producer = connector.producer()
    with open(f"{dir_path}/tests/test_config.yaml", "r") as f:
        device_manager._session = create_session_from_config(yaml.safe_load(f))
    device_manager._load_session()
    return ScanServerMock(device_manager, connector)


class WorkerMock:
    def __init__(self) -> None:
        self.scan_id = None
        self.scan_motors = []
        self.current_scanID = None
        self.current_scan_info = None
        self.status = InstructionQueueStatus.IDLE


class ScanServerMock(ScanServer):
    def __init__(self, device_manager, connector) -> None:
        self.device_manager = device_manager
        super().__init__(bootstrap_server="dummy", connector_cls=ConnectorMock, scibec_url="dummy")
        self.scan_worker = WorkerMock()

    def _start_device_manager(self):
        pass

    def _start_scan_server(self):
        pass

    def shutdown(self):
        pass