mockforge_bench/

parallel_executor.rs

//! Parallel execution engine for multi-target bench testing
//!
//! Executes load tests against multiple targets in parallel with configurable
//! concurrency limits. Uses tokio for async execution and semaphores for
//! backpressure control.

use crate::command::BenchCommand;
use crate::error::{BenchError, Result};
use crate::executor::{K6Executor, K6Results};
use crate::k6_gen::{K6Config, K6ScriptGenerator};
use crate::reporter::TerminalReporter;
use crate::request_gen::RequestGenerator;
use crate::scenarios::LoadScenario;
use crate::spec_parser::SpecParser;
use crate::target_parser::TargetConfig;
use indicatif::{MultiProgress, ProgressBar, ProgressStyle};
use mockforge_openapi::spec::OpenApiSpec;
use std::collections::HashMap;
use std::path::{Path, PathBuf};
use std::str::FromStr;
use std::sync::Arc;
use tokio::sync::Semaphore;
use tokio::task::JoinHandle;

/// Result for a single target execution
#[derive(Debug, Clone)]
pub struct TargetResult {
    /// Target URL that was tested
    pub target_url: String,
    /// Index of the target (for ordering)
    pub target_index: usize,
    /// k6 test results
    pub results: K6Results,
    /// Output directory for this target
    pub output_dir: PathBuf,
    /// Whether the test succeeded
    pub success: bool,
    /// Error message if test failed
    pub error: Option<String>,
}

/// Aggregated results from all target executions
#[derive(Debug, Clone)]
pub struct AggregatedResults {
    /// Results for each target
    pub target_results: Vec<TargetResult>,
    /// Overall statistics
    pub total_targets: usize,
    pub successful_targets: usize,
    pub failed_targets: usize,
    /// Aggregated metrics across all targets
    pub aggregated_metrics: AggregatedMetrics,
}

/// Aggregated metrics across all targets
#[derive(Debug, Clone)]
pub struct AggregatedMetrics {
    /// Total requests across all targets
    pub total_requests: u64,
    /// Total failed requests across all targets
    pub total_failed_requests: u64,
    /// Average response time across all targets (ms)
    pub avg_duration_ms: f64,
    /// p95 response time across all targets (ms)
    pub p95_duration_ms: f64,
    /// p99 response time across all targets (ms)
    pub p99_duration_ms: f64,
    /// Overall error rate percentage
    pub error_rate: f64,
    /// Total RPS across all targets
    pub total_rps: f64,
    /// Average RPS per target
    pub avg_rps: f64,
    /// Total max VUs across all targets
    pub total_vus_max: u32,
    /// Total connections opened across all targets (Issue #79 round 12 —
    /// Srikanth's multi-target bench output was missing the CPS / connection
    /// counts that single-target runs surface).
    pub total_connections_opened: u64,
    /// Total iterations completed across all targets.
    pub total_iterations_completed: u64,
}

impl AggregatedMetrics {
    /// Calculate aggregated metrics from target results
    fn from_results(results: &[TargetResult]) -> Self {
        let mut total_requests = 0u64;
        let mut total_failed_requests = 0u64;
        let mut durations = Vec::new();
        let mut p95_values = Vec::new();
        let mut p99_values = Vec::new();
        let mut total_rps = 0.0f64;
        let mut total_vus_max = 0u32;
        let mut total_connections_opened = 0u64;
        let mut total_iterations_completed = 0u64;
        let mut successful_count = 0usize;

        for result in results {
            if result.success {
                total_requests += result.results.total_requests;
                total_failed_requests += result.results.failed_requests;
                durations.push(result.results.avg_duration_ms);
                p95_values.push(result.results.p95_duration_ms);
                p99_values.push(result.results.p99_duration_ms);
                total_rps += result.results.rps;
                total_vus_max += result.results.vus_max;
                total_connections_opened += result.results.tcp_connect_samples;
                total_iterations_completed += result.results.iterations_completed;
                successful_count += 1;
            }
        }

        let avg_duration_ms = if !durations.is_empty() {
            durations.iter().sum::<f64>() / durations.len() as f64
        } else {
            0.0
        };

        let p95_duration_ms = if !p95_values.is_empty() {
            p95_values.sort_by(|a, b| a.partial_cmp(b).unwrap_or(std::cmp::Ordering::Equal));
            let index = (p95_values.len() as f64 * 0.95).ceil() as usize - 1;
            p95_values[index.min(p95_values.len() - 1)]
        } else {
            0.0
        };

        let p99_duration_ms = if !p99_values.is_empty() {
            p99_values.sort_by(|a, b| a.partial_cmp(b).unwrap_or(std::cmp::Ordering::Equal));
            let index = (p99_values.len() as f64 * 0.99).ceil() as usize - 1;
            p99_values[index.min(p99_values.len() - 1)]
        } else {
            0.0
        };

        let error_rate = if total_requests > 0 {
            (total_failed_requests as f64 / total_requests as f64) * 100.0
        } else {
            0.0
        };

        let avg_rps = if successful_count > 0 {
            total_rps / successful_count as f64
        } else {
            0.0
        };

        Self {
            total_requests,
            total_failed_requests,
            avg_duration_ms,
            p95_duration_ms,
            p99_duration_ms,
            error_rate,
            total_rps,
            avg_rps,
            total_vus_max,
            total_connections_opened,
            total_iterations_completed,
        }
    }
}

/// Round 67 (#79) — aggregated-metrics JSON object shared by
/// `aggregated_summary.json`, per-round `round_summary.json` and the
/// `campaign.jsonl` lines so the three can't drift.
fn aggregated_metrics_json(m: &AggregatedMetrics) -> serde_json::Value {
    serde_json::json!({
        "total_requests": m.total_requests,
        "total_failed_requests": m.total_failed_requests,
        "avg_duration_ms": m.avg_duration_ms,
        "p95_duration_ms": m.p95_duration_ms,
        "p99_duration_ms": m.p99_duration_ms,
        "error_rate": m.error_rate,
        "total_rps": m.total_rps,
        "avg_rps": m.avg_rps,
        "total_vus_max": m.total_vus_max,
        "total_connections_opened": m.total_connections_opened,
        "total_iterations_completed": m.total_iterations_completed,
    })
}

