import os
from unittest import mock

import pytest

import bec_lib
from bec_lib import messages
from bec_lib.bec_service import BECService
from bec_lib.endpoints import MessageEndpoints
from bec_lib.messages import BECStatus
from bec_lib.service_config import ServiceConfig

# pylint: disable=no-member
# pylint: disable=missing-function-docstring
# pylint: disable=redefined-outer-name
# pylint: disable=protected-access

dir_path = os.path.dirname(bec_lib.__file__)


def test_bec_service_init_with_service_config():
    config = ServiceConfig(redis={"host": "localhost", "port": 6379})
    service = BECService(config=config, connector_cls=mock.MagicMock())
    assert service._service_config == config
    assert service.bootstrap_server == "localhost:6379"
    assert service._unique_service is False


def test_bec_service_init_raises_for_invalid_config():
    with pytest.raises(TypeError):
        BECService(config=mock.MagicMock(), connector_cls=mock.MagicMock())


def test_bec_service_init_with_service_config_path():
    service = BECService(
        config=f"{dir_path}/tests/test_service_config.yaml", connector_cls=mock.MagicMock()
    )
    assert isinstance(service._service_config, ServiceConfig)
    assert service.bootstrap_server == "localhost:6379"
    assert service._unique_service is False


def test_init_runs_service_check():
    with mock.patch.object(
        BECService, "_update_existing_services", return_value=False
    ) as mock_update_existing_services:
        service = BECService(
            config=f"{dir_path}/tests/test_service_config.yaml",
            connector_cls=mock.MagicMock(),
            unique_service=True,
        )
        mock_update_existing_services.assert_called_once()


def test_run_service_check_raises_for_existing_service():
    with mock.patch.object(
        BECService, "_update_existing_services", return_value=False
    ) as mock_update_existing_services:
        service = BECService(
            config=f"{dir_path}/tests/test_service_config.yaml",
            connector_cls=mock.MagicMock(),
            unique_service=True,
        )
        service._services_info = {"BECService": mock.MagicMock()}
        with pytest.raises(RuntimeError):
            service._run_service_check(timeout_time=0, elapsed_time=10)


def test_run_service_check_repeats():
    with mock.patch.object(
        BECService, "_update_existing_services", return_value=False
    ) as mock_update_existing_services:
        service = BECService(
            config=f"{dir_path}/tests/test_service_config.yaml",
            connector_cls=mock.MagicMock(),
            unique_service=True,
        )
        service._services_info = {"BECService": mock.MagicMock()}
        assert service._run_service_check(timeout_time=0.5, elapsed_time=0) is True


def test_bec_service_shutdown():
    with mock.patch.object(
        BECService, "_update_existing_services", return_value=False
    ) as mock_update_existing_services:
        service = BECService(
            config=f"{dir_path}/tests/test_service_config.yaml",
            connector_cls=mock.MagicMock(),
            unique_service=True,
        )
        service._service_info_event = mock.MagicMock()
        service._metrics_emitter_event = mock.MagicMock()
        service._service_info_thread = mock.MagicMock()
        service._metrics_emitter_thread = mock.MagicMock()

        service.shutdown()
        service._service_info_event.set.assert_called_once()
        service._metrics_emitter_event.set.assert_called_once()
        service._service_info_thread.join.assert_called_once()
        service._metrics_emitter_thread.join.assert_called_once()


def test_bec_service_service_status():
    with mock.patch.object(
        BECService, "_update_existing_services", return_value=False
    ) as mock_update_existing_services:
        service = BECService(
            config=f"{dir_path}/tests/test_service_config.yaml",
            connector_cls=mock.MagicMock(),
            unique_service=True,
        )
        mock_update_existing_services.reset_mock()
        status = service.service_status
        mock_update_existing_services.assert_called_once()


def test_bec_service_update_existing_services():
    service_keys = [
        MessageEndpoints.service_status("service1").encode(),
        MessageEndpoints.service_status("service2").encode(),
    ]
    service_msgs = [
        messages.StatusMessage(name="service1", status=BECStatus.RUNNING, info={}, metadata={}),
        messages.StatusMessage(name="service2", status=BECStatus.IDLE, info={}, metadata={}),
    ]
    connector_cls = mock.MagicMock()
    connector_cls().producer().keys.return_value = service_keys
    connector_cls().producer().get.side_effect = [msg for msg in service_msgs]
    service = BECService(
        config=f"{os.path.dirname(bec_lib.__file__)}/tests/test_service_config.yaml",
        connector_cls=connector_cls,
        unique_service=True,
    )
    assert service._services_info == {"service1": service_msgs[0], "service2": service_msgs[1]}


def test_bec_service_update_existing_services_ignores_wrong_msgs():
    service_keys = [
        MessageEndpoints.service_status("service1").encode(),
        MessageEndpoints.service_status("service2").encode(),
    ]
    service_msgs = [
        messages.StatusMessage(name="service1", status=BECStatus.RUNNING, info={}, metadata={}),
        None,
    ]
    connector_cls = mock.MagicMock()
    connector_cls().producer().keys.return_value = service_keys
    connector_cls().producer().get.side_effect = [service_msgs[0], None]
    service = BECService(
        config=f"{os.path.dirname(bec_lib.__file__)}/tests/test_service_config.yaml",
        connector_cls=connector_cls,
        unique_service=True,
    )
    assert service._services_info == {"service1": service_msgs[0]}