anti
f78104e1c8
Ruff fixes (20 errors → 0): - F401: Remove unused imports (DeckyConfig, random_hostname, IniConfig, COMPOSE_FILE, sys, patch) across cli.py, mutator/engine.py, templates/ftp, templates/rdp, test_mysql.py, test_postgres.py - F541: Remove extraneous f-prefixes on strings with no placeholders in templates/imap, test_ftp_live, test_http_live - E741: Rename ambiguous variable 'l' to descriptive names (line, entry, part) across conftest.py, test_ftp_live, test_http_live, test_mongodb_live, test_pop3, test_ssh SQLite fix: - Change _initialize_sync() admin seeding from SELECT-then-INSERT to INSERT OR IGNORE, preventing IntegrityError when admin user already exists from a previous run
169 lines
4.8 KiB
Python
169 lines
4.8 KiB
Python
"""
|
|
Tests for the SSHService plugin (real OpenSSH, Cowrie removed).
|
|
"""
|
|
|
|
from decnet.services.registry import all_services, get_service
|
|
from decnet.archetypes import get_archetype
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Helpers
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def _fragment(service_cfg: dict | None = None, log_target: str | None = None) -> dict:
|
|
return get_service("ssh").compose_fragment(
|
|
"test-decky", log_target=log_target, service_cfg=service_cfg
|
|
)
|
|
|
|
|
|
def _dockerfile_text() -> str:
|
|
return (get_service("ssh").dockerfile_context() / "Dockerfile").read_text()
|
|
|
|
|
|
def _entrypoint_text() -> str:
|
|
return (get_service("ssh").dockerfile_context() / "entrypoint.sh").read_text()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Registration
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_ssh_registered():
|
|
assert "ssh" in all_services()
|
|
|
|
|
|
def test_real_ssh_not_registered():
|
|
assert "real_ssh" not in all_services()
|
|
|
|
|
|
def test_ssh_ports():
|
|
assert get_service("ssh").ports == [22]
|
|
|
|
|
|
def test_ssh_is_build_service():
|
|
assert get_service("ssh").default_image == "build"
|
|
|
|
|
|
def test_ssh_dockerfile_context_exists():
|
|
svc = get_service("ssh")
|
|
ctx = svc.dockerfile_context()
|
|
assert ctx.is_dir(), f"Dockerfile context missing: {ctx}"
|
|
assert (ctx / "Dockerfile").exists()
|
|
assert (ctx / "entrypoint.sh").exists()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# No Cowrie env vars
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_no_cowrie_vars():
|
|
env = _fragment()["environment"]
|
|
cowrie_keys = [k for k in env if k.startswith("COWRIE_") or k == "NODE_NAME"]
|
|
assert cowrie_keys == [], f"Unexpected Cowrie vars: {cowrie_keys}"
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# compose_fragment structure
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_fragment_has_build():
|
|
frag = _fragment()
|
|
assert "build" in frag and "context" in frag["build"]
|
|
|
|
|
|
def test_fragment_container_name():
|
|
assert _fragment()["container_name"] == "test-decky-ssh"
|
|
|
|
|
|
def test_fragment_restart_policy():
|
|
assert _fragment()["restart"] == "unless-stopped"
|
|
|
|
|
|
def test_fragment_cap_add():
|
|
assert "NET_BIND_SERVICE" in _fragment().get("cap_add", [])
|
|
|
|
|
|
def test_default_password():
|
|
assert _fragment()["environment"]["SSH_ROOT_PASSWORD"] == "admin"
|
|
|
|
|
|
def test_custom_password():
|
|
assert _fragment(service_cfg={"password": "h4x!"})["environment"]["SSH_ROOT_PASSWORD"] == "h4x!"
|
|
|
|
|
|
def test_custom_hostname():
|
|
assert _fragment(service_cfg={"hostname": "prod-db-01"})["environment"]["SSH_HOSTNAME"] == "prod-db-01"
|
|
|
|
|
|
def test_no_hostname_by_default():
|
|
assert "SSH_HOSTNAME" not in _fragment()["environment"]
|
|
|
|
|
|
def test_no_log_target_in_env():
|
|
assert "LOG_TARGET" not in _fragment(log_target="10.0.0.1:5140").get("environment", {})
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Logging pipeline wiring (Dockerfile + entrypoint)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_dockerfile_has_rsyslog():
|
|
assert "rsyslog" in _dockerfile_text()
|
|
|
|
|
|
def test_dockerfile_runs_as_root():
|
|
lines = [line.strip() for line in _dockerfile_text().splitlines()]
|
|
user_lines = [line for line in lines if line.startswith("USER ")]
|
|
assert user_lines == [], f"Unexpected USER directive(s): {user_lines}"
|
|
|
|
|
|
def test_dockerfile_rsyslog_conf_created():
|
|
df = _dockerfile_text()
|
|
assert "99-decnet.conf" in df
|
|
assert "RFC5424fmt" in df
|
|
|
|
|
|
def test_dockerfile_sudoers_syslog():
|
|
df = _dockerfile_text()
|
|
assert "syslog=auth" in df
|
|
assert "log_input" in df
|
|
assert "log_output" in df
|
|
|
|
|
|
def test_dockerfile_prompt_command_logger():
|
|
df = _dockerfile_text()
|
|
assert "PROMPT_COMMAND" in df
|
|
assert "logger" in df
|
|
|
|
|
|
def test_entrypoint_creates_named_pipe():
|
|
assert "mkfifo" in _entrypoint_text()
|
|
|
|
|
|
def test_entrypoint_starts_rsyslogd():
|
|
assert "rsyslogd" in _entrypoint_text()
|
|
|
|
|
|
def test_entrypoint_sshd_no_dash_e():
|
|
ep = _entrypoint_text()
|
|
assert "sshd -D" in ep
|
|
assert "sshd -D -e" not in ep
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Deaddeck archetype
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def test_deaddeck_uses_ssh():
|
|
arch = get_archetype("deaddeck")
|
|
assert "ssh" in arch.services
|
|
assert "real_ssh" not in arch.services
|
|
|
|
|
|
def test_deaddeck_nmap_os():
|
|
assert get_archetype("deaddeck").nmap_os == "linux"
|
|
|
|
|
|
def test_deaddeck_preferred_distros_not_empty():
|
|
assert len(get_archetype("deaddeck").preferred_distros) >= 1
|