import pytest
import subprocess
from unittest.mock import patch, MagicMock
from autossh_mgr.check import check_connectivity, build_ssh_check_cmd
from autossh_mgr.config import TunnelConfig


@pytest.fixture
def tunnel():
    return TunnelConfig(
        name="web", host="relay.example.com", user="deploy",
        local_port=8080, remote_port=18080,
    )


def test_build_ssh_check_cmd(tunnel):
    cmd = build_ssh_check_cmd(tunnel)
    assert cmd[0] == "ssh"
    assert "-o" in cmd and "BatchMode=yes" in cmd
    assert "ConnectTimeout=5" in " ".join(cmd)
    assert f"{tunnel.user}@{tunnel.host}" in cmd
    assert "true" in cmd
    assert "-p" in cmd and str(tunnel.port) in cmd


def test_check_connectivity_success(tunnel):
    mock_result = MagicMock()
    mock_result.returncode = 0
    mock_result.stderr = ""
    with patch("autossh_mgr.check.subprocess.run", return_value=mock_result) as mock_run:
        success, msg = check_connectivity(tunnel)
    assert success is True
    assert msg == ""
    called_cmd = mock_run.call_args[0][0]
    assert called_cmd == build_ssh_check_cmd(tunnel)


def test_check_connectivity_failure(tunnel):
    mock_result = MagicMock()
    mock_result.returncode = 255
    mock_result.stderr = "Connection refused"
    with patch("autossh_mgr.check.subprocess.run", return_value=mock_result):
        success, msg = check_connectivity(tunnel)
    assert success is False
    assert "Connection refused" in msg