release: nexus btpanel session fix and app-v2
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
"""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()
|
||||
Reference in New Issue
Block a user