finish tests for main report

hovey · hovey · commit 4787bcdb4655 · 2025-08-12T14:51:42.000-04:00
diff --git a/src/rattlesnake/cicd/reports_main_page.py b/src/rattlesnake/cicd/reports_main_page.py
@@ -3,7 +3,6 @@
 """
 
 import argparse
-import datetime
 import sys
 from typing import Final
 
@@ -82,17 +81,6 @@
 """
 
 
-# def get_timestamp() -> str:
-#     """
-#     Get a formatted timestamp string.
-#
-#     Returns:
-#         Formatted timestamp string.
-#     """
-#     now: datetime.datetime = datetime.datetime.now(datetime.timezone.utc)
-#     return now.strftime("%Y-%m-%d %H:%M:%S %Z")
-
-
 def get_report_html(github_repo: str, pylint_score: str) -> str:
     """Generates the HTML for the reports landing page.
 
@@ -190,7 +178,7 @@ def main() -> int:
             pylint_score=args.pylint_score,
             output_file=args.output_file,
         )
-        print(f"✅ GitHub Pages main reports page generated: {args.output_file}")
+        print(f"✅ GitHub Pages reports main page generated: {args.output_file}")
     except IOError as e:
         print(f"❌ I/O error occurred: {e}", file=sys.stderr)
         return 1
diff --git a/tests/test_report_lint.py b/tests/test_report_lint.py
@@ -13,7 +13,6 @@
 
 import re
 
-# import sys  # unused import
 import types
 from pathlib import Path
 from typing import Final
diff --git a/tests/test_reports_main_page.py b/tests/test_reports_main_page.py
@@ -0,0 +1,134 @@
+"""
+Unit tests for generate_reports_landing_page.py — GitHub Pages Landing Page Generator.
+
+This test suite verifies the correctness of the functions used to generate
+the main landing page for code quality reports.
+"""
+
+import types
+
+import pytest
+from rattlesnake.cicd.reports_main_page import (
+    get_report_html,
+    main,
+    run_reports_main_page,
+    write_report,
+)
+
+
+def test_write_report_success(tmp_path):
+    """Test that write_report successfully writes to a file."""
+    content = "<html></html>"
+    file_path = tmp_path / "report.html"
+    write_report(content, str(file_path))
+    assert file_path.read_text() == content
+
+
+def test_write_report_io_error(monkeypatch):
+    """Test that write_report raises IOError when file writing fails."""
+
+    def mock_open_raises_io_error(*args, **kwargs):
+        raise IOError("Permission denied")
+
+    monkeypatch.setattr("builtins.open", mock_open_raises_io_error)
+    with pytest.raises(IOError) as excinfo:
+        write_report("<html></html>", "protected_file.html")
+    assert 'Error writing output file "protected_file.html"' in str(excinfo.value)
+    assert "Permission denied" in str(excinfo.value)
+
+
+def test_get_report_html():
+    """Test get_report_html with valid inputs."""
+    report = get_report_html(
+        github_repo="test/repo",
+        pylint_score="9.5",
+    )
+    assert "<!DOCTYPE html>" in report
+    assert "Rattlesnake Code Quality Reports" in report
+    assert "https://github.com/test/repo" in report
+    assert "<strong>Latest Score:</strong> 9.5/10" in report
+
+
+def test_run_generate_landing_page(tmp_path):
+    """Tests the main report creation function."""
+    output_file = tmp_path / "index.html"
+    run_reports_main_page(
+        github_repo="owner/repo",
+        pylint_score="8.8",
+        output_file=str(output_file),
+    )
+    assert output_file.is_file()
+    content = output_file.read_text()
+    assert "<h1>Rattlesnake Code Quality Reports Summary</h1>" in content
+    assert "<strong>Latest Score:</strong> 8.8/10" in content
+    assert 'href="https://github.com/owner/repo"' in content
+
+
+def test_main_success(monkeypatch, capsys):
+    """Test the main function for a successful run."""
+    mock_args = types.SimpleNamespace(
+        github_repo="owner/repo",
+        pylint_score="9.9",
+        output_file="landing.html",
+    )
+
+    monkeypatch.setattr("argparse.ArgumentParser.parse_args", lambda self: mock_args)
+
+    def mock_run(*args, **kwargs):
+        pass
+
+    monkeypatch.setattr(
+        "rattlesnake.cicd.reports_main_page.run_reports_main_page",
+        mock_run,
+    )
+
+    exit_code = main()
+    assert exit_code == 0
+    captured = capsys.readouterr()
+    assert "✅ GitHub Pages reports main page generated: landing.html" in captured.out
+
+
+def test_main_io_error(monkeypatch, capsys):
+    """Test the main function when an IOError occurs."""
+    mock_args = types.SimpleNamespace(
+        github_repo="owner/repo",
+        pylint_score="9.9",
+        output_file="landing.html",
+    )
+    monkeypatch.setattr("argparse.ArgumentParser.parse_args", lambda self: mock_args)
+
+    def mock_run_raises_io_error(*args, **kwargs):
+        raise IOError("Disk full")
+
+    monkeypatch.setattr(
+        "rattlesnake.cicd.reports_main_page.run_reports_main_page",
+        mock_run_raises_io_error,
+    )
+
+    exit_code = main()
+    assert exit_code == 1
+    captured = capsys.readouterr()
+    assert "❌ I/O error occurred: Disk full" in captured.err
+
+
+def test_main_unexpected_error(monkeypatch, capsys):
+    """Test the main function for an unexpected error."""
+    mock_args = types.SimpleNamespace(
+        github_repo="owner/repo",
+        pylint_score="9.9",
+        output_file="landing.html",
+    )
+    monkeypatch.setattr("argparse.ArgumentParser.parse_args", lambda self: mock_args)
+
+    def mock_run_raises_exception(*args, **kwargs):
+        raise Exception("Something bad happened")
+
+    monkeypatch.setattr(
+        "rattlesnake.cicd.reports_main_page.run_reports_main_page",
+        mock_run_raises_exception,
+    )
+
+    exit_code = main()
+    assert exit_code == 1
+    captured = capsys.readouterr()
+    assert "❌ An unexpected error occurred: Something bad happened" in captured.err
