khmodify: add test

1 files changed, 43 insertions, 5 deletions

diff --git a/src/actions/modify.rs b/src/actions/modify.rs
index cffe6fd..0763aaa 100644
--- a/src/actions/modify.rs
+++ b/src/actions/modify.rs
@@ -6,27 +6,65 @@ use crate::utils::fileutil::write_cal;
 pub fn do_modify(args: &[&str]) -> KhResult<()> {
   info!("do_modify");
 
+  let khlines = input::default_input_khlines()?;
+
   if args[0] == "removeprop" && args[1] == "xlicerror" {
     let dry_run = args.len() >= 3 && args[2] == "--dry-run";
 
-    let lines = input::default_input_khlines()?;
-    for khline in lines {
+    for khline in khlines {
       let cal = khline.to_cal()?.with_remove_property("X-LIC-ERROR");
       if cal.1 > 0 {
         if !dry_run {
-          println!("Modifying {}", cal.0.get_path_as_string().unwrap());
+          info!("Modifying {}", cal.0.get_path_as_string().unwrap());
 
           let backup_path = backup(&khline).unwrap();
           info!("Backup written to {}", backup_path.display());
           write_cal(&cal.0)?
         } else {
-          println!("Would modify {}", cal.0.get_path_as_string().unwrap());
+          info!("Would modify {}", cal.0.get_path_as_string().unwrap());
         };
       }
-    };
+    }
   } else {
     error!("not supported");
   }
 
   Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+  use super::*;
+
+  use crate::testutils::*;
+  use assert_fs::prelude::*;
+  use predicates::prelude::*;
+
+  #[test]
+  fn test_do_modify() {
+    let testdir = prepare_testdir("testdir_with_xlicerror");
+    let args = ["removeprop", "xlicerror"];
+
+    do_modify(&args).unwrap();
+
+    let expected = indoc!("
+      BEGIN:VCALENDAR
+      PRODID:CommuniGate Pro 6.2.5
+      VERSION:2.0
+      BEGIN:VEVENT
+      DTSTAMP:20180813T160004Z
+      UID:1c441c1b-8ca7-4898-b670-49ce30a7137b
+      SEQUENCE:2
+      SUMMARY:some summary
+      DTSTART:20161007T073000Z
+      DTEND:20161007T160000Z
+      LAST-MODIFIED:20161018T095049Z
+      CREATED:20161018T094913Z
+      PRIORITY:5
+      END:VEVENT
+      END:VCALENDAR
+    ").replace("\n", "\r\n");
+    let predicate = predicate::str::similar(expected);
+    testdir.child(".khaleesi/cal/xlicerror.ics").assert(predicate);
+  }
+}