Feature-339: Accept --max-run-time-sec in test_with_docker.sh

cover_agent/cover_agent.py

@@ -316,7 +316,7 @@ def finalize_test_generation(self, iteration_count):
         if self.test_validator.current_coverage >= (desired_coverage / 100):
             self.logger.info(
                 f"Reached above target coverage of {desired_coverage}% "
-                f"(Current Coverage: {current_coverage}%) in {iteration_count} iterations."
+                f"(Current Coverage: {current_coverage}%) in {iteration_count + 1} iterations."
             )
         elif iteration_count == self.config.max_iterations:
             coverage_type = "diff coverage" if self.config.diff_coverage else "coverage"

cover_agent/lsp_logic/ContextHelper.py

@@ -2,7 +2,7 @@
 from contextlib import asynccontextmanager
 from pathlib import Path
 from typing import AsyncIterator, List, Tuple, Optional
-from cover_agent.AICaller import AICaller
+from cover_agent.ai_caller import AICaller
 from cover_agent.lsp_logic.utils.utils_context import (
     analyze_context,
     find_test_file_context,

templated_tests/python_fastapi/test_app.py

@@ -13,3 +13,9 @@ def test_root():
     response = client.get("/")
     assert response.status_code == 200
     assert response.json() == {"message": "Welcome to the FastAPI application!"}
+
+
+def test_divide_by_zero():
+    response = client.get("/divide/10/0")
+    assert response.status_code == 400
+    assert response.json() == {"detail": "Cannot divide by zero"}

tests_integration/test_all.sh

@@ -98,6 +98,7 @@ sh tests_integration/test_with_docker.sh \
   --coverage-type "jacoco" \
   --code-coverage-report-path "build/reports/jacoco/test/jacocoTestReport.csv" \
   --model $MODEL \
+  --max-run-time-sec 240 \
   $log_db_arg \
   $SUPPRESS_LOG_FILES
 

tests_integration/test_with_docker.sh

@@ -17,6 +17,7 @@ COVERAGE_TYPE="cobertura"
 CODE_COVERAGE_REPORT_PATH="coverage.xml"
 MAX_ITERATIONS=3  # Default value
 DESIRED_COVERAGE=70  # Default value
+MAX_RUN_TIME_SEC=30 # Default value
 LOG_DB_PATH="${LOG_DB_PATH:-}"
 SUPPRESS_LOG_FILES=""
 
@@ -46,6 +47,7 @@ while [ "$#" -gt 0 ]; do
     --coverage-type) COVERAGE_TYPE="$2"; shift ;;
     --code-coverage-report-path) CODE_COVERAGE_REPORT_PATH="$2"; shift ;;
     --max-iterations) MAX_ITERATIONS="$2"; shift ;;
+    --max-run-time-sec) MAX_RUN_TIME_SEC="$2"; shift ;;
     --desired-coverage) DESIRED_COVERAGE="$2"; shift ;;
     --log-db-path) LOG_DB_PATH="$2"; shift ;;
     --suppress-log-files) SUPPRESS_LOG_FILES="--suppress-log-files" ;;
@@ -115,6 +117,10 @@ COMMAND="/usr/local/bin/cover-agent \
   --max-iterations $MAX_ITERATIONS \
   --strict-coverage"
 
+if [ -n "$MAX_RUN_TIME_SEC" ]; then
+  COMMAND="$COMMAND --max-run-time-sec $MAX_RUN_TIME_SEC"
+fi
+
 if [ -n "$MODEL" ]; then
   COMMAND="$COMMAND --model \"$MODEL\""
 fi