68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
"""Tests for template utilities."""
|
|
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from sslysze_scan.reporter.template_utils import (
|
|
build_template_context,
|
|
format_tls_version,
|
|
generate_report_id,
|
|
)
|
|
|
|
|
|
class TestFormatTlsVersion:
|
|
"""Tests for format_tls_version function."""
|
|
|
|
def test_format_tls_version_all_versions(self) -> None:
|
|
"""Test formatting all known TLS versions."""
|
|
versions = ["1.0", "1.1", "1.2", "1.3", "ssl_3.0", "unknown"]
|
|
expected = ["TLS 1.0", "TLS 1.1", "TLS 1.2", "TLS 1.3", "SSL 3.0", "unknown"]
|
|
assert [format_tls_version(v) for v in versions] == expected
|
|
|
|
|
|
class TestGenerateReportId:
|
|
"""Tests for generate_report_id function."""
|
|
|
|
def test_generate_report_id_valid_and_invalid(self) -> None:
|
|
"""Test report ID generation with valid and invalid timestamps."""
|
|
# Valid timestamp
|
|
metadata = {"timestamp": "2025-01-08T10:30:00.123456", "scan_id": 5}
|
|
result = generate_report_id(metadata)
|
|
assert result == "20250108_5"
|
|
|
|
# Invalid timestamp falls back to current date
|
|
metadata = {"timestamp": "invalid", "scan_id": 5}
|
|
result = generate_report_id(metadata)
|
|
today = datetime.now().strftime("%Y%m%d")
|
|
assert result == f"{today}_5"
|
|
|
|
|
|
class TestBuildTemplateContext:
|
|
"""Tests for build_template_context function."""
|
|
|
|
def test_build_template_context_complete_and_partial(
|
|
self, mock_scan_data: dict[str, Any]
|
|
) -> None:
|
|
"""Test context building with complete and partial data."""
|
|
# Complete data
|
|
context = build_template_context(mock_scan_data)
|
|
assert context["scan_id"] == 5
|
|
assert context["hostname"] == "example.com"
|
|
assert context["fqdn"] == "example.com"
|
|
assert context["ipv4"] == "192.168.1.1"
|
|
assert context["ipv6"] == "2001:db8::1"
|
|
assert context["timestamp"] == "08.01.2025 10:30"
|
|
assert context["duration"] == "12.34"
|
|
assert context["ports"] == "443, 636"
|
|
assert "summary" in context
|
|
assert "ports_data" in context
|
|
|
|
# Verify ports_data sorted by port
|
|
ports = [p["port"] for p in context["ports_data"]]
|
|
assert ports == sorted(ports)
|
|
|
|
# Missing duration
|
|
mock_scan_data["metadata"]["duration"] = None
|
|
context = build_template_context(mock_scan_data)
|
|
assert context["duration"] == "N/A"
|