blockifier: add a client side proving test state

Author: meship-starkware

crates/blockifier/src/test_utils/initial_test_state.rs

@@ -18,6 +18,7 @@ use crate::state::state_reader_and_contract_manager::{
 };
 use crate::test_utils::contracts::FeatureContractData;
 use crate::test_utils::dict_state_reader::DictStateReader;
+use crate::test_utils::generate_block_hash_storage_updates;
 
 /// Utility to fund an account.
 pub fn fund_account(
@@ -83,7 +84,6 @@ pub fn setup_test_state(
     }
 }
 
-// TODO(Meshi): Add block hash storage updates.
 pub fn test_state(
     chain_info: &ChainInfo,
     initial_balances: Fee,
@@ -104,13 +104,16 @@ pub fn test_state(
         .iter()
         .map(|(feature_contract, i)| ((*feature_contract).into(), *i))
         .collect();
-    test_state_inner(
+    let mut state = test_state_inner(
         chain_info,
         initial_balances,
         &contract_instances_vec[..],
         &HashVersion::V2,
         erc20_version,
-    )
+    );
+    let block_hash_mapping = generate_block_hash_storage_updates();
+    state.state.storage_view.extend(block_hash_mapping.storage);
+    state
 }
 
 pub fn test_state_with_contract_manager(

crates/blockifier/src/transaction/transactions_test.rs

@@ -599,6 +599,8 @@ fn test_invoke_tx(
     #[case] mut expected_arguments: ExpectedResultTestInvokeTx,
     #[case] account_cairo_version: CairoVersion,
     #[values(false, true)] use_kzg_da: bool,
+    #[values(ProofFacts::default(), ProofFacts::snos_proof_facts_for_testing())]
+    proof_facts: ProofFacts,
 ) {
     let block_context = &BlockContext::create_for_account_testing_with_kzg(use_kzg_da);
     let versioned_constants = &block_context.versioned_constants;
@@ -614,6 +616,7 @@ fn test_invoke_tx(
         sender_address: account_contract_address,
         calldata: Calldata(Arc::clone(&calldata.0)),
         resource_bounds,
+        proof_facts,
     });
 
     // Extract invoke transaction fields for testing, as it is consumed when creating an account