/// Round 67 (#79) — per-target JSON rows shared by every stats file.
fn target_results_json(results: &AggregatedResults) -> Vec<serde_json::Value> {
    results
        .target_results
        .iter()
        .map(|r| {
            serde_json::json!({
                "target_url": r.target_url,
                "target_index": r.target_index,
                "success": r.success,
                "error": r.error,
                "total_requests": r.results.total_requests,
                "failed_requests": r.results.failed_requests,
                "avg_duration_ms": r.results.avg_duration_ms,
                "min_duration_ms": r.results.min_duration_ms,
                "med_duration_ms": r.results.med_duration_ms,
                "p90_duration_ms": r.results.p90_duration_ms,
                "p95_duration_ms": r.results.p95_duration_ms,
                "p99_duration_ms": r.results.p99_duration_ms,
                "max_duration_ms": r.results.max_duration_ms,
                "rps": r.results.rps,
                "vus_max": r.results.vus_max,
                "output_dir": r.output_dir.to_string_lossy(),
            })
        })
        .collect()
}

/// JSON body of `aggregated_summary.json` (the campaign-level file written
/// once at the end of a multi-target run).
pub(crate) fn targets_summary_json(
    results: &AggregatedResults,
    elapsed_secs: u64,
) -> serde_json::Value {
    serde_json::json!({
        "total_elapsed_seconds": elapsed_secs,
        "total_targets": results.total_targets,
        "successful_targets": results.successful_targets,
        "failed_targets": results.failed_targets,
        "aggregated_metrics": aggregated_metrics_json(&results.aggregated_metrics),
        "target_results": target_results_json(results),
    })
}

/// Per-target CSV rows shared by `all_targets.csv` (base output and the
/// per-round copy inside each `round_*` dir).
pub(crate) fn targets_csv(results: &AggregatedResults) -> String {
    let mut csv = String::from(
        "target_url,success,requests,failed,rps,vus,min_ms,avg_ms,med_ms,p90_ms,p95_ms,p99_ms,max_ms,error\n",
    );
    for r in &results.target_results {
        csv.push_str(&format!(
            "{},{},{},{},{:.1},{},{:.1},{:.1},{:.1},{:.1},{:.1},{:.1},{:.1},{}\n",
            r.target_url,
            r.success,
            r.results.total_requests,
            r.results.failed_requests,
            r.results.rps,
            r.results.vus_max,
            r.results.min_duration_ms,
            r.results.avg_duration_ms,
            r.results.med_duration_ms,
            r.results.p90_duration_ms,
            r.results.p95_duration_ms,
            r.results.p99_duration_ms,
            r.results.max_duration_ms,
            r.error.as_deref().unwrap_or(""),
        ));
    }
    csv
}

/// Parallel executor for multi-target bench testing
pub struct ParallelExecutor {
    /// Base command configuration (shared across all targets)
    base_command: BenchCommand,
    /// List of targets to test
    targets: Vec<TargetConfig>,
    /// Explicit `--max-concurrency` override. `None` means auto (10, or 3
    /// for huge specs — round 65 / #79).
    max_concurrency_override: Option<usize>,
    /// Base output directory
    base_output: PathBuf,
}

impl ParallelExecutor {
    /// Create a new parallel executor
    pub fn new(
        base_command: BenchCommand,
        targets: Vec<TargetConfig>,
        max_concurrency: Option<usize>,
    ) -> Self {
        let base_output = base_command.output.clone();
        Self {
            base_command,
            targets,
            max_concurrency_override: max_concurrency,
            base_output,
        }
    }

    /// Round 63 (#79): `--max-concurrency` is a semaphore, not a shared VU
    /// pool. Each batch of up to `max_concurrency` targets runs the full
    /// duration, then the next batch starts. Wall clock ≈
    /// ceil(targets / concurrency) * duration.
    pub(crate) fn estimated_wall_clock(
        n_targets: usize,
        max_concurrency: usize,
        duration_secs: u64,
    ) -> (usize, u64) {
        let conc = max_concurrency.max(1);
        let batches = if n_targets == 0 {
            0
        } else {
            n_targets.div_ceil(conc)
        };
        (batches, (batches as u64).saturating_mul(duration_secs))
    }

    /// Human-readable duration for the wall-clock estimate (`35m00s`).
    pub(crate) fn format_wall_clock(secs: u64) -> String {
        let hours = secs / 3600;
        let mins = (secs % 3600) / 60;
        let rem = secs % 60;
        if hours > 0 {
            format!("{hours}h{mins:02}m{rem:02}s")
        } else if mins > 0 {
            format!("{mins}m{rem:02}s")
        } else {
            format!("{rem}s")
        }
    }

