Tests: Significantly improved coverage of the unit tests

benchmarks/automation_profile_cli_bench.py

@@ -14,7 +14,6 @@ import os
 import sys
 import tempfile
 from pathlib import Path
-from unittest.mock import patch
 
 # Ensure the local *source* tree is importable even when ASV has an
 # older build of the package installed.
@@ -30,10 +29,9 @@ from typer.testing import CliRunner  # noqa: E402
 
 from cleveragents.cli.commands.automation_profile import (  # noqa: E402
     _InMemoryProfileRepository,
-    app as profile_app,
 )
-from cleveragents.domain.models.core.automation_profile import (  # noqa: E402
-    AutomationProfile,
+from cleveragents.cli.commands.automation_profile import (
+    app as profile_app,
 )
 
 _VALID_YAML = """\

benchmarks/cleanup_bench.py

@@ -7,10 +7,8 @@ cleanup overhead stays within acceptable bounds.
 from __future__ import annotations
 
 import importlib
-import os
 import sys
 import tempfile
-import time
 from pathlib import Path
 
 # Ensure the local *source* tree is importable.
@@ -22,6 +20,8 @@ import cleveragents  # noqa: E402
 
 importlib.reload(cleveragents)
 
+from datetime import UTC
+
 from cleveragents.application.services.cleanup_service import (  # noqa: E402
     CleanupService,
 )
@@ -84,7 +84,7 @@ class SessionScanBenchmark:
     """Benchmark session inactivity scanning."""
 
     def setup(self) -> None:
-        from datetime import datetime, timedelta, timezone
+        from datetime import datetime, timedelta
 
         self.settings = _make_settings(
             cleanup_session_inactivity_days=30,
@@ -93,7 +93,7 @@ class SessionScanBenchmark:
             settings=self.settings,
             active_plan_ids=frozenset(),
         )
-        now = datetime.now(tz=timezone.utc)
+        now = datetime.now(tz=UTC)
         self.sessions = [
             {
                 "session_id": f"sess-{i:04d}",

benchmarks/plan_diff_bench.py

@@ -26,7 +26,6 @@ importlib.reload(cleveragents)
 
 from cleveragents.application.services.plan_apply_service import (  # noqa: E402
     PlanApplyService,
-    _build_artifacts_dict,
     _render_diff_json,
     _render_diff_plain,
     _render_diff_rich,
@@ -41,7 +40,7 @@ from cleveragents.domain.models.core.change import (  # noqa: E402
     InMemoryChangeSetStore,
     SpecChangeSet,
 )
-from cleveragents.domain.models.core.plan import PlanPhase, ProcessingState  # noqa: E402
+from cleveragents.domain.models.core.plan import PlanPhase  # noqa: E402
 
 
 def _build_changeset(plan_id: str, n_entries: int = 10) -> SpecChangeSet:

benchmarks/project_repository_bench.py

@@ -15,7 +15,6 @@ from sqlalchemy.orm import sessionmaker
 from cleveragents.domain.models.core.project import NamespacedProject
 from cleveragents.infrastructure.database.models import (
     Base,
-    NamespacedProjectModel,
     ResourceModel,
     ResourceTypeModel,
 )

benchmarks/security_audit_bench.py

@@ -11,14 +11,11 @@ from datetime import UTC, datetime, timedelta
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
-from cleveragents.application.services.audit_service import AuditService
+from cleveragents.application.services.audit_service import _TIMESTAMP_FMT, AuditService
 from cleveragents.config.settings import Settings
 from cleveragents.infrastructure.database.models import AuditLogModel, Base
 
 
-from cleveragents.application.services.audit_service import _TIMESTAMP_FMT
-
-
 def _make_service() -> AuditService:
     """Create a service backed by in-memory SQLite."""
     Settings._instance = None

benchmarks/skill_context_bench.py

@@ -35,7 +35,6 @@ except ModuleNotFoundError:
     from cleveragents.skills.context import SkillContext
     from cleveragents.skills.protocol import (
         SkillDefinition,
-        SkillError,
         SkillMetadata,
     )
     from cleveragents.skills.registry import SkillRegistry