45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import pytest
|
|
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
|