#![allow(
clippy::missing_docs_in_private_items,
reason = "test helpers and fixtures do not need doc comments"
)]
use super::*;
use crate::routines::new_store;
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 create_req_with_title(title: &str) -> CreateRoutineRequest {
CreateRoutineRequest {
model: None,
schedule: "@daily".into(),
title: title.into(),
agent: "claude".into(),
prompt: "p".into(),
goal: None,
repositories: vec![],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
}
}
#[test]
fn svc_create_rejects_blank_and_punctuation_titles() {
let _home = TempHome::set();
for title in ["", " \n\t", "!!!"] {
let store = new_store();
let result = svc_create(&store, create_req_with_title(title));
assert!(
matches!(result, Err(AppError::BadRequest(_))),
"title {title:?} should be rejected"
);
assert!(store.lock().unwrap().is_empty());
}
}
#[test]
fn svc_create_rejects_overlong_title() {
let _home = TempHome::set();
let store = new_store();
let title = "a".repeat(MAX_TITLE_LEN + 1);
let result = svc_create(&store, create_req_with_title(&title));
assert!(matches!(result, Err(AppError::BadRequest(_))));
assert!(store.lock().unwrap().is_empty());
}
#[test]
fn svc_create_rejects_unknown_agent() {
let _home = TempHome::set();
let store = new_store();
let result = svc_create(
&store,
CreateRoutineRequest {
model: None,
schedule: "@daily".into(),
title: "Svc Create Unknown Agent ZZZ".into(),
agent: "no-such-agent-zzz".into(),
prompt: "p".into(),
goal: None,
repositories: vec![],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
assert!(store.lock().unwrap().is_empty());
}
#[test]
fn svc_update_rejects_blank_and_punctuation_titles() {
let _home = TempHome::set();
let original = "Svc Update Title Guard ZZZ";
for title in ["", " ", "!!!"] {
let store = new_store();
let routine = make_routine("title-guard-id", original, 1, 1);
crate::routine_storage::write_routine(&routine).unwrap();
store
.lock()
.unwrap()
.insert("title-guard-id".into(), routine);
let result = svc_update(
&store,
"title-guard-id",
UpdateRoutineRequest {
model: None,
schedule: None,
title: Some(title.into()),
agent: None,
prompt: None,
goal: None,
repositories: None,
machines: None,
enabled: None,
ttl_secs: None,
max_runtime_secs: None,
tags: None,
},
);
assert!(
matches!(result, Err(AppError::BadRequest(_))),
"update to title {title:?} should be rejected"
);
assert_eq!(
store.lock().unwrap().get("title-guard-id").unwrap().title,
original
);
}
}
#[test]
fn svc_create_accepts_builtin_agent() {
let _home = TempHome::set();
crate::routines::ensure_default_agents();
let title = "Svc Create Valid Agent ZZZ";
let store = new_store();
let created = svc_create(
&store,
CreateRoutineRequest {
model: None,
schedule: "@daily".into(),
title: title.into(),
agent: "claude".into(),
prompt: "p".into(),
goal: None,
repositories: vec![],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
},
)
.unwrap();
assert_eq!(created.routine.agent, "claude");
svc_delete(&store, &created.routine.id).unwrap();
}
#[test]
fn svc_update_rejects_unknown_agent() {
let _home = TempHome::set();
let title = "Svc Update Unknown Agent ZZZ";
let store = new_store();
let routine = make_routine("upd-agent-id", title, 1, 1);
crate::routine_storage::write_routine(&routine).unwrap();
store.lock().unwrap().insert("upd-agent-id".into(), routine);
let result = svc_update(
&store,
"upd-agent-id",
UpdateRoutineRequest {
model: None,
schedule: None,
title: None,
agent: Some("no-such-agent-zzz".into()),
prompt: None,
goal: None,
repositories: None,
machines: None,
enabled: None,
ttl_secs: None,
max_runtime_secs: None,
tags: None,
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
assert_eq!(
store.lock().unwrap().get("upd-agent-id").unwrap().agent,
"claude"
);
}
#[test]
fn svc_create_rejects_blank_repository_url() {
let _home = TempHome::set();
let store = new_store();
for url in ["", " "] {
let result = svc_create(
&store,
CreateRoutineRequest {
model: None,
schedule: "@daily".into(),
title: "Svc Create Blank Repo ZZZ".into(),
agent: "claude".into(),
prompt: "p".into(),
goal: None,
repositories: vec![Repository {
repository: url.into(),
branch: None,
}],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
}
assert!(store.lock().unwrap().is_empty());
}
#[test]
fn svc_create_rejects_blank_repository_branch() {
let _home = TempHome::set();
let store = new_store();
let result = svc_create(
&store,
CreateRoutineRequest {
model: None,
schedule: "@daily".into(),
title: "Svc Create Blank Branch ZZZ".into(),
agent: "claude".into(),
prompt: "p".into(),
goal: None,
repositories: vec![Repository {
repository: "https://github.com/octocat/Hello-World".into(),
branch: Some(" ".into()),
}],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
assert!(store.lock().unwrap().is_empty());
}
#[test]
fn svc_create_trims_repository_entries() {
let _home = TempHome::set();
crate::routines::ensure_default_agents();
let title = "Svc Create Trim Repo ZZZ";
let store = new_store();
let created = svc_create(
&store,
CreateRoutineRequest {
model: None,
schedule: "@daily".into(),
title: title.into(),
agent: "claude".into(),
prompt: "p".into(),
goal: None,
repositories: vec![Repository {
repository: " https://github.com/octocat/Hello-World ".into(),
branch: Some(" main ".into()),
}],
machines: vec![crate::machine::current_machine()],
enabled: true,
ttl_secs: None,
max_runtime_secs: None,
tags: vec![],
},
)
.unwrap();
let repo = &created.routine.repositories[0];
assert_eq!(repo.repository, "https://github.com/octocat/Hello-World");
assert_eq!(repo.branch.as_deref(), Some("main"));
svc_delete(&store, &created.routine.id).unwrap();
}
#[test]
fn svc_update_rejects_blank_repository_url() {
let _home = TempHome::set();
let title = "Svc Update Blank Repo ZZZ";
let store = new_store();
let routine = make_routine("upd-repo-id", title, 1, 1);
crate::routine_storage::write_routine(&routine).unwrap();
store.lock().unwrap().insert("upd-repo-id".into(), routine);
let result = svc_update(
&store,
"upd-repo-id",
UpdateRoutineRequest {
model: None,
schedule: None,
title: None,
agent: None,
prompt: None,
goal: None,
repositories: Some(vec![Repository {
repository: " ".into(),
branch: None,
}]),
machines: None,
enabled: None,
ttl_secs: None,
max_runtime_secs: None,
tags: None,
},
);
assert!(matches!(result, Err(AppError::BadRequest(_))));
assert!(store
.lock()
.unwrap()
.get("upd-repo-id")
.unwrap()
.repositories
.is_empty());
}