add extension to downloaded snapshots

input-output-hk · Jul 11, 2023 · 109f7ac · 109f7ac
1 parent a8febdd
commit 109f7ac
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/mithril-client/src/aggregator_client/snapshot_client.rs b/mithril-client/src/aggregator_client/snapshot_client.rs
@@ -67,7 +67,7 @@ impl SnapshotClient {
     ) -> StdResult<PathBuf> {
         let filepath = PathBuf::new()
             .join(download_dir)
-            .join(format!("snapshot-{}", snapshot.digest));
+            .join(format!("snapshot-{}.tar.gz", snapshot.digest));
 
         for url in snapshot.locations.as_slice() {
             if self.http_client.probe(url).await.is_ok() {

diff --git a/mithril-client/src/services/snapshot.rs b/mithril-client/src/services/snapshot.rs
@@ -500,7 +500,11 @@ mod tests {
 
         let (_, verifier) = setup_genesis();
         let genesis_verification_key = verifier.to_verification_key();
-        build_dummy_snapshot("digest-10", "1234567890".repeat(124).as_str(), &test_path);
+        build_dummy_snapshot(
+            "digest-10.tar.gz",
+            "1234567890".repeat(124).as_str(),
+            &test_path,
+        );
         let filepath = snapshot_service
             .download(
                 &snapshot,
@@ -551,7 +555,11 @@ mod tests {
 
         let (_, verifier) = setup_genesis();
         let genesis_verification_key = verifier.to_verification_key();
-        build_dummy_snapshot("digest-10", "1234567890".repeat(124).as_str(), &test_path);
+        build_dummy_snapshot(
+            "digest-10.tar.gz",
+            "1234567890".repeat(124).as_str(),
+            &test_path,
+        );
         let err = snapshot_service
             .download(
                 &signed_entity,
@@ -577,7 +585,7 @@ mod tests {
                 "Expected a SnapshotServiceError when snapshot can not be verified. Got {err:?}: '{err}'"
             );
         }
-        let filepath = test_path.join("snapshot-digest-10");
+        let filepath = test_path.join("snapshot-digest-10.tar.gz");
         assert!(filepath.exists());
         let unpack_dir = filepath
             .parent()