feat: initial release

tests/test_template_utils.py

@@ -0,0 +1,67 @@
+"""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"