use crate::config::ThresholdsConfig;
use crate::detect::{Finding, FindingType, Severity};
use crate::report::{GreenSummary, QualityGate, QualityRule};
#[must_use]
pub fn rule_label(rule: &str) -> Option<&'static str> {
match rule {
"n_plus_one_sql_critical_max" => Some("Max critical N+1 SQL"),
"n_plus_one_http_warning_max" => Some("Max N+1 HTTP (warning+)"),
"n_plus_one_messaging_warning_max" => Some("Max N+1 messaging (warning+)"),
"io_waste_ratio_max" => Some("Max I/O waste ratio"),
_ => None,
}
}
#[must_use]
pub fn evaluate(
findings: &[Finding],
green_summary: &GreenSummary,
thresholds: &ThresholdsConfig,
) -> QualityGate {
let mut rules = Vec::with_capacity(4);
let critical_sql_count = findings
.iter()
.filter(|f| f.finding_type == FindingType::NPlusOneSql && f.severity == Severity::Critical)
.count();
let threshold_sql = thresholds.n_plus_one_sql_critical_max;
rules.push(QualityRule {
rule: "n_plus_one_sql_critical_max".to_string(),
threshold: f64::from(threshold_sql),
actual: critical_sql_count as f64,
passed: critical_sql_count <= threshold_sql as usize,
});
let warning_plus_http_count = findings
.iter()
.filter(|f| {
f.finding_type == FindingType::NPlusOneHttp
&& matches!(f.severity, Severity::Warning | Severity::Critical)
})
.count();
let threshold_http = thresholds.n_plus_one_http_warning_max;
rules.push(QualityRule {
rule: "n_plus_one_http_warning_max".to_string(),
threshold: f64::from(threshold_http),
actual: warning_plus_http_count as f64,
passed: warning_plus_http_count <= threshold_http as usize,
});
let warning_plus_messaging_count = findings
.iter()
.filter(|f| {
f.finding_type == FindingType::NPlusOneMessaging
&& matches!(f.severity, Severity::Warning | Severity::Critical)
})
.count();
let threshold_messaging = thresholds.n_plus_one_messaging_warning_max;
rules.push(QualityRule {
rule: "n_plus_one_messaging_warning_max".to_string(),
threshold: f64::from(threshold_messaging),
actual: warning_plus_messaging_count as f64,
passed: warning_plus_messaging_count <= threshold_messaging as usize,
});
rules.push(QualityRule {
rule: "io_waste_ratio_max".to_string(),
threshold: thresholds.io_waste_ratio_max,
actual: green_summary.io_waste_ratio,
passed: green_summary.io_waste_ratio <= thresholds.io_waste_ratio_max,
});
let passed = rules.iter().all(|r| r.passed);
QualityGate { passed, rules }
}
#[cfg(test)]
mod tests {
use super::*;
use crate::config::Config;
use crate::test_helpers::{make_finding, make_test_green_summary};
fn empty_green_summary() -> GreenSummary {
GreenSummary::disabled(0)
}
#[test]
fn all_rules_pass_with_no_findings() {
let config = Config::default();
let summary = empty_green_summary();
let gate = evaluate(&[], &summary, &config.thresholds);
assert!(gate.passed);
assert_eq!(gate.rules.len(), 4);
assert!(gate.rules.iter().all(|r| r.passed));
}
#[test]
fn every_evaluated_rule_has_a_label() {
let config = Config::default();
let gate = evaluate(&[], &empty_green_summary(), &config.thresholds);
for r in &gate.rules {
assert!(
rule_label(&r.rule).is_some(),
"rule {} has no display label",
r.rule
);
}
assert!(rule_label("some_future_rule").is_none());
}
#[test]
fn messaging_n_plus_one_fails_its_own_rule() {
let config = Config::default(); let summary = empty_green_summary();
let findings: Vec<_> = (0..4)
.map(|_| make_finding(FindingType::NPlusOneMessaging, Severity::Warning))
.collect();
let gate = evaluate(&findings, &summary, &config.thresholds);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "n_plus_one_messaging_warning_max")
.expect("the messaging rule is evaluated");
assert!(!rule.passed, "4 warnings exceed the default of 3");
assert!((rule.actual - 4.0).abs() < f64::EPSILON);
assert!(!gate.passed);
}
#[test]
fn critical_sql_fails_gate() {
let config = Config::default(); let findings = vec![make_finding(FindingType::NPlusOneSql, Severity::Critical)];
let summary = empty_green_summary();
let gate = evaluate(&findings, &summary, &config.thresholds);
assert!(!gate.passed);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "n_plus_one_sql_critical_max")
.unwrap();
assert!(!rule.passed);
assert!((rule.actual - 1.0).abs() < f64::EPSILON);
assert!((rule.threshold - 0.0).abs() < f64::EPSILON);
}
#[test]
fn warning_sql_does_not_fail_sql_critical_rule() {
let config = Config::default();
let findings = vec![make_finding(FindingType::NPlusOneSql, Severity::Warning)];
let summary = empty_green_summary();
let gate = evaluate(&findings, &summary, &config.thresholds);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "n_plus_one_sql_critical_max")
.unwrap();
assert!(
rule.passed,
"warning SQL should not trigger critical-only rule"
);
}
#[test]
fn warning_http_under_threshold() {
let config = Config {
thresholds: ThresholdsConfig {
n_plus_one_http_warning_max: 3,
..ThresholdsConfig::default()
},
..Config::default()
};
let findings = vec![
make_finding(FindingType::NPlusOneHttp, Severity::Warning),
make_finding(FindingType::NPlusOneHttp, Severity::Warning),
];
let summary = empty_green_summary();
let gate = evaluate(&findings, &summary, &config.thresholds);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "n_plus_one_http_warning_max")
.unwrap();
assert!(rule.passed);
}
#[test]
fn warning_http_over_threshold() {
let config = Config {
thresholds: ThresholdsConfig {
n_plus_one_http_warning_max: 3,
..ThresholdsConfig::default()
},
..Config::default()
};
let findings = vec![
make_finding(FindingType::NPlusOneHttp, Severity::Warning),
make_finding(FindingType::NPlusOneHttp, Severity::Warning),
make_finding(FindingType::NPlusOneHttp, Severity::Warning),
make_finding(FindingType::NPlusOneHttp, Severity::Warning),
];
let summary = empty_green_summary();
let gate = evaluate(&findings, &summary, &config.thresholds);
assert!(!gate.passed);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "n_plus_one_http_warning_max")
.unwrap();
assert!(!rule.passed);
assert!((rule.actual - 4.0).abs() < f64::EPSILON);
}
#[test]
fn io_waste_ratio_fails_gate() {
let config = Config::default(); let summary = make_test_green_summary(10, 5, 0.5);
let gate = evaluate(&[], &summary, &config.thresholds);
assert!(!gate.passed);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "io_waste_ratio_max")
.unwrap();
assert!(!rule.passed);
assert!((rule.actual - 0.5).abs() < f64::EPSILON);
}
#[test]
fn custom_thresholds() {
let config = Config {
thresholds: ThresholdsConfig {
n_plus_one_sql_critical_max: 5,
io_waste_ratio_max: 0.90,
..ThresholdsConfig::default()
},
..Config::default()
};
let findings = vec![
make_finding(FindingType::NPlusOneSql, Severity::Critical),
make_finding(FindingType::NPlusOneSql, Severity::Critical),
];
let summary = make_test_green_summary(10, 8, 0.8);
let gate = evaluate(&findings, &summary, &config.thresholds);
assert!(gate.passed, "2 critical SQL <= 5, 0.8 <= 0.90");
}
#[test]
fn critical_http_counts_as_warning_plus() {
let config = Config {
thresholds: ThresholdsConfig {
n_plus_one_http_warning_max: 0,
..ThresholdsConfig::default()
},
..Config::default()
};
let findings = vec![make_finding(FindingType::NPlusOneHttp, Severity::Critical)];
let summary = empty_green_summary();
let gate = evaluate(&findings, &summary, &config.thresholds);
let rule = gate
.rules
.iter()
.find(|r| r.rule == "n_plus_one_http_warning_max")
.unwrap();
assert!(
!rule.passed,
"critical HTTP should count toward warning+ threshold"
);
}
}