Fix test and assert

solana-labs · Apr 23, 2021 · e37d33b · e37d33b
1 parent d432e0f
commit e37d33b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/core/src/replay_stage.rs b/core/src/replay_stage.rs
@@ -2948,6 +2948,7 @@ pub(crate) mod tests {
             let gibberish = [0xa5u8; PACKET_DATA_SIZE];
             let mut data_header = DataShredHeader::default();
             data_header.flags |= DATA_COMPLETE_SHRED;
+            data_header.size = payload_len as u16;
             let mut shred = Shred::new_empty_from_header(
                 ShredCommonHeader::default(),
                 data_header,

diff --git a/core/src/serve_repair.rs b/core/src/serve_repair.rs
@@ -939,6 +939,7 @@ mod tests {
             assert_eq!(shreds[0].slot(), 1);
             assert_eq!(shreds[0].index(), 0);
             shreds[0].payload.push(10);
+            shreds[0].data_header.size = shreds[0].payload.len() as u16;
             blockstore
                 .insert_shreds(shreds, None, false)
                 .expect("Expect successful ledger write");

diff --git a/ledger/src/shred.rs b/ledger/src/shred.rs
@@ -398,6 +398,7 @@ impl Shred {
                 &data_header,
             )
             .expect("Failed to write data header into shred buffer");
+            assert!(data_header.size as usize >= start);
         } else if common_header.shred_type == ShredType(CODING_SHRED) {
             Self::serialize_obj_into(
                 &mut start,
@@ -407,7 +408,6 @@ impl Shred {
             )
             .expect("Failed to write data header into shred buffer");
         }
-        assert!(data_header.size as usize >= start);
         Shred {
             common_header,
             data_header,