"""Tests for SSH preset resolution on Server objects."""

from unittest.mock import AsyncMock, MagicMock, patch

import pytest

from server.domain.models import Server
from server.infrastructure.ssh.resolve_server_auth import materialize_server_ssh_auth


@pytest.mark.asyncio
async def test_materialize_password_preset():
    server = Server(
        id=1,
        name="s",
        domain="1.2.3.4",
        auth_method="password",
        preset_id=1,
    )
    preset = MagicMock()
    preset.encrypted_pw = "enc-pw"
    preset.username = "root"

    mock_session = MagicMock()
    with (
        patch(
            "server.infrastructure.database.session.AsyncSessionLocal",
        ) as session_cls,
        patch(
            "server.infrastructure.database.password_preset_repo.PasswordPresetRepositoryImpl"
        ) as repo_cls,
    ):
        session_cls.return_value.__aenter__ = AsyncMock(return_value=mock_session)
        session_cls.return_value.__aexit__ = AsyncMock(return_value=False)
        repo_cls.return_value.get_by_id = AsyncMock(return_value=preset)
        await materialize_server_ssh_auth(server)

    assert server.password == "enc-pw"
    assert server.username == "root"


@pytest.mark.asyncio
async def test_materialize_skips_when_inline_password():
    server = Server(
        id=1,
        name="s",
        domain="1.2.3.4",
        auth_method="password",
        password="already-set",
        preset_id=1,
    )
    with patch(
        "server.infrastructure.database.session.AsyncSessionLocal",
    ) as session_cls:
        session_cls.return_value.__aenter__ = AsyncMock(return_value=MagicMock())
        session_cls.return_value.__aexit__ = AsyncMock(return_value=False)
        with patch(
            "server.infrastructure.database.password_preset_repo.PasswordPresetRepositoryImpl"
        ) as repo_cls:
            repo_cls.return_value.get_by_id = AsyncMock()
            await materialize_server_ssh_auth(server)
            repo_cls.return_value.get_by_id.assert_not_called()