Add add reset method to immutable digest cache provider

Author: Alenar

mithril-common/src/digesters/cache/json_provider.rs

@@ -14,6 +14,7 @@ use std::{
     path::{Path, PathBuf},
 };
 use tokio::{
+    fs,
     fs::File,
     io::{AsyncReadExt, AsyncWriteExt},
 };
@@ -95,6 +96,14 @@ impl ImmutableFileDigestCacheProvider for JsonImmutableFileDigestCacheProvider {
 
         Ok(result)
     }
+
+    async fn reset(&self) -> CacheProviderResult<()> {
+        fs::remove_file(&self.filepath)
+            .await
+            .map_err(ImmutableDigesterCacheStoreError::from)?;
+
+        Ok(())
+    }
 }
 
 #[cfg(test)]
@@ -249,4 +258,38 @@ mod tests {
 
         assert_eq!(expected, result);
     }
+
+    #[tokio::test]
+    async fn reset_clear_existing_values() {
+        let file = get_test_dir("store_erase_existing_values").join("immutable-cache-store.json");
+        let provider = JsonImmutableFileDigestCacheProvider::new(&file);
+        let values_to_store = vec![
+            ("0.chunk".to_string(), "digest 0".to_string()),
+            ("1.chunk".to_string(), "digest 1".to_string()),
+        ];
+        let expected: BTreeMap<_, _> = BTreeMap::from([
+            (
+                ImmutableFile::dummy(PathBuf::default(), 0, "0.chunk".to_string()),
+                Some("digest 0".to_string()),
+            ),
+            (
+                ImmutableFile::dummy(PathBuf::default(), 1, "1.chunk".to_string()),
+                Some("digest 1".to_string()),
+            ),
+        ]);
+        let immutables = expected.keys().cloned().collect();
+
+        provider
+            .store(values_to_store)
+            .await
+            .expect("Cache write should not fail");
+        provider.reset().await.expect("reset should not fails");
+
+        let result: BTreeMap<_, _> = provider
+            .get(immutables)
+            .await
+            .expect("Cache read should not fail");
+
+        assert!(result.into_iter().all(|(_, cache)| cache.is_none()));
+    }
 }

mithril-common/src/digesters/cache/memory_provider.rs

@@ -59,6 +59,12 @@ impl ImmutableFileDigestCacheProvider for MemoryImmutableFileDigestCacheProvider
 
         Ok(result)
     }
+
+    async fn reset(&self) -> CacheProviderResult<()> {
+        let mut store = self.store.write().await;
+        store.clear();
+        Ok(())
+    }
 }
 
 #[cfg(test)]
@@ -185,4 +191,37 @@ mod tests {
 
         assert_eq!(expected, result);
     }
+
+    #[tokio::test]
+    async fn reset_clear_existing_values() {
+        let provider = MemoryImmutableFileDigestCacheProvider::default();
+        let values_to_store = vec![
+            ("0.chunk".to_string(), "digest 0".to_string()),
+            ("1.chunk".to_string(), "digest 1".to_string()),
+        ];
+        let expected: BTreeMap<_, _> = BTreeMap::from([
+            (
+                ImmutableFile::dummy(PathBuf::default(), 0, "0.chunk".to_string()),
+                Some("digest 0".to_string()),
+            ),
+            (
+                ImmutableFile::dummy(PathBuf::default(), 1, "1.chunk".to_string()),
+                Some("digest 1".to_string()),
+            ),
+        ]);
+        let immutables = expected.keys().cloned().collect();
+
+        provider
+            .store(values_to_store)
+            .await
+            .expect("Cache write should not fail");
+        provider.reset().await.expect("reset should not fails");
+
+        let result: BTreeMap<_, _> = provider
+            .get(immutables)
+            .await
+            .expect("Cache read should not fail");
+
+        assert!(result.into_iter().all(|(_, cache)| cache.is_none()));
+    }
 }

mithril-common/src/digesters/cache/provider.rs

@@ -63,4 +63,7 @@ pub trait ImmutableFileDigestCacheProvider: Sync + Send {
         &self,
         immutables: Vec<ImmutableFile>,
     ) -> CacheProviderResult<BTreeMap<ImmutableFile, Option<HexEncodedDigest>>>;
+
+    /// Reset the stored values
+    async fn reset(&self) -> CacheProviderResult<()>;
 }