"""Auto-generate per-server agent_api_key when missing (batch/single install)."""

from unittest.mock import AsyncMock, MagicMock, patch

import pytest

from server.application.server_batch_common import ensure_agent_api_key, generate_agent_api_key_value
from server.domain.models import Server


def test_generate_agent_api_key_format():
    key = generate_agent_api_key_value()
    assert key.startswith("nxs-")
    assert len(key) > 20


@pytest.mark.asyncio
async def test_ensure_agent_api_key_returns_existing():
    server = Server(id=1, name="s1", agent_api_key="nxs-existing")
    assert await ensure_agent_api_key(server) == "nxs-existing"


@pytest.mark.asyncio
async def test_ensure_agent_api_key_generates_and_persists():
    server = Server(id=42, name="机器人", agent_api_key=None)
    row = Server(id=42, name="机器人", agent_api_key=None)

    mock_repo = MagicMock()
    mock_repo.get_by_id = AsyncMock(return_value=row)

    mock_session = MagicMock()
    mock_session.commit = AsyncMock()

    mock_ctx = MagicMock()
    mock_ctx.__aenter__ = AsyncMock(return_value=mock_session)
    mock_ctx.__aexit__ = AsyncMock(return_value=None)

    with patch(
        "server.infrastructure.database.session.AsyncSessionLocal",
        return_value=mock_ctx,
    ), patch(
        "server.infrastructure.database.server_repo.ServerRepositoryImpl",
        return_value=mock_repo,
    ):
        key = await ensure_agent_api_key(server)

    assert key.startswith("nxs-")
    assert row.agent_api_key == key
    assert server.agent_api_key == key
    mock_session.commit.assert_awaited_once()