#![allow(
clippy::missing_docs_in_private_items,
reason = "test helpers and fixtures do not need doc comments"
)]
use super::*;
use crate::routines::new_store;
use std::collections::HashMap;
use std::sync::{Arc, Mutex};
struct TempHome(std::path::PathBuf);
impl TempHome {
fn set() -> Self {
let dir = std::env::temp_dir().join(format!("moadim-svctest-{}", uuid::Uuid::new_v4()));
std::fs::create_dir_all(&dir).expect("create temp home");
unsafe {
std::env::set_var("MOADIM_HOME_OVERRIDE", &dir);
}
Self(dir)
}
}
impl Drop for TempHome {
fn drop(&mut self) {
unsafe {
std::env::remove_var("MOADIM_HOME_OVERRIDE");
}
let _ = std::fs::remove_dir_all(&self.0);
}
}
fn make_routine(id: &str, title: &str, created_at: u64, updated_at: u64) -> Routine {
Routine {
model: None,
id: id.to_string(),
schedule: "@daily".to_string(),
title: title.to_string(),
agent: "claude".to_string(),
prompt: "do the thing".to_string(),
goal: None,
repositories: vec![],
machines: vec![crate::machine::current_machine()],
enabled: true,
source: "managed".to_string(),
created_at,
updated_at,
last_manual_trigger_at: None,
last_scheduled_trigger_at: None,
snoozed_until: None,
skip_runs: None,
power_saving: false,
tags: vec![],
ttl_secs: None,
max_runtime_secs: None,
}
}
fn store_with(routines: Vec<Routine>) -> RoutineStore {
let mut map = HashMap::new();
for routine in routines {
map.insert(routine.id.clone(), routine);
}
Arc::new(Mutex::new(map))
}
fn valid_create_request() -> CreateRoutineRequest {
CreateRoutineRequest {
model: None,
goal: None,
schedule: "@daily".into(),
title: "Valid Title".into(),
agent: "claude".into(),
prompt: "do the thing".into(),
repositories: vec![],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
}
}
fn empty_update_request() -> UpdateRoutineRequest {
UpdateRoutineRequest {
model: None,
goal: None,
schedule: None,
title: None,
agent: None,
prompt: None,
repositories: None,
machines: None,
enabled: None,
ttl_secs: None,
max_runtime_secs: None,
tags: None,
}
}
#[test]
fn svc_create_rejects_ttl_above_cron_ceiling() {
let _home = TempHome::set();
let store = new_store();
let result = svc_create(
&store,
CreateRoutineRequest {
model: None,
goal: None,
schedule: "*/5 * * * *".into(),
ttl_secs: Some(1800),
..valid_create_request()
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
}
#[test]
fn svc_create_rejects_max_runtime_above_cron_ceiling() {
let _home = TempHome::set();
let store = new_store();
let result = svc_create(
&store,
CreateRoutineRequest {
model: None,
goal: None,
schedule: "*/5 * * * *".into(),
max_runtime_secs: Some(1800),
..valid_create_request()
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
}
#[test]
fn svc_create_accepts_secs_at_cron_ceiling() {
let _home = TempHome::set();
let store = store_with(vec![make_routine(
"at-ceiling-dupe",
"At Ceiling ZZZ",
1,
1,
)]);
let result = svc_create(
&store,
CreateRoutineRequest {
model: None,
goal: None,
schedule: "*/5 * * * *".into(),
title: " at ceiling ZZZ ".into(),
ttl_secs: Some(300),
max_runtime_secs: Some(300),
..valid_create_request()
},
);
assert!(matches!(result, Err(AppError::Conflict(_))));
}
#[test]
fn svc_update_rejects_ttl_above_current_schedule_ceiling() {
let _home = TempHome::set();
let store = store_with(vec![Routine {
schedule: "*/5 * * * *".to_string(),
..make_routine("upd-ttl-ceiling", "Keep Ceiling", 1, 1)
}]);
let result = svc_update(
&store,
"upd-ttl-ceiling",
UpdateRoutineRequest {
model: None,
goal: None,
ttl_secs: Some(1800),
..empty_update_request()
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
assert_eq!(
store
.lock()
.unwrap()
.get("upd-ttl-ceiling")
.unwrap()
.ttl_secs,
None
);
}
#[test]
fn svc_update_rejects_secs_above_new_schedule_ceiling() {
let _home = TempHome::set();
let store = store_with(vec![make_routine("upd-new-sched", "Keep New Sched", 1, 1)]);
let result = svc_update(
&store,
"upd-new-sched",
UpdateRoutineRequest {
model: None,
goal: None,
schedule: Some("*/5 * * * *".into()),
max_runtime_secs: Some(1800),
..empty_update_request()
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
}