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(
@@ -104,13 +105,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/account_transactions_test.rs

@@ -55,6 +55,7 @@ use starknet_api::transaction::fields::{
     ContractAddressSalt,
     Fee,
     GasVectorComputationMode,
+    ProofFacts,
     Resource,
     ResourceBounds,
     ValidResourceBounds,
@@ -2191,3 +2192,31 @@ fn test_missing_validate_entrypoint_rejects(
         if ret == retdata![Felt::from_hex(ENTRYPOINT_NOT_FOUND_ERROR).unwrap()]
     );
 }
+
+#[rstest]
+fn test_invoke_tx_with_proof_facts(
+    block_context: BlockContext,
+    default_all_resource_bounds: ValidResourceBounds,
+) {
+    let TestInitData { mut state, account_address, contract_address, mut nonce_manager } =
+        create_test_init_data(
+            &block_context.chain_info,
+            CairoVersion::Cairo1(RunnableCairo1::Casm),
+        );
+
+    // Create an invoke transaction with valid proof facts.
+    let tx_execution_info = run_invoke_tx(
+        &mut state,
+        &block_context,
+        invoke_tx_args! {
+            sender_address: account_address,
+            calldata: create_trivial_calldata(contract_address),
+            resource_bounds: default_all_resource_bounds,
+            nonce: nonce_manager.next(account_address),
+            proof_facts: ProofFacts::snos_proof_facts_for_testing(),
+        },
+    )
+    .unwrap();
+
+    assert!(!tx_execution_info.is_reverted());
+}