Add tests for withdrawals provider

paradigmxyz · Aug 22, 2024 · 2b4a6da · 2b4a6da
1 parent ed6ee0c
commit 2b4a6da
Show file tree

Hide file tree

Showing 28 changed files with 293 additions and 98 deletions.
diff --git a/crates/blockchain-tree/src/block_buffer.rs b/crates/blockchain-tree/src/block_buffer.rs
@@ -191,7 +191,7 @@ mod tests {
 
     /// Create random block with specified number and parent hash.
     fn create_block<R: Rng>(rng: &mut R, number: u64, parent: BlockHash) -> SealedBlockWithSenders {
-        let block = random_block(rng, number, Some(parent), None, None, None);
+        let block = random_block(rng, number, Some(parent), None, None, None, None);
         block.seal_with_senders().unwrap()
     }
 

diff --git a/crates/consensus/beacon/src/engine/mod.rs b/crates/consensus/beacon/src/engine/mod.rs
@@ -2230,8 +2230,8 @@ mod tests {
                 })]))
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
-            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
+            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
             let (_static_dir, static_dir_path) = create_test_static_files_dir();
 
             insert_blocks(
@@ -2289,8 +2289,8 @@ mod tests {
                 .disable_blockchain_tree_sync()
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
-            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
+            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
 
             let (_static_dir, static_dir_path) = create_test_static_files_dir();
 
@@ -2305,7 +2305,8 @@ mod tests {
 
             let mut engine_rx = spawn_consensus_engine(consensus_engine);
 
-            let next_head = random_block(&mut rng, 2, Some(block1.hash()), None, Some(0), None);
+            let next_head =
+                random_block(&mut rng, 2, Some(block1.hash()), None, Some(0), None, None);
             let next_forkchoice_state = ForkchoiceState {
                 head_block_hash: next_head.hash(),
                 finalized_block_hash: block1.hash(),
@@ -2357,8 +2358,8 @@ mod tests {
                 .disable_blockchain_tree_sync()
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
-            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
+            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
 
             let (_static_dir, static_dir_path) = create_test_static_files_dir();
 
@@ -2404,16 +2405,19 @@ mod tests {
                 ]))
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
-            let mut block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
+            let mut block1 =
+                random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
             block1.header.set_difficulty(U256::from(1));
 
             // a second pre-merge block
-            let mut block2 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let mut block2 =
+                random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
             block2.header.set_difficulty(U256::from(1));
 
             // a transition block
-            let mut block3 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let mut block3 =
+                random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
             block3.header.set_difficulty(U256::from(1));
 
             let (_static_dir, static_dir_path) = create_test_static_files_dir();
@@ -2461,8 +2465,8 @@ mod tests {
                 ]))
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
-            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
+            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
 
             let (_temp_dir, temp_dir_path) = create_test_static_files_dir();
 
@@ -2525,7 +2529,7 @@ mod tests {
             // Send new payload
             let res = env
                 .send_new_payload(
-                    block_to_payload_v1(random_block(&mut rng, 0, None, None, Some(0), None)),
+                    block_to_payload_v1(random_block(&mut rng, 0, None, None, Some(0), None, None)),
                     None,
                 )
                 .await;
@@ -2536,7 +2540,7 @@ mod tests {
             // Send new payload
             let res = env
                 .send_new_payload(
-                    block_to_payload_v1(random_block(&mut rng, 1, None, None, Some(0), None)),
+                    block_to_payload_v1(random_block(&mut rng, 1, None, None, Some(0), None, None)),
                     None,
                 )
                 .await;
@@ -2565,9 +2569,9 @@ mod tests {
                 })]))
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
-            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None);
-            let block2 = random_block(&mut rng, 2, Some(block1.hash()), None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
+            let block1 = random_block(&mut rng, 1, Some(genesis.hash()), None, Some(0), None, None);
+            let block2 = random_block(&mut rng, 2, Some(block1.hash()), None, Some(0), None, None);
 
             let (_static_dir, static_dir_path) = create_test_static_files_dir();
             insert_blocks(
@@ -2635,8 +2639,15 @@ mod tests {
 
             let genesis =
                 SealedBlock { header: chain_spec.sealed_genesis_header(), ..Default::default() };
-            let block1 =
-                random_block(&mut rng, 1, Some(chain_spec.genesis_hash()), None, Some(0), None);
+            let block1 = random_block(
+                &mut rng,
+                1,
+                Some(chain_spec.genesis_hash()),
+                None,
+                Some(0),
+                None,
+                None,
+            );
 
             // TODO: add transactions that transfer from the alloc accounts, generating the new
             // block tx and state root
@@ -2686,7 +2697,7 @@ mod tests {
                 })]))
                 .build();
 
