tests/prune.rs: regression tests fro prune

diff --git a/tests/prune.rs b/tests/prune.rs
new file mode 100644 (file)
index 0000000..b9b1386
--- /dev/null
+++ b/tests/prune.rs
@@ -0,0 +1,137 @@
+use failure::*;
+use std::path::PathBuf;
+
+use proxmox_backup::backup::*;
+
+const TESTDIR: &str = "/tmp/prunetest";
+
+fn get_prune_list(
+    list: Vec<BackupInfo>,
+    keep_last: Option<u64>,
+    keep_daily: Option<u64>,
+    keep_weekly: Option<u64>,
+    keep_monthly: Option<u64>,
+    keep_yearly: Option<u64>,
+) -> Vec<PathBuf> {
+
+    let remove_list = BackupGroup::compute_prune_list(
+        list, keep_last, keep_daily, keep_weekly, keep_monthly, keep_yearly).unwrap();
+
+    remove_list
+        .iter()
+        .map(|d| d.backup_dir.relative_path())
+        .collect()
+}
+
+#[test]
+fn test_prune_simple() -> Result<(), Error> {
+
+    let _ = std::fs::remove_dir_all(TESTDIR);
+    std::fs::create_dir(TESTDIR)?;
+
+    std::fs::create_dir_all(format!("{}/{}", TESTDIR, "host/elsa/2019-12-02T11:59:15Z"))?;
+    std::fs::create_dir_all(format!("{}/{}", TESTDIR, "host/elsa/2019-12-03T11:59:15Z"))?;
+    std::fs::create_dir_all(format!("{}/{}", TESTDIR, "host/elsa/2019-12-04T11:59:15Z"))?;
+    std::fs::create_dir_all(format!("{}/{}", TESTDIR, "host/elsa/2019-12-04T12:59:15Z"))?;
+
+
+    let orig_list = BackupInfo::list_backups(std::path::Path::new(TESTDIR))?;
+
+    // keep-last tests
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, Some(4), None, None, None, None);
+    let expect: Vec<PathBuf> = Vec::new();
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, Some(3), None, None, None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, Some(2), None, None, None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, Some(1), None, None, None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, Some(0), None, None, None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T12:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    // keep-last, keep-daily mixed
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, Some(2), Some(2), None, None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    // keep-daily test
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, None, Some(3), None, None, None);
+    let expect: Vec<PathBuf> = vec![PathBuf::from("host/elsa/2019-12-04T11:59:15Z")];
+    assert_eq!(remove_list, expect);
+
+    // keep-daily test
+
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, None, Some(2), None, None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    // keep-weekly
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, None, None, Some(5), None, None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+     // keep-monthly
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, None, None, None, Some(6), None);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+     // keep-yearly
+    let list = orig_list.clone();
+    let remove_list = get_prune_list(list, None, None, None, None, Some(7));
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    Ok(())
+}