78 lines
2.6 KiB
Rust
78 lines
2.6 KiB
Rust
use crate::argparse::DescriptionMod;
|
|
use crate::invocation::{apply_modification, ResolvedModification};
|
|
use taskchampion::{Replica, Status};
|
|
use termcolor::WriteColor;
|
|
|
|
pub(in crate::invocation) fn execute<W: WriteColor>(
|
|
w: &mut W,
|
|
replica: &mut Replica,
|
|
mut modification: ResolvedModification,
|
|
) -> Result<(), crate::Error> {
|
|
// extract the description from the modification to handle it specially
|
|
let description = match modification.0.description {
|
|
DescriptionMod::Set(ref s) => s.clone(),
|
|
_ => "(no description)".to_owned(),
|
|
};
|
|
modification.0.description = DescriptionMod::None;
|
|
|
|
let task = replica.new_task(Status::Pending, description).unwrap();
|
|
let mut task = task.into_mut(replica);
|
|
apply_modification(&mut task, &modification)?;
|
|
writeln!(w, "added task {}", task.get_uuid())?;
|
|
Ok(())
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
use crate::argparse::Modification;
|
|
use crate::invocation::test::*;
|
|
use pretty_assertions::assert_eq;
|
|
|
|
#[test]
|
|
fn test_add() {
|
|
let mut w = test_writer();
|
|
let mut replica = test_replica();
|
|
let modification = ResolvedModification(Modification {
|
|
description: DescriptionMod::Set(s!("my description")),
|
|
..Default::default()
|
|
});
|
|
execute(&mut w, &mut replica, modification).unwrap();
|
|
|
|
// check that the task appeared..
|
|
let working_set = replica.working_set().unwrap();
|
|
let task = replica
|
|
.get_task(working_set.by_index(1).unwrap())
|
|
.unwrap()
|
|
.unwrap();
|
|
assert_eq!(task.get_description(), "my description");
|
|
assert_eq!(task.get_status(), Status::Pending);
|
|
|
|
assert_eq!(w.into_string(), format!("added task {}\n", task.get_uuid()));
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_with_tags() {
|
|
let mut w = test_writer();
|
|
let mut replica = test_replica();
|
|
let modification = ResolvedModification(Modification {
|
|
description: DescriptionMod::Set(s!("my description")),
|
|
add_tags: vec![tag!("tag1")].drain(..).collect(),
|
|
..Default::default()
|
|
});
|
|
execute(&mut w, &mut replica, modification).unwrap();
|
|
|
|
// check that the task appeared..
|
|
let working_set = replica.working_set().unwrap();
|
|
let task = replica
|
|
.get_task(working_set.by_index(1).unwrap())
|
|
.unwrap()
|
|
.unwrap();
|
|
assert_eq!(task.get_description(), "my description");
|
|
assert_eq!(task.get_status(), Status::Pending);
|
|
assert!(task.has_tag(&tag!("tag1")));
|
|
|
|
assert_eq!(w.into_string(), format!("added task {}\n", task.get_uuid()));
|
|
}
|
|
}
|