update to first proxmox crate split

[proxmox-backup.git] / tests / prune.rs

diff --git a/tests/prune.rs b/tests/prune.rs
index 2648930f26a22e39444075a86010b3d04bf25745..4853ee49905ca9e8ce8014328fddc5b569d81734 100644 (file)
--- a/tests/prune.rs
+++ b/tests/prune.rs
@@ -1,26 +1,26 @@
-use failure::*;
 use std::path::PathBuf;
 
-use proxmox_backup::backup::*;
+use anyhow::Error;
+
+use pbs_api_types::PruneOptions;
+use pbs_datastore::manifest::MANIFEST_BLOB_NAME;
+use pbs_datastore::prune::compute_prune_info;
+use pbs_datastore::{BackupDir, BackupInfo};
 
 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>,
+    return_kept: bool,
+    options: &PruneOptions,
 ) -> Vec<PathBuf> {
 
-   let mut prune_info = BackupGroup::compute_prune_info(
-        list, keep_last, keep_daily, keep_weekly, keep_monthly, keep_yearly).unwrap();
+    let mut prune_info = compute_prune_info(list, options).unwrap();
 
     prune_info.reverse();
-    
+
     prune_info
         .iter()
         .filter_map(|(info, keep)| {
-            if *keep {
+            if *keep != return_kept {
                 None
             } else {
                 Some(info.backup_dir.relative_path())
@@ -34,17 +34,126 @@ fn create_info(
     partial: bool,
 ) -> BackupInfo {
 
-    let backup_dir = BackupDir::parse(snapshot).unwrap();
+    let backup_dir: BackupDir = snapshot.parse().unwrap();
 
     let mut files = Vec::new();
 
     if !partial {
         files.push(String::from(MANIFEST_BLOB_NAME));
     }
-    
+
     BackupInfo { backup_dir, files }
 }
 
+#[test]
+fn test_prune_hourly() -> Result<(), Error> {
+
+    let mut orig_list = Vec::new();
+
+    orig_list.push(create_info("host/elsa/2019-11-15T09:39:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-15T10:49:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-15T10:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-15T11:39:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-15T11:49:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-15T11:59:15Z", false));
+
+    let list = orig_list.clone();
+    let mut options = PruneOptions::default();
+    options.keep_hourly = Some(3);
+    let remove_list = get_prune_list(list, false, &options);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-11-15T10:49:15Z"),
+        PathBuf::from("host/elsa/2019-11-15T11:39:15Z"),
+        PathBuf::from("host/elsa/2019-11-15T11:49:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list;
+    let mut options = PruneOptions::default();
+    options.keep_hourly = Some(2);
+    let remove_list = get_prune_list(list, true, &options);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-11-15T10:59:15Z"),
+        PathBuf::from("host/elsa/2019-11-15T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    Ok(())
+}
+
+#[test]
+fn test_prune_simple2() -> Result<(), Error> {
+
+    let mut orig_list = Vec::new();
+
+    orig_list.push(create_info("host/elsa/2018-11-15T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-15T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-21T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-22T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-11-29T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-12-01T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-12-02T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-12-03T11:59:15Z", false));
+    orig_list.push(create_info("host/elsa/2019-12-04T11:59:15Z", false));
+
+    let list = orig_list.clone();
+    let mut options = PruneOptions::default();
+    options.keep_daily = Some(1);
+    let remove_list = get_prune_list(list, true, &options);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list.clone();
+    let mut options = PruneOptions::default();
+    options.keep_last = Some(1);
+    options.keep_daily = Some(1);
+    let remove_list = get_prune_list(list, true, &options);
+    let expect: Vec<PathBuf> = vec![
+        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 mut options = PruneOptions::default();
+    options.keep_daily = Some(1);
+    options.keep_weekly = Some(1);
+    let remove_list = get_prune_list(list, true, &options);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-12-01T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list.clone();
+    let mut options = PruneOptions::default();
+    options.keep_daily = Some(1);
+    options.keep_weekly = Some(1);
+    options.keep_monthly = Some(1);
+    let remove_list = get_prune_list(list, true, &options);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2019-11-22T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-01T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    let list = orig_list;
+    let mut options = PruneOptions::default();
+    options.keep_monthly = Some(1);
+    options.keep_yearly = Some(1);
+    let remove_list = get_prune_list(list, true, &options);
+    let expect: Vec<PathBuf> = vec![
+        PathBuf::from("host/elsa/2018-11-15T11:59:15Z"),
+        PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
+    ];
+    assert_eq!(remove_list, expect);
+
+    Ok(())
+}
+
 #[test]
 fn test_prune_simple() -> Result<(), Error> {
 
@@ -55,23 +164,28 @@ fn test_prune_simple() -> Result<(), Error> {
     orig_list.push(create_info("host/elsa/2019-12-04T11:59:15Z", false));
     orig_list.push(create_info("host/elsa/2019-12-04T12:59:15Z", false));
 
-    
     // keep-last tests
 
     let list = orig_list.clone();
-    let remove_list = get_prune_list(list, Some(4), None, None, None, None);
+    let mut options = PruneOptions::default();
+    options.keep_last = Some(4);
+    let remove_list = get_prune_list(list, false, &options);
     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 mut options = PruneOptions::default();
+    options.keep_last = Some(3);
+    let remove_list = get_prune_list(list, false, &options);
     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 mut options = PruneOptions::default();
+    options.keep_last = Some(2);
+    let remove_list = get_prune_list(list, false, &options);
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
         PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
@@ -79,7 +193,9 @@ fn test_prune_simple() -> Result<(), Error> {
     assert_eq!(remove_list, expect);
 
     let list = orig_list.clone();
-    let remove_list = get_prune_list(list, Some(1), None, None, None, None);
+    let mut options = PruneOptions::default();
+    options.keep_last = Some(1);
+    let remove_list = get_prune_list(list, false, &options);
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
         PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
@@ -88,7 +204,9 @@ fn test_prune_simple() -> Result<(), Error> {
     assert_eq!(remove_list, expect);
 
     let list = orig_list.clone();
-    let remove_list = get_prune_list(list, Some(0), None, None, None, None);
+    let mut options = PruneOptions::default();
+    options.keep_last = Some(0);
+    let remove_list = get_prune_list(list, false, &options);
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
         PathBuf::from("host/elsa/2019-12-03T11:59:15Z"),
@@ -99,19 +217,26 @@ fn test_prune_simple() -> Result<(), Error> {
 
     // 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 mut options = PruneOptions::default();
+    options.keep_last = Some(2);
+    options.keep_daily = Some(2);
+    let remove_list = get_prune_list(list, false, &options);
     let expect: Vec<PathBuf> = vec![];
     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 mut options = PruneOptions::default();
+    options.keep_daily = Some(3);
+    let remove_list = get_prune_list(list, false, &options);
     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 mut options = PruneOptions::default();
+    options.keep_daily = Some(2);
+    let remove_list = get_prune_list(list, false, &options);
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
         PathBuf::from("host/elsa/2019-12-04T11:59:15Z"),
@@ -120,7 +245,9 @@ fn test_prune_simple() -> Result<(), Error> {
 
     // keep-weekly
     let list = orig_list.clone();
-    let remove_list = get_prune_list(list, None, None, Some(5), None, None);
+    let mut options = PruneOptions::default();
+    options.keep_weekly = Some(5);
+    let remove_list = get_prune_list(list, false, &options);
     // all backup are within the same week, so we only keep a single file
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
@@ -131,16 +258,22 @@ fn test_prune_simple() -> Result<(), Error> {
 
     // keep-daily + keep-weekly
     let list = orig_list.clone();
-    let remove_list = get_prune_list(list, None, Some(1), Some(5), None, None);
+    let mut options = PruneOptions::default();
+    options.keep_daily = Some(1);
+    options.keep_weekly = Some(5);
+    let remove_list = get_prune_list(list, false, &options);
     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 mut options = PruneOptions::default();
+    options.keep_monthly = Some(6);
+    let remove_list = get_prune_list(list, false, &options);
     // all backup are within the same month, so we only keep a single file
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
@@ -151,7 +284,9 @@ fn test_prune_simple() -> Result<(), Error> {
 
     // keep-yearly
     let list = orig_list.clone();
-    let remove_list = get_prune_list(list, None, None, None, None, Some(7));
+    let mut options = PruneOptions::default();
+    options.keep_yearly = Some(7);
+    let remove_list = get_prune_list(list, false, &options);
     // all backup are within the same year, so we only keep a single file
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),
@@ -161,8 +296,12 @@ fn test_prune_simple() -> Result<(), Error> {
     assert_eq!(remove_list, expect);
 
     // keep-weekly + keep-monthly + keep-yearly
-    let list = orig_list.clone();
-    let remove_list = get_prune_list(list, None, None, Some(5), Some(6), Some(7));
+    let list = orig_list;
+    let mut options = PruneOptions::default();
+    options.keep_weekly = Some(5);
+    options.keep_monthly = Some(6);
+    options.keep_yearly = Some(7);
+    let remove_list = get_prune_list(list, false, &options);
     // all backup are within one week, so we only keep a single file
     let expect: Vec<PathBuf> = vec![
         PathBuf::from("host/elsa/2019-12-02T11:59:15Z"),