test(blockifier): global validate and execute sierra_gas; deploy

starkware-libs · Dec 10, 2024 · ca1fe92 · ca1fe92
1 parent 23fd4c4
commit ca1fe92
Show file tree

Hide file tree

Showing 4 changed files with 155 additions and 85 deletions.
diff --git a/crates/blockifier/src/transaction/account_transactions_test.rs b/crates/blockifier/src/transaction/account_transactions_test.rs
@@ -17,7 +17,7 @@ use starknet_api::executable_transaction::{
     AccountTransaction as ApiExecutableTransaction,
     DeclareTransaction as ApiExecutableDeclareTransaction,
 };
-use starknet_api::execution_resources::GasAmount;
+use starknet_api::execution_resources::{GasAmount, GasVector};
 use starknet_api::hash::StarkHash;
 use starknet_api::state::StorageKey;
 use starknet_api::test_utils::declare::executable_declare_tx;
@@ -79,7 +79,6 @@ use crate::test_utils::{
     DEFAULT_L1_DATA_GAS_MAX_AMOUNT,
     DEFAULT_L1_GAS_AMOUNT,
     DEFAULT_L2_GAS_MAX_AMOUNT,
-    DEFAULT_STRK_L1_DATA_GAS_PRICE,
     DEFAULT_STRK_L1_GAS_PRICE,
     DEFAULT_STRK_L2_GAS_PRICE,
     MAX_FEE,
@@ -95,6 +94,7 @@ use crate::transaction::test_utils::{
     calculate_class_info_for_testing,
     create_account_tx_for_validate_test_nonce_0,
     create_all_resource_bounds,
+    create_gas_amount_bounds_with_default_price,
     create_test_init_data,
     default_all_resource_bounds,
     default_l1_resource_bounds,
@@ -194,6 +194,8 @@ fn test_fee_enforcement(
     #[case] gas_bounds_mode: GasVectorComputationMode,
     #[values(true, false)] zero_bounds: bool,
 ) {
+    use crate::transaction::test_utils::create_gas_amount_bounds_with_default_price;
+
     let account = FeatureContract::AccountWithoutValidations(CairoVersion::Cairo0);
     let state = &mut test_state(&block_context.chain_info, BALANCE, &[(account, 1)]);
     let tx = executable_deploy_account_tx(
@@ -205,13 +207,12 @@ fn test_fee_enforcement(
                     (if zero_bounds { 0 } else { DEFAULT_L1_GAS_AMOUNT.0 }).into(),
                     DEFAULT_STRK_L1_GAS_PRICE.into()
                 ),
-                GasVectorComputationMode::All => create_all_resource_bounds(
-                    (if zero_bounds { 0 } else { DEFAULT_L1_GAS_AMOUNT.0 }).into(),
-                    DEFAULT_STRK_L1_GAS_PRICE.into(),
-                    (if zero_bounds { 0 } else { DEFAULT_L2_GAS_MAX_AMOUNT.0 }).into(),
-                    DEFAULT_STRK_L2_GAS_PRICE.into(),
-                    (if zero_bounds { 0 } else { DEFAULT_L1_DATA_GAS_MAX_AMOUNT.0 }).into(),
-                    DEFAULT_STRK_L1_DATA_GAS_PRICE.into(),
+                GasVectorComputationMode::All => create_gas_amount_bounds_with_default_price(
+                    GasVector{
+                        l1_gas: (if zero_bounds { 0 } else { DEFAULT_L1_GAS_AMOUNT.0 }).into(),
+                        l2_gas: (if zero_bounds { 0 } else { DEFAULT_L2_GAS_MAX_AMOUNT.0 }).into(),
+                        l1_data_gas: (if zero_bounds { 0 } else { DEFAULT_L1_DATA_GAS_MAX_AMOUNT.0 }).into(),
+                    },
                 ),
             },
             version,
@@ -237,13 +238,12 @@ fn test_all_bounds_combinations_enforce_fee(
     let expected_enforce_fee = l1_gas_bound + l1_data_gas_bound + l2_gas_bound > 0;
     let account_tx = invoke_tx_with_default_flags(invoke_tx_args! {
         version: TransactionVersion::THREE,
-        resource_bounds: create_all_resource_bounds(
-            l1_gas_bound.into(),
-            DEFAULT_STRK_L1_GAS_PRICE.into(),
-            l2_gas_bound.into(),
-            DEFAULT_STRK_L2_GAS_PRICE.into(),
-            l1_data_gas_bound.into(),
-            DEFAULT_STRK_L1_DATA_GAS_PRICE.into(),
+        resource_bounds: create_gas_amount_bounds_with_default_price(
+            GasVector {
+                l1_gas: l1_gas_bound.into(),
+                l2_gas: l2_gas_bound.into(),
+                l1_data_gas: l1_data_gas_bound.into(),
+            },
         ),
     });
     assert_eq!(account_tx.enforce_fee(), expected_enforce_fee);

diff --git a/crates/blockifier/src/transaction/post_execution_test.rs b/crates/blockifier/src/transaction/post_execution_test.rs
@@ -22,20 +22,12 @@ use crate::fee::fee_checks::FeeCheckError;
 use crate::state::state_api::StateReader;
 use crate::test_utils::contracts::FeatureContract;
 use crate::test_utils::initial_test_state::test_state;
-use crate::test_utils::{
-    create_calldata,
-    CairoVersion,
-    BALANCE,
-    DEFAULT_STRK_L1_DATA_GAS_PRICE,
-    DEFAULT_STRK_L1_GAS_PRICE,
-    DEFAULT_STRK_L2_GAS_PRICE,
-};
+use crate::test_utils::{create_calldata, CairoVersion, BALANCE, DEFAULT_STRK_L1_GAS_PRICE};
 use crate::transaction::account_transaction::AccountTransaction;
 use crate::transaction::errors::TransactionExecutionError;
 use crate::transaction::objects::{HasRelatedFeeType, TransactionInfoCreator};
 use crate::transaction::test_utils::{
     block_context,
-    create_all_resource_bounds,
     default_all_resource_bounds,
     default_l1_resource_bounds,
     invoke_tx_with_default_flags,
@@ -268,6 +260,10 @@ fn test_revert_on_resource_overuse(
     #[case] resource_to_decrement: Option<Resource>,
     #[values(CairoVersion::Cairo0)] cairo_version: CairoVersion,
 ) {
+    use starknet_api::execution_resources::GasVector;
+
+    use crate::transaction::test_utils::create_gas_amount_bounds_with_default_price;
+
     block_context.block_info.use_kzg_da = true;
     block_context.versioned_constants.allocation_cost = AllocationCost::ZERO;
     let gas_mode = resource_bounds.get_gas_vector_computation_mode();
@@ -374,14 +370,11 @@ fn test_revert_on_resource_overuse(
                     Resource::L2Gas => l2_gas.0 -= 1,
                     Resource::L1DataGas => l1_data_gas.0 -= 1,
                 }
-                create_all_resource_bounds(
+                create_gas_amount_bounds_with_default_price(GasVector {
                     l1_gas,
-                    DEFAULT_STRK_L1_GAS_PRICE.into(),
                     l2_gas,
-                    DEFAULT_STRK_L2_GAS_PRICE.into(),
                     l1_data_gas,
-                    DEFAULT_STRK_L1_DATA_GAS_PRICE.into(),
-                )
+                })
             }
         }
     };

diff --git a/crates/blockifier/src/transaction/test_utils.rs b/crates/blockifier/src/transaction/test_utils.rs
@@ -3,7 +3,7 @@ use starknet_api::abi::abi_utils::get_fee_token_var_address;
 use starknet_api::block::{FeeType, GasPrice};
 use starknet_api::contract_class::{ClassInfo, ContractClass, SierraVersion};
 use starknet_api::core::{ClassHash, ContractAddress, Nonce};
-use starknet_api::execution_resources::GasAmount;
+use starknet_api::execution_resources::{GasAmount, GasVector};
 use starknet_api::test_utils::declare::executable_declare_tx;
 use starknet_api::test_utils::deploy_account::{executable_deploy_account_tx, DeployAccountTxArgs};
 use starknet_api::test_utils::invoke::{executable_invoke_tx, InvokeTxArgs};
@@ -59,6 +59,16 @@ pub const STORAGE_WRITE: u64 = 8;
 
 /// Test fixtures.
 
+#[fixture]
+pub fn block_context() -> BlockContext {
+    BlockContext::create_for_account_testing()
+}
+
+#[fixture]
+pub fn versioned_constants(block_context: BlockContext) -> VersionedConstants {
+    block_context.versioned_constants().clone()
+}
+
 #[fixture]
 pub fn max_fee() -> Fee {
     MAX_FEE
@@ -80,42 +90,27 @@ pub fn create_resource_bounds(computation_mode: &GasVectorComputationMode) -> Va
         GasVectorComputationMode::NoL2Gas => {
             l1_resource_bounds(DEFAULT_L1_GAS_AMOUNT, DEFAULT_STRK_L1_GAS_PRICE.into())
         }
-        GasVectorComputationMode::All => create_all_resource_bounds(
-            DEFAULT_L1_GAS_AMOUNT,
-            DEFAULT_STRK_L1_GAS_PRICE.into(),
-            DEFAULT_L2_GAS_MAX_AMOUNT,
-            DEFAULT_STRK_L2_GAS_PRICE.into(),
-            DEFAULT_L1_DATA_GAS_MAX_AMOUNT,
-            DEFAULT_STRK_L1_DATA_GAS_PRICE.into(),
-        ),
+        GasVectorComputationMode::All => create_gas_amount_bounds_with_default_price(GasVector {
+            l1_gas: DEFAULT_L1_GAS_AMOUNT,
+            l1_data_gas: DEFAULT_L1_DATA_GAS_MAX_AMOUNT,
+            l2_gas: DEFAULT_L2_GAS_MAX_AMOUNT,
+        }),
     }
 }
 
 pub fn create_gas_amount_bounds_with_default_price(
-    l1_gas_amount: GasAmount,
-    l2_gas_amount: GasAmount,
-    l1_data_gas_amount: GasAmount,
+    GasVector { l1_gas, l1_data_gas, l2_gas }: GasVector,
 ) -> ValidResourceBounds {
     create_all_resource_bounds(
-        l1_gas_amount,
+        l1_gas,
         DEFAULT_STRK_L1_GAS_PRICE.into(),
-        l2_gas_amount,
+        l2_gas,
         DEFAULT_STRK_L2_GAS_PRICE.into(),
-        l1_data_gas_amount,
+        l1_data_gas,
         DEFAULT_STRK_L1_DATA_GAS_PRICE.into(),
     )
 }
 
-#[fixture]
-pub fn block_context() -> BlockContext {
-    BlockContext::create_for_account_testing()
-}
-
-#[fixture]
-pub fn versioned_constants(block_context: BlockContext) -> VersionedConstants {
-    block_context.versioned_constants().clone()
-}
-
 /// Struct containing the data usually needed to initialize a test.
 pub struct TestInitData {
     pub state: CachedState<DictStateReader>,