    /// Execute tests against all targets in parallel
    pub async fn execute_all(&self) -> Result<AggregatedResults> {
        let total_targets = self.targets.len();
        TerminalReporter::print_progress(&format!(
            "Starting parallel execution for {} targets",
            total_targets
        ));

        // Validate k6 installation
        if !K6Executor::is_k6_installed() {
            TerminalReporter::print_error("k6 is not installed");
            TerminalReporter::print_warning(
                "Install k6 from: https://k6.io/docs/get-started/installation/",
            );
            return Err(BenchError::K6NotFound);
        }

        // #79: --targets-file used to ignore --wafbench-verbatim. It always
        // required a spec and built templates from spec operations, so the
        // command either died with "No spec files provided" or fuzzed spec
        // URLs instead of sending the traffic file as written.
        let spec_supplied =
            !self.base_command.spec.is_empty() || self.base_command.spec_dir.is_some();
        let verbatim = self.base_command.wafbench_verbatim;

        let (templates, parser) = if verbatim {
            let verbatim_templates = self.base_command.load_verbatim_templates()?;
            if verbatim_templates.is_empty() {
                return Err(BenchError::Other(
                    "--wafbench-verbatim was set but no traffic cases were loaded. Check \
                     --wafbench-dir points at a file, directory or glob containing cases with \
                     a `request.uri`."
                        .to_string(),
                ));
            }
            TerminalReporter::print_success(&format!(
                "Verbatim mode: {} request(s) will be sent exactly as written (spec endpoints not used)",
                verbatim_templates.len()
            ));
            let parser = if spec_supplied {
                TerminalReporter::print_progress("Loading OpenAPI specification(s)...");
                let merged_spec = self.base_command.load_and_merge_specs().await?;
                TerminalReporter::print_success("Specification(s) loaded (base path only)");
                SpecParser::from_spec(merged_spec)
            } else {
                SpecParser::from_spec(OpenApiSpec {
                    spec: Default::default(),
                    file_path: None,
                    raw_document: None,
                })
            };
            (verbatim_templates, parser)
        } else {
            TerminalReporter::print_progress("Loading OpenAPI specification(s)...");
            let merged_spec = self.base_command.load_and_merge_specs().await?;
            let parser = SpecParser::from_spec(merged_spec);
            TerminalReporter::print_success("Specification(s) loaded");

            let operations = if let Some(filter) = &self.base_command.operations {
                parser.filter_operations(filter)?
            } else {
                parser.get_operations()
            };

            if operations.is_empty() {
                return Err(BenchError::Other("No operations found in spec".to_string()));
            }

            TerminalReporter::print_success(&format!("Found {} operations", operations.len()));

            TerminalReporter::print_progress("Generating request templates...");
            let templates: Vec<_> = operations
                .iter()
                .map(RequestGenerator::generate_template)
                .collect::<Result<Vec<_>>>()?;
            TerminalReporter::print_success("Request templates generated");
            (templates, parser)
        };

        // Pre-load per-target specs. In verbatim mode a per-target spec may
        // still resolve --base-path, but it must not replace the traffic-file
        // templates. That swap is how --targets-file kept fuzzing spec URLs
        // after --wafbench-verbatim was added (#79).
        let mut per_target_data: HashMap<
            PathBuf,
            (Vec<crate::request_gen::RequestTemplate>, Option<String>),
        > = HashMap::new();
        if !verbatim {
            let mut unique_specs: Vec<PathBuf> = Vec::new();
            for t in &self.targets {
                if let Some(spec_path) = &t.spec {
                    if !unique_specs.contains(spec_path) {
                        unique_specs.push(spec_path.clone());
                    }
                }
            }
            for spec_path in &unique_specs {
                TerminalReporter::print_progress(&format!(
                    "Loading per-target spec: {}",
                    spec_path.display()
                ));
                match SpecParser::from_file(spec_path).await {
                    Ok(target_parser) => {
                        let target_ops = if let Some(filter) = &self.base_command.operations {
                            match target_parser.filter_operations(filter) {
                                Ok(ops) => ops,
                                Err(e) => {
                                    TerminalReporter::print_warning(&format!(
                                        "Failed to filter operations from {}: {}. Using shared spec.",
                                        spec_path.display(),
                                        e
                                    ));
                                    continue;
                                }
                            }
                        } else {
                            target_parser.get_operations()
                        };
                        let target_templates: Vec<_> = match target_ops
                            .iter()
                            .map(RequestGenerator::generate_template)
                            .collect::<Result<Vec<_>>>()
                        {
                            Ok(t) => t,
                            Err(e) => {
                                TerminalReporter::print_warning(&format!(
                                    "Failed to generate templates from {}: {}. Using shared spec.",
                                    spec_path.display(),
                                    e
                                ));
                                continue;
                            }
                        };
                        let target_base_path = if let Some(cli_bp) = &self.base_command.base_path {
                            if cli_bp.is_empty() {
                                None
                            } else {
                                Some(cli_bp.clone())
                            }
                        } else {
                            target_parser.get_base_path()
                        };
                        TerminalReporter::print_success(&format!(
                            "Loaded {} operations from {}",
                            target_templates.len(),
                            spec_path.display()
                        ));
                        per_target_data
                            .insert(spec_path.clone(), (target_templates, target_base_path));
                    }
                    Err(e) => {
                        TerminalReporter::print_warning(&format!(
                            "Failed to load per-target spec {}: {}. Targets using this spec will use the shared spec.",
                            spec_path.display(),
                            e
                        ));
                    }
                }
            }
        }

        // Parse base headers
        let base_headers = self.base_command.parse_headers()?;

        // Resolve base path (CLI option takes priority over spec's servers URL)
        let base_path = self.resolve_base_path(&parser);
        if let Some(ref bp) = base_path {
            TerminalReporter::print_progress(&format!("Using base path: {}", bp));
        }

        // Parse scenario
        let scenario = LoadScenario::from_str(&self.base_command.scenario)
            .map_err(BenchError::InvalidScenario)?;

        let duration_secs_val = BenchCommand::parse_duration(&self.base_command.duration)?;

        // Round 65 (#79) — resolve concurrency after we know op count so huge
        // specs auto-cap parallel heavyweight k6 processes.
        let (max_concurrency, conc_warn) = crate::k6_gen::resolve_max_concurrency(
            self.max_concurrency_override,
            templates.len(),
            total_targets,
        );
        if let Some(msg) = conc_warn {
            TerminalReporter::print_warning(&msg);
        } else {
            TerminalReporter::print_progress(&format!(
                "Max concurrency: {} ({} target{})",
                max_concurrency,
                total_targets,
                if total_targets == 1 { "" } else { "s" },
            ));
        }

        // Round 65 (#79) — collapse per-op metrics once for all targets.
        let (per_op_metrics, per_op_warn) = crate::k6_gen::resolve_per_op_metrics(
            self.base_command.per_op_metrics,
            templates.len(),
            duration_secs_val,
        );
        if let Some(msg) = per_op_warn {
            TerminalReporter::print_warning(&msg);
        }

        // Round 63 (#79): batches of `--max-concurrency` each run the full
        // `--duration`. Wall clock is ceil(targets / concurrency) * duration,
        // not duration alone. --vus and --rps are per target, not shared.
        let (batches, one_pass_wall_secs) =
            Self::estimated_wall_clock(total_targets, max_concurrency, duration_secs_val);
        TerminalReporter::print_progress(&format!(
            "One pass: {batches} batch(es) × {duration_secs_val}s ≈ {} ({one_pass_wall_secs}s). --vus and --rps are per target, not shared.",
            Self::format_wall_clock(one_pass_wall_secs),
        ));

        // Round 66 (#79) — optional campaign loop so low concurrency can still
        // hit every target over a longevity window without manual restarts.
        let repeat_until_secs = match &self.base_command.repeat_until {
            Some(d) => Some(BenchCommand::parse_duration(d)?),
            None => None,
        };
        if let Some(n) = self.base_command.rounds {
            if n == 0 {
                return Err(BenchError::Other(
                    "--rounds must be >= 1 (or omit it for a single pass / --repeat-until)".into(),
                ));
            }
        }
        let max_rounds = self.base_command.rounds.unwrap_or(if repeat_until_secs.is_some() {
            u32::MAX
        } else {
            1
        });
        let looping = max_rounds > 1 || repeat_until_secs.is_some();
        // Round 67 (#79) — campaign disk control. Scripts are identical every
        // round, so they get written once per target under `<out>/scripts/`
        // instead of a fresh copy per `round_*/target_*`. `keep_rounds`
        // bounds how many `round_*` dirs survive (rotation); per-round stats
        // always land in `campaign.jsonl` + `round-summaries/` first.
        let keep_rounds = self.base_command.keep_rounds;
        let no_k6_logs = self.base_command.no_k6_logs;
        let scripts_base = looping.then(|| self.base_output.join("scripts"));
        if let Some(keep) = keep_rounds {
            if !looping {
                TerminalReporter::print_warning(
                    "--keep-rounds only applies to campaign runs (--repeat-until / --rounds > 1); ignoring.",
                );
            } else {
                TerminalReporter::print_progress(&format!(
                    "Round pruning: keeping newest {keep} round_* dir(s); per-round stats persist in campaign.jsonl + round-summaries/"
                ));
            }
        }
        if let Some(until) = repeat_until_secs {
            TerminalReporter::print_progress(&format!(
                "Campaign loop: will re-run all targets until wall clock reaches {} \
                 (or --rounds {}). Each k6 process only lives `--duration` ({duration_secs_val}s), \
                 so RSS resets between batches.",
                Self::format_wall_clock(until),
                self.base_command
                    .rounds
                    .map(|n| n.to_string())
                    .unwrap_or_else(|| "unlimited".into()),
            ));
        } else if max_rounds > 1 {
            TerminalReporter::print_progress(&format!(
                "Campaign loop: will re-run all targets for {max_rounds} round(s)."
            ));
        }

        let security_testing_enabled_val = self.base_command.security_testing_enabled();

        if crate::request_gen::should_force_k6_http1(verbatim, &templates, &base_headers) {
            TerminalReporter::print_progress(
                "Forcing HTTP/1.1 (GODEBUG=http2client=0): Connection headers are hop-by-hop and HTTP/2 rejects them. The header stays on the wire.",
            );
        }

        // Pre-compute enhancement code once (same for all targets)
        let has_advanced_features = self.base_command.data_file.is_some()
            || self.base_command.error_rate.is_some()
            || self.base_command.security_testing_enabled()
            || self.base_command.parallel_create.is_some();

        let enhancement_code = if has_advanced_features {
            let dummy_script = "export const options = {};";
            let enhanced = self.base_command.generate_enhanced_script(dummy_script)?;
            if let Some(pos) = enhanced.find("export const options") {
                enhanced[..pos].to_string()
            } else {
                String::new()
            }
        } else {
            String::new()
        };

        let campaign_start = std::time::Instant::now();
        let mut last_results: Option<AggregatedResults> = None;
        let mut round: u32 = 0;

        while round < max_rounds {
            // Stop before starting a new round once wall clock is exhausted.
            if let Some(until) = repeat_until_secs {
                if round > 0 && campaign_start.elapsed().as_secs() >= until {
                    TerminalReporter::print_progress(&format!(
                        "Reached --repeat-until wall clock ({}); stopping after {} round(s).",
                        Self::format_wall_clock(until),
                        round,
                    ));
                    break;
                }
            }

            round += 1;
            let round_output = if looping {
                let dir = self.base_output.join(format!("round_{round}"));
                let wall_note = repeat_until_secs
                    .map(|u| {
                        format!(
                            " (wall {} / {})",
                            Self::format_wall_clock(campaign_start.elapsed().as_secs()),
                            Self::format_wall_clock(u)
                        )
                    })
                    .unwrap_or_default();
                TerminalReporter::print_progress(&format!(
                    "Round {round}{wall_note} — results under {}",
                    dir.display()
                ));
                dir
            } else {
                self.base_output.clone()
            };

            // Create semaphore for concurrency control
            let semaphore = Arc::new(Semaphore::new(max_concurrency));
            let multi_progress = MultiProgress::new();

            // Create progress bars for each target
            let progress_bars: Vec<ProgressBar> = (0..total_targets)
                .map(|i| {
                    let pb = multi_progress.add(ProgressBar::new(1));
                    pb.set_style(
                        ProgressStyle::default_bar()
                            .template("{spinner:.green} [{elapsed_precise}] [{wide_bar:.cyan/blue}] {pos}/{len} {msg}")
                            .unwrap(),
                    );
                    pb.set_message(format!("Target {}", i + 1));
                    pb
                })
                .collect();

            // Spawn tasks for each target
            let mut handles: Vec<JoinHandle<Result<TargetResult>>> = Vec::new();

            for (index, target) in self.targets.iter().enumerate() {
                let target = target.clone();
                // Clone necessary fields from base_command instead of passing reference
                let duration = self.base_command.duration.clone();
                let vus = self.base_command.vus;
                let scenario_str = self.base_command.scenario.clone();
                let operations = self.base_command.operations.clone();
                let auth = self.base_command.auth.clone();
                let headers = self.base_command.headers.clone();
                let threshold_percentile = self.base_command.threshold_percentile.clone();
                let threshold_ms = self.base_command.threshold_ms;
                let max_error_rate = self.base_command.max_error_rate;
                // Issue #79 r62 (Srikanth on 0.3.209): the round-60 abort valve
                // stopped his WAF stress runs at ~2min because a legitimate high
                // rejection rate crossed 0.95. Thread the opt-out/threshold through
                // the per-target path so `--no-abort-on-error` reaches the k6 script.
                let abort_on_error = self.base_command.abort_on_error;
                let abort_on_error_rate = self.base_command.abort_on_error_rate;
                let verbose = self.base_command.verbose;
                let skip_tls_verify = self.base_command.skip_tls_verify;
                let chunked_request_bodies = self.base_command.chunked_request_bodies;
                let target_rps = self.base_command.target_rps;
                let no_keep_alive = self.base_command.no_keep_alive;
                // Issue #79 r54 (Srikanth on 0.3.200): `--source-ip` was silently
                // dropped in multi-target (`--targets-file`) mode because this
                // executor built `K6Executor::new()` without `.with_local_ips` and
                // the per-target BenchCommand clone zeroed `source_ips`. Thread the
                // source IPs (as the k6 `--local-ips` list) and geo config through
                // to each target's run.
                let local_ips = self.base_command.source_ips.join(",");
                let dns_policy = self.base_command.dns_policy.clone().unwrap_or_default();
                let geo_source_ips = self.base_command.geo_source_ips.clone();
                let geo_source_headers = self.base_command.geo_source_headers.clone();

                // Select per-target templates/base_path if this target has a custom spec.
                // Verbatim templates stay the traffic file's requests even when a
                // target lists its own spec.
                let (templates, base_path) = if verbatim {
                    (templates.clone(), base_path.clone())
                } else if let Some(spec_path) = &target.spec {
                    if let Some((t, bp)) = per_target_data.get(spec_path) {
                        (t.clone(), bp.clone())
                    } else {
                        (templates.clone(), base_path.clone())
                    }
                } else {
                    (templates.clone(), base_path.clone())
                };

                let base_headers = base_headers.clone();
                let scenario = scenario.clone();
                let duration_secs = duration_secs_val;
                let base_output = round_output.clone();
                let semaphore = semaphore.clone();
                let progress_bar = progress_bars[index].clone();
                let target_index = index;
                let security_testing_enabled = security_testing_enabled_val;
                let enhancement_code = enhancement_code.clone();
                // Round 67 (#79) — one stable script file per target for the
                // whole campaign instead of a per-round duplicate.
                let script_dir =
                    scripts_base.as_ref().map(|d| d.join(format!("target_{}", index + 1)));

                let handle = tokio::spawn(async move {
                    // Acquire semaphore permit
                    let _permit = semaphore.acquire().await.map_err(|e| {
                        BenchError::Other(format!("Failed to acquire semaphore: {}", e))
                    })?;

                    progress_bar.set_message(format!("Testing {}", target.url));

                    // Execute test for this target
                    let result = Self::execute_single_target_internal(
                        &duration,
                        vus,
                        &scenario_str,
                        &operations,
                        &auth,
                        &headers,
                        &threshold_percentile,
                        threshold_ms,
                        max_error_rate,
                        abort_on_error,
                        abort_on_error_rate,
                        per_op_metrics,
                        verbose,
                        skip_tls_verify,
                        base_path.as_ref(),
                        &target,
                        target_index,
                        &templates,
                        &base_headers,
                        &scenario,
                        duration_secs,
                        &base_output,
                        security_testing_enabled,
                        chunked_request_bodies,
                        target_rps,
                        no_keep_alive,
                        &enhancement_code,
                        &local_ips,
                        &dns_policy,
                        &geo_source_ips,
                        &geo_source_headers,
                        verbatim,
                        no_k6_logs,
                        script_dir.as_deref(),
                    )
                    .await;

                    progress_bar.inc(1);
                    progress_bar.finish_with_message(format!("Completed {}", target.url));

                    result
                });

                handles.push(handle);
            }

            // Wait for all tasks to complete and collect results
            let mut target_results = Vec::new();
            for (index, handle) in handles.into_iter().enumerate() {
                match handle.await {
                    Ok(Ok(result)) => {
                        target_results.push(result);
                    }
                    Ok(Err(e)) => {
                        // Create error result
                        let target_url = self.targets[index].url.clone();
                        target_results.push(TargetResult {
                            target_url: target_url.clone(),
                            target_index: index,
                            results: K6Results::default(),
                            output_dir: round_output.join(format!("target_{}", index + 1)),
                            success: false,
                            error: Some(e.to_string()),
                        });
                    }
                    Err(e) => {
                        // Join error
                        let target_url = self.targets[index].url.clone();
                        target_results.push(TargetResult {
                            target_url: target_url.clone(),
                            target_index: index,
                            results: K6Results::default(),
                            output_dir: round_output.join(format!("target_{}", index + 1)),
                            success: false,
                            error: Some(format!("Task join error: {}", e)),
                        });
                    }
                }
            }

            // Sort results by target index
            target_results.sort_by_key(|r| r.target_index);

            // Calculate aggregated metrics
            let aggregated_metrics = AggregatedMetrics::from_results(&target_results);

            let successful_targets = target_results.iter().filter(|r| r.success).count();
            let failed_targets = total_targets - successful_targets;

            last_results = Some(AggregatedResults {
                target_results,
                total_targets,
                successful_targets,
                failed_targets,
                aggregated_metrics,
            });

            // Round 67 (#79) — persist per-round stats before any pruning so
            // a killed or rotated campaign still leaves an auditable trail.
            if looping {
                let agg = last_results.as_ref().expect("just assigned");
                Self::write_round_artifacts(
                    round,
                    &round_output,
                    &self.base_output,
                    agg,
                    campaign_start.elapsed(),
                )?;
                if let Some(keep) = keep_rounds {
                    Self::prune_old_rounds(&self.base_output, keep);
                }
                TerminalReporter::print_success(&format!(
                    "Round {round} done: {}/{} targets ok, {} requests ({:.2}% err) — stats in {}",
                    agg.successful_targets,
                    agg.total_targets,
                    agg.aggregated_metrics.total_requests,
                    agg.aggregated_metrics.error_rate,
                    self.base_output.join("campaign.jsonl").display(),
                ));
            }
        }

        last_results.ok_or_else(|| {
            BenchError::Other("No rounds executed (check --rounds / --repeat-until)".to_string())
        })
    }

