test: add unit tests for JSONSummaryReporter (#149)

Signed-off-by: Ayush-Patel-56 <ayushpatel2731@gmail.com>

Author: Ayush-Patel-56

tests/unit/reporter/test_json_summary_reporter.py

@@ -0,0 +1,111 @@
+"""
+Unit tests for JSONSummaryReporter
+"""
+
+import os
+import json
+import datetime
+
+from krkn_ai.reporter.json_summary_reporter import JSONSummaryReporter
+from krkn_ai.models.app import CommandRunResult, FitnessResult
+from krkn_ai.models.scenario.scenario_dummy import DummyScenario
+
+
+class TestJSONSummaryReporter:
+    def _create_results(self, gen_id, start_score, count, scenario, now):
+        results = {}
+        for i in range(count):
+            sid = (gen_id * 100) + i
+            score = float(start_score + (i * 10))
+            res = CommandRunResult(
+                generation_id=gen_id,
+                scenario_id=sid,
+                scenario=scenario,
+                cmd="test",
+                log="test",
+                returncode=0,
+                start_time=now,
+                end_time=now,
+                fitness_result=FitnessResult(fitness_score=score),
+            )
+            results[sid] = res
+        return results
+
+    def test_generate_summary_content(self, minimal_config):
+        now = datetime.datetime(2023, 1, 1, 12, 0, 0)
+        cc = minimal_config.cluster_components
+        scenario = DummyScenario(cluster_components=cc)
+        
+        gen0 = self._create_results(0, 0, 3, scenario, now)
+        gen1 = self._create_results(1, 40, 3, scenario, now)
+        
+        pop = {**gen0, **gen1}
+        best = [gen0[2], gen1[102]]
+
+        reporter = JSONSummaryReporter(
+            run_uuid="test-run",
+            config=minimal_config,
+            seen_population=pop,
+            best_of_generation=best,
+            start_time=now,
+            end_time=now + datetime.timedelta(seconds=100),
+            completed_generations=2,
+            seed=123
+        )
+        
+        summary = reporter.generate_summary()
+
+        assert summary["run_id"] == "test-run"
+        assert summary["duration_seconds"] == 100.0
+        assert summary["summary"]["total_scenarios_executed"] == 6
+        assert summary["summary"]["best_fitness_score"] == 60.0
+        
+        assert summary["fitness_progression"][0]["average"] == 10.0
+        assert summary["fitness_progression"][1]["average"] == 50.0
+        assert summary["fitness_progression"][1]["best"] == 60.0
+
+    def test_best_scenarios_ranking(self, minimal_config):
+        now = datetime.datetime.now()
+        cc = minimal_config.cluster_components
+        scenario = DummyScenario(cluster_components=cc)
+        
+        pop = self._create_results(0, 0, 15, scenario, now)
+        
+        reporter = JSONSummaryReporter(
+            run_uuid="test",
+            config=minimal_config,
+            seen_population=pop,
+            best_of_generation=[],
+        )
+        
+        best = reporter.generate_summary()["best_scenarios"]
+        assert len(best) == 10
+        assert best[0]["rank"] == 1
+        assert best[0]["fitness_score"] == 140.0
+        assert best[0]["parameters"]["end"] == 10
+
+    def test_empty_population(self, minimal_config):
+        reporter = JSONSummaryReporter(
+            run_uuid="empty",
+            config=minimal_config,
+            seen_population={},
+            best_of_generation=[],
+        )
+        summary = reporter.generate_summary()
+        assert summary["summary"]["total_scenarios_executed"] == 0
+        assert summary["best_scenarios"] == []
+
+    def test_save_json_file(self, minimal_config, temp_output_dir):
+        reporter = JSONSummaryReporter(
+            run_uuid="save-test",
+            config=minimal_config,
+            seen_population={},
+            best_of_generation=[],
+        )
+        reporter.save(temp_output_dir)
+        
+        path = os.path.join(temp_output_dir, "results.json")
+        assert os.path.exists(path)
+        with open(path, "r") as f:
+            content = json.load(f)
+            assert content["run_id"] == "save-test"