132 lines
3.9 KiB
Rust
132 lines
3.9 KiB
Rust
use scriptherder::check::Check;
|
|
use scriptherder::checkstatus::CheckStatus;
|
|
use scriptherder::job::Job;
|
|
use scriptherder::jobs_list::JobsList;
|
|
use std::collections::HashMap;
|
|
|
|
fn move_back(job: &mut Job, seconds: f64) {
|
|
let s = job.start_time().unwrap();
|
|
let e = job.end_time().unwrap();
|
|
job.set_times_for_test(s - seconds, e - seconds);
|
|
}
|
|
|
|
fn make_checkstatus(jobs: Vec<Job>, ok: &str, warn: &str) -> CheckStatus {
|
|
let mut checks = HashMap::new();
|
|
checks.insert(
|
|
"test1".to_string(),
|
|
Check::new(ok, warn, "unit_testing", false).unwrap(),
|
|
);
|
|
let mut cs = CheckStatus::new(false, String::new(), checks);
|
|
cs.check_jobs(JobsList::from_jobs(jobs, false));
|
|
cs
|
|
}
|
|
|
|
#[test]
|
|
fn two_delayed_jobs_critical() {
|
|
let ok = "exit_status=0, max_age=1m";
|
|
let warn = "exit_status=0, max_age=5m";
|
|
let mut j1 = Job::new("test1", vec!["/usr/bin/true".into()]).unwrap();
|
|
let mut j2 = Job::new("test1", vec!["/usr/bin/true".into()]).unwrap();
|
|
j1.run();
|
|
j2.run();
|
|
let rc = Check::new(ok, warn, "unit_testing", true).unwrap();
|
|
j1.check(&rc);
|
|
j2.check(&rc);
|
|
move_back(&mut j1, 19.0 * 60.0);
|
|
move_back(&mut j2, 20.0 * 60.0);
|
|
let cs = make_checkstatus(vec![j1, j2], ok, warn);
|
|
assert_eq!(cs.num_jobs(), 1);
|
|
assert_eq!(
|
|
cs.aggregate_status(),
|
|
(
|
|
"CRITICAL".to_string(),
|
|
Some("age=19m>1m, age=19m>5m".to_string())
|
|
)
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn two_delayed_jobs_warning() {
|
|
let ok = "exit_status=0, max_age=1m";
|
|
let warn = "exit_status=0, max_age=2h";
|
|
let mut j1 = Job::new("test1", vec!["/usr/bin/true".into()]).unwrap();
|
|
let mut j2 = Job::new("test1", vec!["/usr/bin/true".into()]).unwrap();
|
|
j1.run();
|
|
j2.run();
|
|
let rc = Check::new(ok, warn, "unit_testing", true).unwrap();
|
|
j1.check(&rc);
|
|
j2.check(&rc);
|
|
move_back(&mut j1, 2.0 * 60.0);
|
|
move_back(&mut j2, 4.0 * 60.0);
|
|
let cs = make_checkstatus(vec![j1, j2], ok, warn);
|
|
assert_eq!(cs.num_jobs(), 1);
|
|
assert_eq!(
|
|
cs.aggregate_status(),
|
|
(
|
|
"WARNING".to_string(),
|
|
Some("age=2m>1m, max_age=2h, stored_status=OK==True".to_string())
|
|
)
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn positive_then_negative_critical() {
|
|
let ok = "exit_status=0";
|
|
let warn = "exit_status=0";
|
|
let mut j1 = Job::new("test1", vec!["/usr/bin/true".into()]).unwrap();
|
|
let mut j2 = Job::new("test1", vec!["/usr/bin/false".into()]).unwrap();
|
|
j1.run();
|
|
j2.run();
|
|
let rc = Check::new(ok, warn, "unit_testing", true).unwrap();
|
|
j1.check(&rc);
|
|
j2.check(&rc);
|
|
let cs = make_checkstatus(vec![j1, j2], ok, warn);
|
|
assert_eq!(cs.num_jobs(), 1);
|
|
assert_eq!(
|
|
cs.aggregate_status(),
|
|
(
|
|
"CRITICAL".to_string(),
|
|
Some("stored_status=OK==False".to_string())
|
|
)
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn job_failed_job() {
|
|
let ok = "exit_status=0,max_age=50m";
|
|
let warn = "exit_status=0,max_age=1h";
|
|
let mut j1 = Job::new("test1", vec!["/usr/bin/false".into()]).unwrap();
|
|
j1.run();
|
|
move_back(&mut j1, 10.0);
|
|
let rc = Check::new(ok, warn, "unit_testing", true).unwrap();
|
|
j1.check(&rc);
|
|
let cs = make_checkstatus(vec![j1], ok, warn);
|
|
assert_eq!(cs.num_jobs(), 1);
|
|
assert_eq!(
|
|
cs.aggregate_status(),
|
|
(
|
|
"CRITICAL".to_string(),
|
|
Some("stored_status=OK==False".to_string())
|
|
)
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn misconfigured_criteria() {
|
|
let ok = "exit_status=0,max_age=50m";
|
|
let warn = "exit_status=0,max_age=1";
|
|
let mut j1 = Job::new("test1", vec!["/usr/bin/false".into()]).unwrap();
|
|
j1.run();
|
|
move_back(&mut j1, 10.0);
|
|
let rc = Check::new(ok, warn, "unit_testing", true).unwrap();
|
|
j1.check(&rc);
|
|
let cs = make_checkstatus(vec![j1], ok, warn);
|
|
assert_eq!(cs.num_jobs(), 1);
|
|
assert_eq!(
|
|
cs.aggregate_status(),
|
|
(
|
|
"CRITICAL".to_string(),
|
|
Some("stored_status=OK==False, age=10s>1s".to_string())
|
|
)
|
|
);
|
|
}
|