nightshift/tests/test_reports.py

61 lines
2.0 KiB
Python

from pathlib import Path
import tempfile
import unittest
from nightshift.artifacts import ArtifactStore
from nightshift.reports import ReportGenerator
from nightshift.stages import StageResult
from nightshift.tasks import parse_tasks
TASK_MD = """# Tasks
- [ ] TASK-001: Report results
Description:
Write summaries.
Acceptance Criteria:
- Final notes explain status
- Run summary includes artifacts
"""
class ReportGeneratorTests(unittest.TestCase):
def test_writes_final_notes_stage_results_and_run_summary(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = Path(directory)
artifacts = ArtifactStore(root, ".nightshift", run_id="test-run")
reporter = ReportGenerator(root, artifacts)
task = parse_tasks(TASK_MD)[0]
context_out = artifacts.write_stage_output(task.id, "context-out.md", "# Context Out\n")
report = reporter.write_reports(
task,
"complete",
"done",
1,
[
StageResult(
stage_id="test",
status="pass",
reason="ok",
output_path=".nightshift/runs/test-run/tasks/TASK-001/test-output.txt",
)
],
context_out_path=context_out,
)
self.assertTrue(report.final_notes_path.exists())
self.assertTrue(report.stage_results_path.exists())
self.assertTrue(report.run_summary_path.exists())
self.assertTrue(report.devlog_path.exists())
self.assertIn("Retry count: 1", report.final_notes_path.read_text(encoding="utf-8"))
self.assertIn("test", report.stage_results_path.read_text(encoding="utf-8"))
self.assertIn("Final notes", report.run_summary_path.read_text(encoding="utf-8"))
self.assertIn("Tests reported", report.devlog_path.read_text(encoding="utf-8"))
if __name__ == "__main__":
unittest.main()