#![allow(clippy::missing_docs_in_private_items)]
use super::*;
use crate::routines::{new_store, slugify, Repository};
use std::collections::HashMap;
use std::sync::{Arc, Mutex};
fn make_routine(id: &str, title: &str, created_at: u64, updated_at: u64) -> Routine {
Routine {
id: id.to_string(),
schedule: "@daily".to_string(),
title: title.to_string(),
agent: "claude".to_string(),
prompt: "do the thing".to_string(),
repositories: vec![],
enabled: true,
source: "managed".to_string(),
created_at,
updated_at,
last_triggered_at: None,
ttl_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))
}
#[test]
fn svc_list_sorts_by_updated_at() {
let store = store_with(vec![
make_routine("late", "Zeta", 100, 300),
make_routine("early", "Alpha", 100, 100),
make_routine("mid", "Mid", 100, 200),
]);
let query = RoutineListQuery {
sort: RoutineSort::Updated,
..Default::default()
};
let list = svc_list(&store, &query);
assert_eq!(list[0].routine.id, "early");
assert_eq!(list[1].routine.id, "mid");
assert_eq!(list[2].routine.id, "late");
}
#[test]
fn svc_list_sorts_by_title_case_insensitively() {
let store = store_with(vec![
make_routine("banana", "banana", 0, 0),
make_routine("apple", "Apple", 0, 0),
make_routine("cherry", "CHERRY", 0, 0),
]);
let query = RoutineListQuery {
sort: RoutineSort::Title,
..Default::default()
};
let list = svc_list(&store, &query);
assert_eq!(list[0].routine.id, "apple");
assert_eq!(list[1].routine.id, "banana");
assert_eq!(list[2].routine.id, "cherry");
}
#[test]
fn svc_create_rejects_duplicate_slug() {
let title = "Svc Create Dup ZZZ";
let store = new_store();
let first = svc_create(
&store,
CreateRoutineRequest {
schedule: "@daily".into(),
title: title.into(),
agent: "claude".into(),
prompt: "p".into(),
repositories: vec![],
enabled: true,
ttl_secs: None,
},
)
.unwrap();
let conflict = svc_create(
&store,
CreateRoutineRequest {
schedule: "@daily".into(),
title: " svc create DUP zzz ".into(),
agent: "claude".into(),
prompt: "p".into(),
repositories: vec![],
enabled: true,
ttl_secs: None,
},
);
assert!(matches!(conflict, Err(AppError::Conflict(_))));
svc_delete(&store, &first.routine.id).unwrap();
let _ = crate::routine_storage::remove_routine_dir(&slugify(title));
}
#[test]
fn svc_update_rejects_renaming_into_existing_slug() {
let title_keep = "Svc Update Keep ZZZ";
let title_other = "Svc Update Other ZZZ";
let store = new_store();
let routine_keep = make_routine("keep-id", title_keep, 1, 1);
let routine_other = make_routine("other-id", title_other, 2, 2);
crate::routine_storage::write_routine(&routine_keep).unwrap();
crate::routine_storage::write_routine(&routine_other).unwrap();
store.lock().unwrap().insert("keep-id".into(), routine_keep);
store
.lock()
.unwrap()
.insert("other-id".into(), routine_other);
let conflict = svc_update(
&store,
"other-id",
UpdateRoutineRequest {
schedule: None,
title: Some(title_keep.into()),
agent: None,
prompt: None,
repositories: None,
enabled: None,
ttl_secs: None,
},
);
assert!(matches!(conflict, Err(AppError::Conflict(_))));
let _ = crate::routine_storage::remove_routine_dir(&slugify(title_keep));
let _ = crate::routine_storage::remove_routine_dir(&slugify(title_other));
}
#[test]
fn svc_update_sets_ttl_secs() {
let title = "Svc Update Ttl ZZZ";
let store = new_store();
let routine = make_routine("ttl-id", title, 1, 1);
crate::routine_storage::write_routine(&routine).unwrap();
store.lock().unwrap().insert("ttl-id".into(), routine);
let updated = svc_update(
&store,
"ttl-id",
UpdateRoutineRequest {
schedule: None,
title: None,
agent: None,
prompt: None,
repositories: None,
enabled: None,
ttl_secs: Some(4242),
},
)
.unwrap();
assert_eq!(updated.routine.ttl_secs, Some(4242));
let _ = crate::routine_storage::remove_routine_dir(&slugify(title));
}
#[test]
fn svc_logs_returns_newest_workbench_log() {
let title = "Svc Logs Newest ZZZ";
let slug = slugify(title);
let store = new_store();
let mut routine = make_routine("logs-id", title, 1, 1);
routine.repositories = vec![Repository {
repository: "https://example.com/r.git".into(),
branch: None,
}];
store.lock().unwrap().insert("logs-id".into(), routine);
let workbenches = crate::paths::workbenches_dir();
std::fs::create_dir_all(&workbenches).unwrap();
let older = workbenches.join(format!("{slug}-1000"));
let newer = workbenches.join(format!("{slug}-2000"));
std::fs::create_dir_all(&older).unwrap();
std::fs::create_dir_all(&newer).unwrap();
std::fs::write(older.join("agent.log"), "old log contents").unwrap();
std::fs::write(newer.join("agent.log"), "new log contents").unwrap();
let logs = svc_logs(&store, "logs-id").unwrap();
assert_eq!(logs, "new log contents");
let _ = std::fs::remove_dir_all(&older);
let _ = std::fs::remove_dir_all(&newer);
}
#[test]
fn svc_logs_skips_foreign_and_unparseable_workbenches() {
let title = "Svc Logs Mixed ZZQ";
let slug = slugify(title);
let store = new_store();
let routine = make_routine("logs-mixed-id", title, 1, 1);
store
.lock()
.unwrap()
.insert("logs-mixed-id".into(), routine);
let workbenches = crate::paths::workbenches_dir();
std::fs::create_dir_all(&workbenches).unwrap();
let unparseable = workbenches.join("not-a-workbench-name");
std::fs::create_dir_all(&unparseable).unwrap();
std::fs::write(unparseable.join("agent.log"), "ignored").unwrap();
let foreign = workbenches.join("some-other-routine-9999");
std::fs::create_dir_all(&foreign).unwrap();
std::fs::write(foreign.join("agent.log"), "foreign log").unwrap();
let mine = workbenches.join(format!("{slug}-4242"));
std::fs::create_dir_all(&mine).unwrap();
std::fs::write(mine.join("agent.log"), "mine log contents").unwrap();
let logs = svc_logs(&store, "logs-mixed-id").unwrap();
assert_eq!(logs, "mine log contents");
let _ = std::fs::remove_dir_all(&unparseable);
let _ = std::fs::remove_dir_all(&foreign);
let _ = std::fs::remove_dir_all(&mine);
}
#[test]
fn svc_logs_empty_when_workbenches_dir_absent() {
let title = "Svc Logs No Workbenches ZZQ";
let store = new_store();
store.lock().unwrap().insert(
"logs-empty-id".into(),
make_routine("logs-empty-id", title, 1, 1),
);
let fresh_home = std::env::temp_dir().join(format!("moadim-no-wb-{}", uuid::Uuid::new_v4()));
std::fs::create_dir_all(&fresh_home).unwrap();
let saved = std::env::var_os("HOME");
unsafe {
std::env::set_var("HOME", &fresh_home);
}
assert!(!crate::paths::workbenches_dir().exists());
let logs = svc_logs(&store, "logs-empty-id").unwrap();
unsafe {
match saved {
Some(value) => std::env::set_var("HOME", value),
None => std::env::remove_var("HOME"),
}
}
assert_eq!(logs, "");
let _ = std::fs::remove_dir_all(&fresh_home);
}
#[test]
fn svc_logs_missing_routine_not_found() {
assert!(matches!(
svc_logs(&new_store(), "nope"),
Err(AppError::NotFound)
));
}
static PATH_GUARD: Mutex<()> = Mutex::new(());
fn with_empty_path(body: impl FnOnce()) {
let guard = PATH_GUARD
.lock()
.unwrap_or_else(|poison| poison.into_inner());
let saved = std::env::var_os("PATH");
std::env::set_var("PATH", "");
body();
match saved {
Some(value) => std::env::set_var("PATH", value),
None => std::env::remove_var("PATH"),
}
drop(guard);
}
#[test]
fn svc_create_warns_when_crontab_sync_fails() {
let title = "Svc Create Sync Fail ZZZ";
let store = new_store();
with_empty_path(|| {
let created = svc_create(
&store,
CreateRoutineRequest {
schedule: "@daily".into(),
title: title.into(),
agent: "claude".into(),
prompt: "p".into(),
repositories: vec![],
enabled: true,
ttl_secs: None,
},
)
.unwrap();
assert_eq!(created.routine.title, title);
});
let _ = crate::routine_storage::remove_routine_dir(&slugify(title));
}
#[test]
fn svc_update_warns_when_crontab_sync_fails() {
let title = "Svc Update Sync Fail ZZZ";
let store = new_store();
let routine = make_routine("upd-sync-id", title, 1, 1);
crate::routine_storage::write_routine(&routine).unwrap();
store.lock().unwrap().insert("upd-sync-id".into(), routine);
with_empty_path(|| {
let updated = svc_update(
&store,
"upd-sync-id",
UpdateRoutineRequest {
schedule: None,
title: None,
agent: None,
prompt: Some("changed".into()),
repositories: None,
enabled: None,
ttl_secs: None,
},
)
.unwrap();
assert_eq!(updated.routine.prompt, "changed");
});
let _ = crate::routine_storage::remove_routine_dir(&slugify(title));
}
#[test]
fn svc_delete_warns_when_crontab_sync_fails() {
let title = "Svc Delete Sync Fail ZZZ";
let store = new_store();
let routine = make_routine("del-sync-id", title, 1, 1);
crate::routine_storage::write_routine(&routine).unwrap();
store.lock().unwrap().insert("del-sync-id".into(), routine);
with_empty_path(|| {
let deleted = svc_delete(&store, "del-sync-id").unwrap();
assert_eq!(deleted.routine.title, title);
});
let _ = crate::routine_storage::remove_routine_dir(&slugify(title));
}
#[test]
fn svc_trigger_warns_when_spawn_fails() {
let agent_name = "svc-trigger-spawn-fail-agent-zzz";
std::fs::create_dir_all(crate::paths::agents_dir()).unwrap();
let cfg = crate::paths::agent_toml_path(agent_name);
std::fs::write(&cfg, "command = \"true\"\nargs = []\n").unwrap();
let title = "Svc Trigger Spawn Fail ZZZ";
let store = new_store();
let mut routine = make_routine("trig-spawn-id", title, 1, 1);
routine.agent = agent_name.into();
crate::routine_storage::write_routine(&routine).unwrap();
store
.lock()
.unwrap()
.insert("trig-spawn-id".into(), routine);
with_empty_path(|| {
let triggered = svc_trigger(&store, "trig-spawn-id").unwrap();
assert!(triggered.last_triggered_at.is_some());
});
let _ = std::fs::remove_file(&cfg);
let _ = crate::routine_storage::remove_routine_dir(&slugify(title));
}