"""Tests for MCP server connection checks."""

import pytest

from dolibarr_mcp import dolibarr_mcp_server
from dolibarr_mcp.config import Config
from dolibarr_mcp.dolibarr_client import DolibarrAPIError


class _DummyClient:
    """Simple dummy client for test_api_connection success path."""

    async def __aenter__(self):
        return self

    async def __aexit__(self, exc_type, exc_val, exc_tb):
        return False

    async def get_status(self):
        return {"success": {"dolibarr_version": "1.0.0"}}


class _ErrorClient:
    """Dummy client that raises for test_api_connection error path."""

    async def __aenter__(self):
        return self

    async def __aexit__(self, exc_type, exc_val, exc_tb):
        return False

    async def get_status(self):
        raise DolibarrAPIError("boom")


@pytest.mark.asyncio
async def test_api_connection_success(monkeypatch):
    """Yields True when the Dolibarr API status call succeeds."""
    monkeypatch.setattr(dolibarr_mcp_server, "DolibarrClient", lambda config: _DummyClient())
    config = Config(
        dolibarr_url="https://example.com/api/index.php",
        dolibarr_api_key="test_key",
    )

    async with dolibarr_mcp_server.test_api_connection(config) as api_ok:
        assert api_ok is True


@pytest.mark.asyncio
async def test_api_connection_missing_configuration():
    """Yields False when the configuration is incomplete."""
    config = Config(
        dolibarr_url="https://your-dolibarr-instance.com/api/index.php",
        dolibarr_api_key="placeholder_api_key",
    )

    async with dolibarr_mcp_server.test_api_connection(config) as api_ok:
        assert api_ok is False


@pytest.mark.asyncio
async def test_api_connection_with_client_error(monkeypatch):
    """Yields False when the Dolibarr client raises errors."""
    monkeypatch.setattr(dolibarr_mcp_server, "DolibarrClient", lambda config: _ErrorClient())
    config = Config(
        dolibarr_url="https://example.com/api/index.php",
        dolibarr_api_key="test_key",
    )

    async with dolibarr_mcp_server.test_api_connection(config) as api_ok:
        assert api_ok is False