Skip to content
This repository has been archived by the owner on Jan 22, 2025. It is now read-only.

Commit

Permalink
Fix test_concurrent_snapshot_packaging
Browse files Browse the repository at this point in the history
(cherry picked from commit 3acf956)
  • Loading branch information
mvines authored and mergify-bot committed Feb 27, 2020
1 parent 405f9a9 commit 6448384
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 8 deletions.
15 changes: 10 additions & 5 deletions core/tests/bank_forks.rs
Original file line number Diff line number Diff line change
Expand Up @@ -232,9 +232,8 @@ mod tests {
let saved_snapshots_dir = TempDir::new().unwrap();
let saved_accounts_dir = TempDir::new().unwrap();
let saved_slot = 4;
let saved_tar = snapshot_config
.snapshot_package_output_path
.join(saved_slot.to_string());
let mut saved_archive_path = None;

for forks in 0..MAX_CACHE_ENTRIES + 2 {
let bank = Bank::new_from_parent(
&bank_forks[forks as u64],
Expand All @@ -246,6 +245,7 @@ mod tests {
let tx = system_transaction::transfer(&mint_keypair, &key1, 1, genesis_config.hash());
assert_eq!(bank.process_transaction(&tx), Ok(()));
bank.squash();
let accounts_hash = bank.update_accounts_hash();
bank_forks.insert(bank);

let package_sender = {
Expand Down Expand Up @@ -286,6 +286,11 @@ mod tests {
&options,
)
.unwrap();

saved_archive_path = Some(snapshot_utils::get_snapshot_archive_path(
&snapshot_config.snapshot_package_output_path,
&(slot, accounts_hash),
));
}
}

Expand Down Expand Up @@ -323,7 +328,7 @@ mod tests {
.join()
.expect("SnapshotPackagerService exited with error");

// Check the tar we cached the state for earlier was generated correctly
// Check the archive we cached the state for earlier was generated correctly

// before we compare, stick an empty status_cache in this dir so that the package comparision works
// This is needed since the status_cache is added by the packager and is not collected from
Expand All @@ -342,7 +347,7 @@ mod tests {
.unwrap();

snapshot_utils::verify_snapshot_archive(
saved_tar,
saved_archive_path.unwrap(),
saved_snapshots_dir.path(),
saved_accounts_dir
.path()
Expand Down
7 changes: 4 additions & 3 deletions ledger/src/snapshot_utils.rs
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,8 @@ pub fn archive_snapshot_package(snapshot_package: &SnapshotPackage) -> Result<()

timer.stop();
info!(
"Successfully created tarball. slot: {}, elapsed ms: {}, size={}",
"Successfully created {:?}. slot: {}, elapsed ms: {}, size={}",
snapshot_package.tar_output_file,
snapshot_package.root,
timer.as_ms(),
metadata.len()
Expand Down Expand Up @@ -650,7 +651,7 @@ fn get_io_error(error: &str) -> SnapshotError {
}

pub fn verify_snapshot_archive<P, Q, R>(
snapshot_tar: P,
snapshot_archive: P,
snapshots_to_verify: Q,
storages_to_verify: R,
) where
Expand All @@ -660,7 +661,7 @@ pub fn verify_snapshot_archive<P, Q, R>(
{
let temp_dir = tempfile::TempDir::new().unwrap();
let unpack_dir = temp_dir.path();
untar_snapshot_in(snapshot_tar, &unpack_dir).unwrap();
untar_snapshot_in(snapshot_archive, &unpack_dir).unwrap();

// Check snapshots are the same
let unpacked_snapshots = unpack_dir.join(&TAR_SNAPSHOTS_DIR);
Expand Down

0 comments on commit 6448384

Please sign in to comment.