    /// Round 67 (#79) — per-round stats for campaign mode. Writes
    /// `round_summary.json` + `all_targets.csv` inside the round dir, a
    /// pruning-proof copy under `<out>/round-summaries/`, and appends one
    /// JSON line to `<out>/campaign.jsonl`.
    fn write_round_artifacts(
        round: u32,
        round_output: &Path,
        base_output: &Path,
        results: &AggregatedResults,
        campaign_elapsed: std::time::Duration,
    ) -> Result<()> {
        let summary = serde_json::json!({
            "round": round,
            "campaign_elapsed_seconds": campaign_elapsed.as_secs(),
            "total_targets": results.total_targets,
            "successful_targets": results.successful_targets,
            "failed_targets": results.failed_targets,
            "aggregated_metrics": aggregated_metrics_json(&results.aggregated_metrics),
            "target_results": target_results_json(results),
        });
        let pretty =
            serde_json::to_string_pretty(&summary).map_err(|e| BenchError::Other(e.to_string()))?;

        std::fs::write(round_output.join("round_summary.json"), &pretty)?;
        std::fs::write(round_output.join("all_targets.csv"), targets_csv(results))?;

        // Persistent copy — survives --keep-rounds pruning.
        let summaries_dir = base_output.join("round-summaries");
        std::fs::create_dir_all(&summaries_dir)?;
        std::fs::write(summaries_dir.join(format!("round_{round}.json")), &pretty)?;

        // Append-only campaign log: one compact record per round, greppable
        // and complete even if the campaign is killed mid-run.
        let mut line =
            serde_json::to_string(&summary).map_err(|e| BenchError::Other(e.to_string()))?;
        line.push('\n');
        use std::io::Write;
        let mut f = std::fs::OpenOptions::new()
            .create(true)
            .append(true)
            .open(base_output.join("campaign.jsonl"))?;
        f.write_all(line.as_bytes())
            .map_err(|e| BenchError::Other(format!("failed to append campaign.jsonl: {e}")))?;
        Ok(())
    }

