diff --git a/tests/integration/test_pipeline.py b/tests/integration/test_pipeline.py index c38c9fe..fc77ac0 100644 --- a/tests/integration/test_pipeline.py +++ b/tests/integration/test_pipeline.py @@ -1,5 +1,4 @@ import pytest -from unittest.mock import patch, MagicMock from cyberai.core.config import CyberAIConfig from cyberai.core.session import PentestSession, SessionState, Finding, Severity diff --git a/tests/integration/test_reality_probe.py b/tests/integration/test_reality_probe.py index c64b143..a828c6e 100644 --- a/tests/integration/test_reality_probe.py +++ b/tests/integration/test_reality_probe.py @@ -2,8 +2,7 @@ reality-probe integration tests — mock client, no live server needed. """ -import pytest -from unittest.mock import MagicMock, patch +from unittest.mock import MagicMock from cyberai.integrations.reality_probe_client import RealityProbeClient, TLSResult from cyberai.agents.recon.tls_tool import TLSTool from cyberai.agents.intel.tls_cve_mapper import TLSCVEMapper diff --git a/tests/unit/test_attack_metadata.py b/tests/unit/test_attack_metadata.py index 70a0c89..44c556b 100644 --- a/tests/unit/test_attack_metadata.py +++ b/tests/unit/test_attack_metadata.py @@ -1,4 +1,3 @@ -import pytest from cyberai.agents.exploit.attack_metadata import ( enrich_attack_path, enrich_all, diff --git a/tests/unit/test_cache.py b/tests/unit/test_cache.py index c39270f..6a90f45 100644 --- a/tests/unit/test_cache.py +++ b/tests/unit/test_cache.py @@ -1,6 +1,5 @@ import pytest import time -from pathlib import Path from cyberai.core.cache import FileCache diff --git a/tests/unit/test_cve_scorer.py b/tests/unit/test_cve_scorer.py index 375846d..8f81935 100644 --- a/tests/unit/test_cve_scorer.py +++ b/tests/unit/test_cve_scorer.py @@ -3,9 +3,7 @@ score_cve, score_all, CVEScore, - CVSS_WEIGHT, EXPLOIT_WEIGHT, - RECENCY_WEIGHT, EPSS_WEIGHT, ) diff --git a/tests/unit/test_exploit.py b/tests/unit/test_exploit.py index f06563d..31e2e68 100644 --- a/tests/unit/test_exploit.py +++ b/tests/unit/test_exploit.py @@ -1,4 +1,3 @@ -import pytest from cyberai.agents.exploit.cvss_analyzer import analyze_attack_vector from cyberai.agents.exploit.chain_builder import build_exploit_chain diff --git a/tests/unit/test_html_renderer.py b/tests/unit/test_html_renderer.py index 37abed5..fb46ce6 100644 --- a/tests/unit/test_html_renderer.py +++ b/tests/unit/test_html_renderer.py @@ -1,4 +1,3 @@ -import pytest from pathlib import Path from cyberai.agents.report.html_renderer import ( render_html_report, diff --git a/tests/unit/test_intel.py b/tests/unit/test_intel.py index d9cd6ae..6f07a3b 100644 --- a/tests/unit/test_intel.py +++ b/tests/unit/test_intel.py @@ -1,4 +1,3 @@ -import pytest from cyberai.agents.intel.service_mapper import ports_to_queries, score_to_severity from cyberai.agents.intel.nvd_client import _parse_cves diff --git a/tests/unit/test_intel_v2.py b/tests/unit/test_intel_v2.py index 87e8701..92abe93 100644 --- a/tests/unit/test_intel_v2.py +++ b/tests/unit/test_intel_v2.py @@ -9,7 +9,6 @@ from unittest.mock import patch -import pytest from cyberai.agents.intel.agent import IntelAgent, IntelAgentV2, _normalize from cyberai.core.config import CyberAIConfig diff --git a/tests/unit/test_orchestrator.py b/tests/unit/test_orchestrator.py index 69fcce1..fdac622 100644 --- a/tests/unit/test_orchestrator.py +++ b/tests/unit/test_orchestrator.py @@ -1,5 +1,4 @@ -import pytest -from unittest.mock import patch, MagicMock +from unittest.mock import patch from cyberai.core.orchestrator import Orchestrator from cyberai.core.scan_session import ScanPhase, ScanState diff --git a/tests/unit/test_orchestrator_config.py b/tests/unit/test_orchestrator_config.py index 275aa22..499d4c3 100644 --- a/tests/unit/test_orchestrator_config.py +++ b/tests/unit/test_orchestrator_config.py @@ -7,7 +7,6 @@ from __future__ import annotations -import pytest from click.testing import CliRunner from cyberai.__main__ import cli diff --git a/tests/unit/test_phantom_grid.py b/tests/unit/test_phantom_grid.py index bf6c230..5ba6d78 100644 --- a/tests/unit/test_phantom_grid.py +++ b/tests/unit/test_phantom_grid.py @@ -1,4 +1,3 @@ -import pytest from unittest.mock import patch, MagicMock from cyberai.integrations.phantom_grid import PhantomGridClient, OOBInteraction from cyberai.integrations.oob_payloads import ( diff --git a/tests/unit/test_rate_limiter.py b/tests/unit/test_rate_limiter.py index f92240b..11170d8 100644 --- a/tests/unit/test_rate_limiter.py +++ b/tests/unit/test_rate_limiter.py @@ -1,4 +1,3 @@ -import pytest import time from cyberai.core.rate_limiter import ( RateLimiter, diff --git a/tests/unit/test_recon.py b/tests/unit/test_recon.py index 48b08ea..1252cb7 100644 --- a/tests/unit/test_recon.py +++ b/tests/unit/test_recon.py @@ -1,7 +1,4 @@ -import pytest -from unittest.mock import patch, MagicMock from cyberai.agents.recon.nmap_tool import _parse_ports -from cyberai.agents.recon.dns_tool import run_dns def test_parse_ports_empty(): diff --git a/tests/unit/test_report.py b/tests/unit/test_report.py index 89f44ad..9e24a52 100644 --- a/tests/unit/test_report.py +++ b/tests/unit/test_report.py @@ -1,4 +1,3 @@ -import pytest from datetime import datetime, timezone from cyberai.core.session import PentestSession, Finding, Severity from cyberai.agents.report.markdown_renderer import render_markdown diff --git a/tests/unit/test_risk_prioritizer.py b/tests/unit/test_risk_prioritizer.py index 759a31f..46596ac 100644 --- a/tests/unit/test_risk_prioritizer.py +++ b/tests/unit/test_risk_prioritizer.py @@ -1,4 +1,3 @@ -import pytest from cyberai.agents.intel.risk_prioritizer import prioritize, group_by_tier, summarize CVES = [ diff --git a/tests/unit/test_safety_validator.py b/tests/unit/test_safety_validator.py index d50eea7..c76175e 100644 --- a/tests/unit/test_safety_validator.py +++ b/tests/unit/test_safety_validator.py @@ -1,7 +1,5 @@ -import pytest from cyberai.agents.exploit.safety_validator import ( validate_exploit_scope, - _check_target_ip, _target_in_scope, ) diff --git a/tests/unit/test_scan_session.py b/tests/unit/test_scan_session.py index a365051..7e45f8b 100644 --- a/tests/unit/test_scan_session.py +++ b/tests/unit/test_scan_session.py @@ -1,4 +1,3 @@ -import pytest from cyberai.core.scan_session import ScanSession, ScanState, ScanPhase, PhaseResult diff --git a/tests/unit/test_security.py b/tests/unit/test_security.py index 42e7bf5..e4d9b4d 100644 --- a/tests/unit/test_security.py +++ b/tests/unit/test_security.py @@ -1,4 +1,3 @@ -import pytest from cyberai.core.security.injection_detector import detect_injection, scan_messages from cyberai.core.security.input_sanitizer import ( sanitize_target, diff --git a/tests/unit/test_session_shim.py b/tests/unit/test_session_shim.py index 7f27ead..da02b7b 100644 --- a/tests/unit/test_session_shim.py +++ b/tests/unit/test_session_shim.py @@ -7,8 +7,8 @@ def test_legacy_imports_still_work(): from cyberai.core.session import ( - Finding, - PentestSession, + Finding, # noqa: F401 — shim must re-export this legacy name + PentestSession, # noqa: F401 — shim must re-export this legacy name Severity, SessionState, ) diff --git a/tests/unit/test_subdomain_enum.py b/tests/unit/test_subdomain_enum.py index 59682c5..9e73734 100644 --- a/tests/unit/test_subdomain_enum.py +++ b/tests/unit/test_subdomain_enum.py @@ -1,4 +1,3 @@ -import pytest from unittest.mock import patch from cyberai.agents.recon.subdomain_enum import ( enumerate_subdomains, diff --git a/tests/unit/test_tool_calling.py b/tests/unit/test_tool_calling.py index 00646f8..d53ba57 100644 --- a/tests/unit/test_tool_calling.py +++ b/tests/unit/test_tool_calling.py @@ -3,7 +3,7 @@ from __future__ import annotations import json -from typing import Any, Dict, List +from typing import Any, Dict from unittest.mock import MagicMock import pytest