Skip to content

Commit 37eb5cc

Browse files
authored
Blockstore: fix last tests using legacy shreds (#7308)
* fix test_get_slot_entries_with_shred_count_corruption * fix test_merkle_root_metas_data
1 parent d3038f3 commit 37eb5cc

File tree

1 file changed

+38
-22
lines changed

1 file changed

+38
-22
lines changed

ledger/src/blockstore.rs

Lines changed: 38 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -5315,7 +5315,7 @@ pub mod tests {
53155315
crate::{
53165316
genesis_utils::{create_genesis_config, GenesisConfigInfo},
53175317
leader_schedule::{FixedSchedule, IdentityKeyedLeaderSchedule},
5318-
shred::{max_ticks_per_n_shreds, ShredFlags},
5318+
shred::max_ticks_per_n_shreds,
53195319
},
53205320
assert_matches::assert_matches,
53215321
bincode::{serialize, Options},
@@ -7542,18 +7542,25 @@ pub mod tests {
75427542
index
75437543
);
75447544

7545-
// Add a shred from different fec set
7545+
let shredder = Shredder::new(slot, slot.saturating_sub(1), 0, 0).unwrap();
7546+
let keypair = Keypair::new();
7547+
let reed_solomon_cache = ReedSolomonCache::default();
75467548
let new_index = fec_set_index + 31;
7547-
let new_data_shred = Shred::new_from_data(
7548-
slot,
7549-
new_index,
7550-
1, // parent_offset
7551-
&[3, 3, 3], // data
7552-
ShredFlags::empty(),
7553-
0, // reference_tick,
7554-
0, // version
7555-
fec_set_index + 30,
7556-
);
7549+
// Add a shred from different fec set
7550+
let new_data_shred = shredder
7551+
.make_shreds_from_data_slice(
7552+
&keypair,
7553+
&[3, 3, 3],
7554+
false,
7555+
Some(Hash::default()),
7556+
new_index,
7557+
new_index,
7558+
&reed_solomon_cache,
7559+
&mut ProcessShredsStats::default(),
7560+
)
7561+
.unwrap()
7562+
.next()
7563+
.unwrap();
75577564

75587565
let mut shred_insertion_tracker =
75597566
ShredInsertionTracker::new(data_shreds.len(), blockstore.db.batch().unwrap());
@@ -7868,16 +7875,25 @@ pub mod tests {
78687875
);
78697876

78707877
// Insert an empty shred that won't deshred into entries
7871-
let shreds = vec![Shred::new_from_data(
7872-
slot,
7873-
next_shred_index as u32,
7874-
1,
7875-
&[1, 1, 1],
7876-
ShredFlags::LAST_SHRED_IN_SLOT,
7877-
0,
7878-
0,
7879-
next_shred_index as u32,
7880-
)];
7878+
7879+
let shredder = Shredder::new(slot, slot.saturating_sub(1), 0, 0).unwrap();
7880+
let keypair = Keypair::new();
7881+
let reed_solomon_cache = ReedSolomonCache::default();
7882+
7883+
let shreds: Vec<Shred> = shredder
7884+
.make_shreds_from_data_slice(
7885+
&keypair,
7886+
&[1, 1, 1],
7887+
true,
7888+
Some(Hash::default()),
7889+
next_shred_index as u32,
7890+
next_shred_index as u32,
7891+
&reed_solomon_cache,
7892+
&mut ProcessShredsStats::default(),
7893+
)
7894+
.unwrap()
7895+
.take(DATA_SHREDS_PER_FEC_BLOCK)
7896+
.collect();
78817897

78827898
// With the corruption, nothing should be returned, even though an
78837899
// earlier data block was valid

0 commit comments

Comments
 (0)