tests/prune.rs: regression tests fro prune

This commit is contained in:
Dietmar Maurer 2019-12-04 16:08:03 +01:00
parent aeeac29bb1
commit 0c875cf379

137
tests/prune.rs Normal file
View File

@ -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(())
}