fix: unittest generator latest run id null

Author: jungs1

src/mutahunter/core/controller.py

@@ -1,34 +1,30 @@
+import json
 import os
 import time
 from subprocess import CompletedProcess
 from typing import Any, Dict, List, Optional
 from uuid import uuid4
 
-from tqdm import tqdm
-from md2pdf.core import md2pdf
 from jinja2 import Template
-import json
+from md2pdf.core import md2pdf
+from tqdm import tqdm
+
 from mutahunter.core.analyzer import Analyzer
 from mutahunter.core.coverage_processor import CoverageProcessor
 from mutahunter.core.db import MutationDatabase
 from mutahunter.core.entities.config import MutationTestControllerConfig
 from mutahunter.core.error_parser import extract_error_message
-from mutahunter.core.exceptions import (
-    CoverageAnalysisError,
-    MutantKilledError,
-    MutantSurvivedError,
-    MutationTestingError,
-    ReportGenerationError,
-    UnexpectedTestResultError,
-)
+from mutahunter.core.exceptions import (CoverageAnalysisError,
+                                        MutantKilledError, MutantSurvivedError,
+                                        MutationTestingError,
+                                        ReportGenerationError,
+                                        UnexpectedTestResultError)
 from mutahunter.core.git_handler import GitHandler
 from mutahunter.core.io import FileOperationHandler
 from mutahunter.core.llm_mutation_engine import LLMMutationEngine
 from mutahunter.core.logger import logger
 from mutahunter.core.prompts.mutant_generator import (
-    SYSTEM_PROMPT_MUTANT_ANALYSUS,
-    USER_PROMPT_MUTANT_ANALYSIS,
-)
+    SYSTEM_PROMPT_MUTANT_ANALYSUS, USER_PROMPT_MUTANT_ANALYSIS)
 from mutahunter.core.report import MutantReport
 from mutahunter.core.router import LLMRouter
 from mutahunter.core.runner import MutantTestRunner

src/mutahunter/core/db.py

@@ -2,8 +2,8 @@
 import os
 import sqlite3
 from contextlib import contextmanager
-from typing import Any, Dict, List, Optional, Tuple
 from datetime import datetime
+from typing import Any, Dict, List, Optional, Tuple
 
 
 class DatabaseError(Exception):
@@ -502,6 +502,16 @@ def remove_mutants_by_file_version_id(self, file_version_id: int):
                 conn.rollback()
                 raise DatabaseError(f"Error removing mutants: {str(e)}")
 
+    def get_latest_run_id(self) -> Optional[int]:
+        with self.get_connection() as conn:
+            cursor = conn.cursor()
+            try:
+                cursor.execute("SELECT id FROM Runs ORDER BY id DESC LIMIT 1")
+                result = cursor.fetchone()
+                return result[0] if result else None
+            except sqlite3.Error as e:
+                raise DatabaseError(f"Error fetching latest run ID: {str(e)}")
+
     def get_mutation_coverage(self) -> float:
         summary = self.get_mutant_summary()
         if not summary:

src/mutahunter/core/report.py

@@ -3,15 +3,12 @@
 """
 
 import os
+from importlib import resources
 from typing import Any, Dict, List
 
-from jinja2 import (
-    Environment,
-    PackageLoader,
-    FileSystemLoader,
-    select_autoescape,
-)
-from importlib import resources
+from jinja2 import (Environment, FileSystemLoader, PackageLoader,
+                    select_autoescape)
+
 from mutahunter.core.db import MutationDatabase
 from mutahunter.core.logger import logger
 

src/mutahunter/core/unittest_generator.py

@@ -73,16 +73,20 @@ def run(self) -> None:
         initial_line_coverage_rate = self.coverage_processor.line_coverage_rate
         print("Initial line coverage rate:", initial_line_coverage_rate)
         self.increase_line_coverage()
-        self.mutator.run_coverage_analysis()
-        self.mutator.run_mutation_testing()
-        initial_mutation_coverage_rate = self.db.get_mutation_coverage()
+        self.mutator.run()
+        latest_run_id = self.db.get_latest_run_id()
+        print("Latest run ID:", latest_run_id)
+        data = self.db.get_mutant_summary(latest_run_id)
+        initial_mutation_coverage_rate = data["mutation_coverage"]
         print("Initial mutation coverage rate:", initial_mutation_coverage_rate)
         self.increase_mutation_coverage()
         print(
             f"Line coverage increased from {initial_line_coverage_rate*100:.2f}% to {self.coverage_processor.line_coverage_rate*100:.2f}%"
         )
+        data = self.db.get_mutant_summary(latest_run_id)
+        final_mutation_coverage_rate = data["mutation_coverage"]
         print(
-            f"Mutation coverage increased from {initial_mutation_coverage_rate*100:.2f}% to {self.db.get_mutation_coverage()*100:.2f}%"
+            f"Mutation coverage increased from {initial_mutation_coverage_rate*100:.2f}% to {final_mutation_coverage_rate*100:.2f}%"
         )
 
     def increase_line_coverage(self):
@@ -99,10 +103,12 @@ def increase_line_coverage(self):
 
     def increase_mutation_coverage(self):
         attempt = 0
-
+        latest_run_id = self.db.get_latest_run_id()
+        data = self.db.get_mutant_summary(latest_run_id)
+        mutation_coverage_rate = data["mutation_coverage"]
         self.failed_unittests = []
         while (
-            self.db.get_mutation_coverage() < self.config.target_mutation_coverage_rate
+            mutation_coverage_rate < self.config.target_mutation_coverage_rate
             and attempt < self.config.max_attempts
         ):
             attempt += 1

src/mutahunter/main.py

@@ -5,10 +5,8 @@
 from mutahunter.core.controller import MutationTestController
 from mutahunter.core.coverage_processor import CoverageProcessor
 from mutahunter.core.db import MutationDatabase
-from mutahunter.core.entities.config import (
-    MutationTestControllerConfig,
-    UnittestGeneratorConfig,
-)
+from mutahunter.core.entities.config import (MutationTestControllerConfig,
+                                             UnittestGeneratorConfig)
 from mutahunter.core.error_parser import extract_error_message
 from mutahunter.core.git_handler import GitHandler
 from mutahunter.core.io import FileOperationHandler