-            let genesis = random_block(&mut rng, 0, None, None, Some(0), None);
+            let genesis = random_block(&mut rng, 0, None, None, Some(0), None, None);
 
             let (_static_dir, static_dir_path) = create_test_static_files_dir();
 
@@ -2715,7 +2726,7 @@ mod tests {
 
             // Send new payload
             let parent = rng.gen();
-            let block = random_block(&mut rng, 2, Some(parent), None, Some(0), None);
+            let block = random_block(&mut rng, 2, Some(parent), None, Some(0), None, None);
             let res = env.send_new_payload(block_to_payload_v1(block), None).await;
             let expected_result = PayloadStatus::from_status(PayloadStatusEnum::Syncing);
             assert_matches!(res, Ok(result) => assert_eq!(result, expected_result));

diff --git a/crates/net/downloaders/src/bodies/bodies.rs b/crates/net/downloaders/src/bodies/bodies.rs
@@ -652,7 +652,7 @@ mod tests {
         // Generate some random blocks
         let db = create_test_rw_db();
         let mut rng = generators::rng();
-        let blocks = random_block_range(&mut rng, 0..=199, B256::ZERO, 1..2, None);
+        let blocks = random_block_range(&mut rng, 0..=199, B256::ZERO, 1..2, None, None);
 
         let headers = blocks.iter().map(|block| block.header.clone()).collect::<Vec<_>>();
         let bodies = blocks

diff --git a/crates/net/downloaders/src/test_utils/mod.rs b/crates/net/downloaders/src/test_utils/mod.rs
@@ -21,7 +21,7 @@ pub(crate) fn generate_bodies(
     range: RangeInclusive<u64>,
 ) -> (Vec<SealedHeader>, HashMap<B256, BlockBody>) {
     let mut rng = generators::rng();
-    let blocks = random_block_range(&mut rng, range, B256::ZERO, 0..2, None);
+    let blocks = random_block_range(&mut rng, range, B256::ZERO, 0..2, None, None);
 
     let headers = blocks.iter().map(|block| block.header.clone()).collect();
     let bodies = blocks

diff --git a/crates/prune/prune/src/segments/receipts.rs b/crates/prune/prune/src/segments/receipts.rs
@@ -99,7 +99,7 @@ mod tests {
         let db = TestStageDB::default();
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, 1..=10, B256::ZERO, 2..3, None);
+        let blocks = random_block_range(&mut rng, 1..=10, B256::ZERO, 2..3, None, None);
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");
 
         let mut receipts = Vec::new();

diff --git a/crates/prune/prune/src/segments/static_file/transactions.rs b/crates/prune/prune/src/segments/static_file/transactions.rs
@@ -105,7 +105,7 @@ mod tests {
         let db = TestStageDB::default();
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, 1..=100, B256::ZERO, 2..3, None);
+        let blocks = random_block_range(&mut rng, 1..=100, B256::ZERO, 2..3, None, None);
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");
 
         let transactions = blocks.iter().flat_map(|block| &block.body).collect::<Vec<_>>();

diff --git a/crates/prune/prune/src/segments/user/account_history.rs b/crates/prune/prune/src/segments/user/account_history.rs
@@ -151,7 +151,7 @@ mod tests {
         let db = TestStageDB::default();
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, 1..=5000, B256::ZERO, 0..1, None);
+        let blocks = random_block_range(&mut rng, 1..=5000, B256::ZERO, 0..1, None, None);
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");
 
         let accounts = random_eoa_accounts(&mut rng, 2).into_iter().collect::<BTreeMap<_, _>>();

diff --git a/crates/prune/prune/src/segments/user/receipts_by_logs.rs b/crates/prune/prune/src/segments/user/receipts_by_logs.rs
@@ -242,9 +242,9 @@ mod tests {
 
         let tip = 20000;
         let blocks = [
-            random_block_range(&mut rng, 0..=100, B256::ZERO, 1..5, None),
-            random_block_range(&mut rng, (100 + 1)..=(tip - 100), B256::ZERO, 0..1, None),
-            random_block_range(&mut rng, (tip - 100 + 1)..=tip, B256::ZERO, 1..5, None),
+            random_block_range(&mut rng, 0..=100, B256::ZERO, 1..5, None, None),
+            random_block_range(&mut rng, (100 + 1)..=(tip - 100), B256::ZERO, 0..1, None, None),
+            random_block_range(&mut rng, (tip - 100 + 1)..=tip, B256::ZERO, 1..5, None, None),
         ]
         .concat();
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");

diff --git a/crates/prune/prune/src/segments/user/sender_recovery.rs b/crates/prune/prune/src/segments/user/sender_recovery.rs
@@ -101,7 +101,7 @@ mod tests {
         let db = TestStageDB::default();
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, 1..=10, B256::ZERO, 2..3, None);
+        let blocks = random_block_range(&mut rng, 1..=10, B256::ZERO, 2..3, None, None);
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");
 
         let mut transaction_senders = Vec::new();

diff --git a/crates/prune/prune/src/segments/user/storage_history.rs b/crates/prune/prune/src/segments/user/storage_history.rs
@@ -157,7 +157,7 @@ mod tests {
         let db = TestStageDB::default();
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, 0..=5000, B256::ZERO, 0..1, None);
+        let blocks = random_block_range(&mut rng, 0..=5000, B256::ZERO, 0..1, None, None);
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");
 
         let accounts = random_eoa_accounts(&mut rng, 2).into_iter().collect::<BTreeMap<_, _>>();

diff --git a/crates/prune/prune/src/segments/user/transaction_lookup.rs b/crates/prune/prune/src/segments/user/transaction_lookup.rs
@@ -130,7 +130,7 @@ mod tests {
         let db = TestStageDB::default();
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, 1..=10, B256::ZERO, 2..3, None);
+        let blocks = random_block_range(&mut rng, 1..=10, B256::ZERO, 2..3, None, None);
         db.insert_blocks(blocks.iter(), StorageKind::Database(None)).expect("insert blocks");
 
         let mut tx_hash_numbers = Vec::new();

diff --git a/crates/rpc/rpc-engine-api/src/engine_api.rs b/crates/rpc/rpc-engine-api/src/engine_api.rs
@@ -1036,6 +1036,7 @@ mod tests {
                 B256::default(),
                 0..2,
                 None,
+                None,
             );
             handle.provider.extend_blocks(blocks.iter().cloned().map(|b| (b.hash(), b.unseal())));
 
@@ -1061,6 +1062,7 @@ mod tests {
                 B256::default(),
                 0..2,
                 None,
+                None,
             );
 
             // Insert only blocks in ranges 1-25 and 50-75
@@ -1153,9 +1155,9 @@ mod tests {
 
             let terminal_block_number = 1000;
             let consensus_terminal_block =
-                random_block(&mut rng, terminal_block_number, None, None, None, None);
+                random_block(&mut rng, terminal_block_number, None, None, None, None, None);
             let execution_terminal_block =
-                random_block(&mut rng, terminal_block_number, None, None, None, None);
+                random_block(&mut rng, terminal_block_number, None, None, None, None, None);
 
             let transition_config = TransitionConfiguration {
                 terminal_total_difficulty: handle
@@ -1196,8 +1198,15 @@ mod tests {
             let (handle, api) = setup_engine_api();
 
             let terminal_block_number = 1000;
-            let terminal_block =
-                random_block(&mut generators::rng(), terminal_block_number, None, None, None, None);
+            let terminal_block = random_block(
+                &mut generators::rng(),
+                terminal_block_number,
+                None,
+                None,
+                None,
+                None,
+                None,
+            );
 
             let transition_config = TransitionConfiguration {
                 terminal_total_difficulty: handle

diff --git a/crates/rpc/rpc-engine-api/tests/it/payload.rs b/crates/rpc/rpc-engine-api/tests/it/payload.rs
@@ -32,7 +32,7 @@ fn transform_block<F: FnOnce(Block) -> Block>(src: SealedBlock, f: F) -> Executi
 #[test]
 fn payload_body_roundtrip() {
     let mut rng = generators::rng();
-    for block in random_block_range(&mut rng, 0..=99, B256::default(), 0..2, None) {
+    for block in random_block_range(&mut rng, 0..=99, B256::default(), 0..2, None, None) {
         let unsealed = block.clone().unseal();
         let payload_body: ExecutionPayloadBodyV1 = convert_to_payload_body_v1(unsealed);
 
@@ -53,7 +53,7 @@ fn payload_body_roundtrip() {
 fn payload_validation() {
     let mut rng = generators::rng();
     let parent = rng.gen();
-    let block = random_block(&mut rng, 100, Some(parent), Some(3), Some(0), None);
+    let block = random_block(&mut rng, 100, Some(parent), Some(3), Some(0), None, None);
 
     // Valid extra data
     let block_with_valid_extra_data = transform_block(block.clone(), |mut b| {

diff --git a/crates/stages/stages/benches/setup/mod.rs b/crates/stages/stages/benches/setup/mod.rs
@@ -116,7 +116,8 @@ pub(crate) fn txs_testdata(num_blocks: u64) -> TestStageDB {
         .into_iter()
         .collect();
 
-        let mut blocks = random_block_range(&mut rng, 0..=num_blocks, B256::ZERO, txs_range, None);
+        let mut blocks =
+            random_block_range(&mut rng, 0..=num_blocks, B256::ZERO, txs_range, None, None);
 
         let (transitions, start_state) = random_changeset_range(
             &mut rng,

diff --git a/crates/stages/stages/src/stages/bodies.rs b/crates/stages/stages/src/stages/bodies.rs
@@ -719,7 +719,7 @@ mod tests {
                 let mut rng = generators::rng();
 
                 // Static files do not support gaps in headers, so we need to generate 0 to end
-                let blocks = random_block_range(&mut rng, 0..=end, GENESIS_HASH, 0..2, None);
+                let blocks = random_block_range(&mut rng, 0..=end, GENESIS_HASH, 0..2, None, None);
                 self.db.insert_headers_with_td(blocks.iter().map(|block| &block.header))?;
                 if let Some(progress) = blocks.get(start as usize) {
                     // Insert last progress data

diff --git a/crates/stages/stages/src/stages/hashing_account.rs b/crates/stages/stages/src/stages/hashing_account.rs
@@ -72,7 +72,8 @@ impl AccountHashingStage {
 
         let mut rng = generators::rng();
 
-        let blocks = random_block_range(&mut rng, opts.blocks.clone(), B256::ZERO, opts.txs, None);
+        let blocks =
+            random_block_range(&mut rng, opts.blocks.clone(), B256::ZERO, opts.txs, None, None);
 
         for block in blocks {
             provider.insert_historical_block(block.try_seal_with_senders().unwrap()).unwrap();

diff --git a/crates/stages/stages/src/stages/hashing_storage.rs b/crates/stages/stages/src/stages/hashing_storage.rs
@@ -342,7 +342,8 @@ mod tests {
             let n_accounts = 31;
             let mut accounts = random_contract_account_range(&mut rng, &mut (0..n_accounts));
 
-            let blocks = random_block_range(&mut rng, stage_progress..=end, B256::ZERO, 0..3, None);
+            let blocks =
+                random_block_range(&mut rng, stage_progress..=end, B256::ZERO, 0..3, None, None);
 
             self.db.insert_headers(blocks.iter().map(|block| &block.header))?;
 

diff --git a/crates/stages/stages/src/stages/index_account_history.rs b/crates/stages/stages/src/stages/index_account_history.rs
@@ -538,7 +538,7 @@ mod tests {
                 .into_iter()
                 .collect::<BTreeMap<_, _>>();
 
-            let blocks = random_block_range(&mut rng, start..=end, B256::ZERO, 0..3, None);
+            let blocks = random_block_range(&mut rng, start..=end, B256::ZERO, 0..3, None, None);
 
             let (changesets, _) = random_changeset_range(
                 &mut rng,

diff --git a/crates/stages/stages/src/stages/index_storage_history.rs b/crates/stages/stages/src/stages/index_storage_history.rs
@@ -560,7 +560,7 @@ mod tests {
                 .into_iter()
                 .collect::<BTreeMap<_, _>>();
 
-            let blocks = random_block_range(&mut rng, start..=end, B256::ZERO, 0..3, None);
+            let blocks = random_block_range(&mut rng, start..=end, B256::ZERO, 0..3, None, None);
 
             let (changesets, _) = random_changeset_range(
                 &mut rng,

diff --git a/crates/stages/stages/src/stages/merkle.rs b/crates/stages/stages/src/stages/merkle.rs
@@ -502,6 +502,7 @@ mod tests {
                     B256::ZERO,
                     0..1,
                     None,
+                    None,
                 ));
                 self.db.insert_blocks(preblocks.iter(), StorageKind::Static)?;
             }
@@ -522,6 +523,7 @@ mod tests {
                 Some(0),
                 None,
                 None,
+                None,
             );
             let mut header = header.unseal();
 
@@ -536,7 +538,7 @@ mod tests {
 
             let head_hash = sealed_head.hash();
             let mut blocks = vec![sealed_head];
-            blocks.extend(random_block_range(&mut rng, start..=end, head_hash, 0..3, None));
+            blocks.extend(random_block_range(&mut rng, start..=end, head_hash, 0..3, None, None));
             let last_block = blocks.last().cloned().unwrap();
             self.db.insert_blocks(blocks.iter(), StorageKind::Static)?;