    /// Round 67 (#79) — log rotation for campaign mode: delete all but the
    /// newest `keep` `round_*` dirs under `base_output`. `round-summaries/`
    /// and `campaign.jsonl` live outside the round dirs and are untouched.
    pub(crate) fn prune_old_rounds(base_output: &Path, keep: u32) {
        let mut round_dirs: Vec<(u32, PathBuf)> = match std::fs::read_dir(base_output) {
            Ok(rd) => rd
                .filter_map(|e| e.ok())
                .filter_map(|e| {
                    let name = e.file_name().to_string_lossy().to_string();
                    let n = name.strip_prefix("round_")?.parse::<u32>().ok()?;
                    if e.file_type().ok()?.is_dir() {
                        Some((n, e.path()))
                    } else {
                        None
                    }
                })
                .collect(),
            Err(_) => return,
        };
        round_dirs.sort_by_key(|(n, _)| *n);
        let removable = round_dirs.len().saturating_sub(keep as usize);
        let mut pruned = 0usize;
        for (_, path) in round_dirs.into_iter().take(removable) {
            if std::fs::remove_dir_all(&path).is_ok() {
                pruned += 1;
            }
        }
        if pruned > 0 {
            TerminalReporter::print_progress(&format!(
                "Pruned {pruned} old round dir(s) (keeping newest {keep})"
            ));
        }
    }

    /// Resolve the effective base path for API endpoints
    fn resolve_base_path(&self, parser: &SpecParser) -> Option<String> {
        // CLI option takes priority (including empty string to disable)
        if let Some(cli_base_path) = &self.base_command.base_path {
            if cli_base_path.is_empty() {
                return None;
            }
            return Some(cli_base_path.clone());
        }
        // Fall back to spec's base path
        parser.get_base_path()
    }

    /// Execute a single target test (internal method that doesn't require BenchCommand)
    #[allow(clippy::too_many_arguments)]
    async fn execute_single_target_internal(
        _duration: &str,
        vus: u32,
        _scenario_str: &str,
        _operations: &Option<String>,
        auth: &Option<String>,
        _headers: &[String],
        threshold_percentile: &str,
        threshold_ms: u64,
        max_error_rate: f64,
        abort_on_error: bool,
        abort_on_error_rate: f64,
        per_op_metrics: bool,
        verbose: bool,
        skip_tls_verify: bool,
        base_path: Option<&String>,
        target: &TargetConfig,
        target_index: usize,
        templates: &[crate::request_gen::RequestTemplate],
        base_headers: &HashMap<String, String>,
        scenario: &LoadScenario,
        duration_secs: u64,
        base_output: &Path,
        security_testing_enabled: bool,
        chunked_request_bodies: bool,
        target_rps: Option<u32>,
        no_keep_alive: bool,
        enhancement_code: &str,
        local_ips: &str,
        dns_policy: &str,
        geo_source_ips: &[String],
        geo_source_headers: &[String],
        wafbench_verbatim: bool,
        no_k6_logs: bool,
        script_dir: Option<&Path>,
    ) -> Result<TargetResult> {
        // Merge target-specific headers with base headers
        let mut custom_headers = base_headers.clone();
        if let Some(target_headers) = &target.headers {
            custom_headers.extend(target_headers.clone());
        }

        // Round 63 (#79): keep Connection on the wire; force HTTP/1.1.
        let force_http1 = crate::request_gen::should_force_k6_http1(
            wafbench_verbatim,
            templates,
            &custom_headers,
        );

        // Use target-specific auth if provided, otherwise use base auth
        let auth_header = target.auth.as_ref().or(auth.as_ref()).cloned();

        // Create k6 config for this target
        let k6_config = K6Config {
            target_url: target.url.clone(),
            base_path: base_path.cloned(),
            scenario: scenario.clone(),
            duration_secs,
            max_vus: vus,
            threshold_percentile: threshold_percentile.to_string(),
            threshold_ms,
            max_error_rate,
            auth_header,
            custom_headers,
            skip_tls_verify,
            security_testing_enabled,
            chunked_request_bodies,
            target_rps,
            no_keep_alive,
            geo_source_ips: geo_source_ips.to_vec(),
            geo_source_headers: geo_source_headers.to_vec(),
        };

        // Generate k6 script
        let generator = K6ScriptGenerator::new(k6_config, templates.to_vec())
            .with_abort_valve(abort_on_error, abort_on_error_rate)
            .with_force_http1(force_http1)
            .with_per_op_metrics(per_op_metrics);
        let mut script = generator.generate()?;

        // Apply pre-computed enhancement code (security definitions, etc.)
        if !enhancement_code.is_empty() {
            if let Some(pos) = script.find("export const options") {
                script.insert_str(pos, enhancement_code);
            }
        }

        // Validate script
        let validation_errors = K6ScriptGenerator::validate_script(&script);
        if !validation_errors.is_empty() {
            return Err(BenchError::Other(format!(
                "Script validation failed for target {}: {}",
                target.url,
                validation_errors.join(", ")
            )));
        }

        // Create output directory for this target
        let output_dir = base_output.join(format!("target_{}", target_index + 1));
        std::fs::create_dir_all(&output_dir)?;

        // Write script to file. Round 67 (#79): in campaign mode the script
        // is identical every round, so it lives once per target under
        // `<out>/scripts/target_N/` instead of a per-round duplicate.
        let script_path = match script_dir {
            Some(d) => {
                std::fs::create_dir_all(d)?;
                d.join("k6-script.js")
            }
            None => output_dir.join("k6-script.js"),
        };
        std::fs::write(&script_path, script)?;

        // Execute k6 with its own REST API server port per target. k6 defaults
        // to localhost:6565 and parallel instances collide on it.
        //
        // Issue #79 r58 — Srikanth on 0.3.205: every one of 10 targets failed
        // with `exit status: 106` (k6 `CannotStartRESTAPI`) and 0 requests, so
        // NO load ran at all (which read as "--discard-response-bodies isn't
        // helping"). The old scheme pinned each k6 to a FIXED port
        // (6565 + index → 6566, 6567, ...), which collides whenever those
        // ports are already taken — e.g. orphaned k6 processes left behind by
        // a previous run that was OOM-killed (his r56/r57 SIGKILL), a second
        // concurrent bench, or any local service. Bind an OS-assigned
        // ephemeral port (`:0`) instead: k6 asks the kernel for a free port at
        // start, so it can never be "already in use". We never query the REST
        // API (results come from summary.json on disk), so a random port is
        // fine. Verified: two concurrent k6 with `--address localhost:0` both
        // start cleanly where the fixed port yields exit 106.
        let api_port = 0; // ephemeral: kernel picks a free port, no collision
                          // Issue #79 r54 — forward `--source-ip` (as k6 `--local-ips`) so each
                          // target's VUs rotate through the configured source IP pool. Without
                          // this, `--source-ip` was a silent no-op in multi-target mode.
                          // Issue #79 r56 — plain multi-target load only checks status codes, so
                          // discard response bodies to keep k6's RSS bounded on long, high-VU runs.
                          // Without this, 10+ targets * 10 VUs * 70 min buffered every body and the
                          // kernel OOM-killer sent k6 `signal: 9 (SIGKILL)`.
        let executor = K6Executor::new()?
            .with_local_ips(local_ips.to_string())
            .with_dns_policy(dns_policy.to_string())
            .with_discard_response_bodies(true)
            .with_force_http1(force_http1)
            .with_capture_logs(!no_k6_logs);
        let results = executor
            .execute_with_port(&script_path, Some(&output_dir), verbose, Some(api_port))
            .await;

        match results {
            Ok(k6_results) => Ok(TargetResult {
                target_url: target.url.clone(),
                target_index,
                results: k6_results,
                output_dir,
                success: true,
                error: None,
            }),
            Err(e) => Ok(TargetResult {
                target_url: target.url.clone(),
                target_index,
                results: K6Results::default(),
                output_dir,
                success: false,
                error: Some(e.to_string()),
            }),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_aggregated_metrics_from_results() {
        let results = vec![
            TargetResult {
                target_url: "http://api1.com".to_string(),
                target_index: 0,
                results: K6Results {
                    total_requests: 100,
                    failed_requests: 5,
                    avg_duration_ms: 100.0,
                    p95_duration_ms: 200.0,
                    p99_duration_ms: 300.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output1"),
                success: true,
                error: None,
            },
            TargetResult {
                target_url: "http://api2.com".to_string(),
                target_index: 1,
                results: K6Results {
                    total_requests: 200,
                    failed_requests: 10,
                    avg_duration_ms: 150.0,
                    p95_duration_ms: 250.0,
                    p99_duration_ms: 350.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output2"),
                success: true,
                error: None,
            },
        ];

        let metrics = AggregatedMetrics::from_results(&results);
        assert_eq!(metrics.total_requests, 300);
        assert_eq!(metrics.total_failed_requests, 15);
        assert_eq!(metrics.avg_duration_ms, 125.0); // (100 + 150) / 2
    }

    #[test]
    fn test_aggregated_metrics_with_failed_targets() {
        let results = vec![
            TargetResult {
                target_url: "http://api1.com".to_string(),
                target_index: 0,
                results: K6Results {
                    total_requests: 100,
                    failed_requests: 5,
                    avg_duration_ms: 100.0,
                    p95_duration_ms: 200.0,
                    p99_duration_ms: 300.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output1"),
                success: true,
                error: None,
            },
            TargetResult {
                target_url: "http://api2.com".to_string(),
                target_index: 1,
                results: K6Results::default(),
                output_dir: PathBuf::from("output2"),
                success: false,
                error: Some("Network error".to_string()),
            },
        ];

        let metrics = AggregatedMetrics::from_results(&results);
        // Only successful target should be counted
        assert_eq!(metrics.total_requests, 100);
        assert_eq!(metrics.total_failed_requests, 5);
        assert_eq!(metrics.avg_duration_ms, 100.0);
    }

    #[test]
    fn test_aggregated_metrics_empty_results() {
        let results = vec![];
        let metrics = AggregatedMetrics::from_results(&results);
        assert_eq!(metrics.total_requests, 0);
        assert_eq!(metrics.total_failed_requests, 0);
        assert_eq!(metrics.avg_duration_ms, 0.0);
        assert_eq!(metrics.error_rate, 0.0);
    }

    #[test]
    fn test_aggregated_metrics_error_rate_calculation() {
        let results = vec![TargetResult {
            target_url: "http://api1.com".to_string(),
            target_index: 0,
            results: K6Results {
                total_requests: 1000,
                failed_requests: 50,
                avg_duration_ms: 100.0,
                p95_duration_ms: 200.0,
                p99_duration_ms: 300.0,
                ..Default::default()
            },
            output_dir: PathBuf::from("output1"),
            success: true,
            error: None,
        }];

        let metrics = AggregatedMetrics::from_results(&results);
        assert_eq!(metrics.error_rate, 5.0); // 50/1000 * 100
    }

    #[test]
    fn test_aggregated_metrics_p95_p99_calculation() {
        let results = vec![
            TargetResult {
                target_url: "http://api1.com".to_string(),
                target_index: 0,
                results: K6Results {
                    total_requests: 100,
                    failed_requests: 0,
                    avg_duration_ms: 100.0,
                    p95_duration_ms: 150.0,
                    p99_duration_ms: 200.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output1"),
                success: true,
                error: None,
            },
            TargetResult {
                target_url: "http://api2.com".to_string(),
                target_index: 1,
                results: K6Results {
                    total_requests: 100,
                    failed_requests: 0,
                    avg_duration_ms: 200.0,
                    p95_duration_ms: 250.0,
                    p99_duration_ms: 300.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output2"),
                success: true,
                error: None,
            },
            TargetResult {
                target_url: "http://api3.com".to_string(),
                target_index: 2,
                results: K6Results {
                    total_requests: 100,
                    failed_requests: 0,
                    avg_duration_ms: 300.0,
                    p95_duration_ms: 350.0,
                    p99_duration_ms: 400.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output3"),
                success: true,
                error: None,
            },
        ];

        let metrics = AggregatedMetrics::from_results(&results);
        // p95 should be the 95th percentile of [150, 250, 350] = index 2 = 350
        // p99 should be the 99th percentile of [200, 300, 400] = index 2 = 400
        assert_eq!(metrics.p95_duration_ms, 350.0);
        assert_eq!(metrics.p99_duration_ms, 400.0);
    }

    #[test]
    fn estimated_wall_clock_is_batches_times_duration() {
        // Srikanth: 64 IPs, --max-concurrency 10, -d 300 → 7 batches × 300s.
        let (batches, wall) = ParallelExecutor::estimated_wall_clock(64, 10, 300);
        assert_eq!(batches, 7);
        assert_eq!(wall, 2100);
        assert_eq!(ParallelExecutor::format_wall_clock(2100), "35m00s");
        assert_eq!(ParallelExecutor::estimated_wall_clock(10, 10, 300), (1, 300));
        assert_eq!(ParallelExecutor::estimated_wall_clock(0, 10, 300), (0, 0));
        assert_eq!(ParallelExecutor::format_wall_clock(45), "45s");
        assert_eq!(ParallelExecutor::format_wall_clock(3661), "1h01m01s");
    }

    #[test]
    fn parallel_k6_spawn_sets_force_http1() {
        let src = include_str!("parallel_executor.rs");
        assert!(
            src.contains("with_force_http1(force_http1)"),
            "multi-target k6 spawn must pass GODEBUG=http2client=0 when Connection headers are present"
        );
        assert!(
            src.contains("One pass:") || src.contains("Estimated wall clock"),
            "multi-target start must print ceil(targets/concurrency)*duration"
        );
        assert!(
            src.contains("repeat_until") && src.contains("Campaign loop"),
            "multi-target must support --repeat-until campaign looping (#79)"
        );
        assert!(
            src.contains("with_per_op_metrics(per_op_metrics)"),
            "multi-target k6 spawn must apply Round-65 per-op metrics collapse (#79)"
        );
        assert!(
            src.contains("resolve_max_concurrency"),
            "multi-target must auto-cap concurrency for huge specs (#79)"
        );
        assert!(
            src.contains("with_capture_logs(!no_k6_logs)"),
            "multi-target k6 spawn must honour --no-k6-logs (#79)"
        );
    }

    fn sample_aggregated_results() -> AggregatedResults {
        let target_results = vec![
            TargetResult {
                target_url: "http://api1.com".to_string(),
                target_index: 0,
                results: K6Results {
                    total_requests: 100,
                    failed_requests: 5,
                    avg_duration_ms: 100.0,
                    p95_duration_ms: 200.0,
                    p99_duration_ms: 300.0,
                    ..Default::default()
                },
                output_dir: PathBuf::from("output1"),
                success: true,
                error: None,
            },
            TargetResult {
                target_url: "http://api2.com".to_string(),
                target_index: 1,
                results: K6Results::default(),
                output_dir: PathBuf::from("output2"),
                success: false,
                error: Some("boom".to_string()),
            },
        ];
        AggregatedResults {
            aggregated_metrics: AggregatedMetrics::from_results(&target_results),
            total_targets: 2,
            successful_targets: 1,
            failed_targets: 1,
            target_results,
        }
    }

    /// Round 67 (#79) — each campaign round must drop a summary + CSV into
    /// the round dir, a pruning-proof copy under round-summaries/, and one
    /// JSONL line per round in campaign.jsonl.
    #[test]
    fn write_round_artifacts_writes_all_surfaces() {
        let base = tempfile::tempdir().unwrap();
        let round_dir = base.path().join("round_3");
        std::fs::create_dir_all(&round_dir).unwrap();
        let results = sample_aggregated_results();

        ParallelExecutor::write_round_artifacts(
            3,
            &round_dir,
            base.path(),
            &results,
            std::time::Duration::from_secs(42),
        )
        .unwrap();
        ParallelExecutor::write_round_artifacts(
            4,
            &round_dir,
            base.path(),
            &results,
            std::time::Duration::from_secs(84),
        )
        .unwrap();

        let summary_raw = std::fs::read_to_string(round_dir.join("round_summary.json")).unwrap();
        let summary: serde_json::Value = serde_json::from_str(&summary_raw).unwrap();
        assert_eq!(summary["round"], 4);
        assert_eq!(summary["total_targets"], 2);
        assert_eq!(summary["successful_targets"], 1);
        assert_eq!(summary["aggregated_metrics"]["total_requests"], 100);

        assert!(base.path().join("round-summaries/round_3.json").exists());
        assert!(base.path().join("round-summaries/round_4.json").exists());

        let csv = std::fs::read_to_string(round_dir.join("all_targets.csv")).unwrap();
        assert!(csv.contains("http://api1.com"));
        assert!(csv.contains("boom"));

        // Two artifact writes → two JSONL lines.
        let jsonl = std::fs::read_to_string(base.path().join("campaign.jsonl")).unwrap();
        let lines: Vec<&str> = jsonl.lines().collect();
        assert_eq!(lines.len(), 2);
        let first: serde_json::Value = serde_json::from_str(lines[0]).unwrap();
        assert_eq!(first["round"], 3);
        assert_eq!(first["campaign_elapsed_seconds"], 42);
        let second: serde_json::Value = serde_json::from_str(lines[1]).unwrap();
        assert_eq!(second["round"], 4);
    }

    /// Round 67 (#79) — --keep-rounds prunes only `round_*` dirs and always
    /// leaves the newest N plus the campaign-level stats.
    #[test]
    fn prune_old_rounds_keeps_newest_and_stats() {
        let base = tempfile::tempdir().unwrap();
        for n in 1..=5 {
            std::fs::create_dir_all(base.path().join(format!("round_{n}"))).unwrap();
        }
        // Things that must survive pruning.
        std::fs::create_dir_all(base.path().join("round-summaries")).unwrap();
        std::fs::write(base.path().join("campaign.jsonl"), "{}\n").unwrap();
        std::fs::create_dir_all(base.path().join("scripts")).unwrap();
        std::fs::create_dir_all(base.path().join("target_1")).unwrap();
        // A round dir that is not a dir must not be deleted.
        std::fs::write(base.path().join("round_9.txt"), "x").unwrap();

        ParallelExecutor::prune_old_rounds(base.path(), 2);

        for n in 1..=3 {
            assert!(!base.path().join(format!("round_{n}")).exists(), "round_{n} pruned");
        }
        assert!(base.path().join("round_4").exists());
        assert!(base.path().join("round_5").exists());
        assert!(base.path().join("round-summaries").exists());
        assert!(base.path().join("campaign.jsonl").exists());
        assert!(base.path().join("scripts").exists());
        assert!(base.path().join("target_1").exists());
        assert!(base.path().join("round_9.txt").exists());
    }

    /// Round 67 (#79) — keep=0 prunes every round dir (campaign stats only).
    #[test]
    fn prune_old_rounds_zero_keeps_stats_only() {
        let base = tempfile::tempdir().unwrap();
        std::fs::create_dir_all(base.path().join("round_1")).unwrap();
        std::fs::write(base.path().join("campaign.jsonl"), "{}\n").unwrap();

        ParallelExecutor::prune_old_rounds(base.path(), 0);

        assert!(!base.path().join("round_1").exists());
        assert!(base.path().join("campaign.jsonl").exists());
    }
}