blockifier: update test contract for test_new_contract_flow

Author: dorimedini-starkware

crates/blockifier/src/execution/syscalls/syscall_tests/constants.rs

@@ -1 +1 @@
-pub const REQUIRED_GAS_GET_BLOCK_HASH_TEST: u64 = 15220;
+pub const REQUIRED_GAS_GET_BLOCK_HASH_TEST: u64 = 16720;

crates/blockifier/src/execution/syscalls/syscall_tests/get_block_hash.rs

@@ -48,7 +48,7 @@ fn positive_flow(runnable_version: RunnableCairo1) {
     let test_contract = FeatureContract::TestContract(CairoVersion::Cairo1(runnable_version));
     let (mut state, block_number, block_hash) = initialize_state(test_contract);
 
-    let calldata = calldata![block_number];
+    let calldata = calldata![block_number, block_hash];
     let entry_point_call = CallEntryPoint {
         entry_point_selector: selector_from_name("test_get_block_hash"),
         calldata,
@@ -83,7 +83,7 @@ fn positive_flow(runnable_version: RunnableCairo1) {
             l2_to_l1_messages: [],
             cairo_native: false,
             failed: false,
-            gas_consumed: 15220,
+            gas_consumed: 16020,
         }
     "#]]
     .assert_debug_eq(&call_info.execution);
@@ -98,7 +98,8 @@ fn negative_flow_block_number_out_of_range(runnable_version: RunnableCairo1) {
 
     let requested_block_number = CURRENT_BLOCK_NUMBER - constants::STORED_BLOCK_HASH_BUFFER + 1;
     let block_number = felt!(requested_block_number);
-    let calldata = calldata![block_number];
+    let dummy_block_hash = felt!(66_u64);
+    let calldata = calldata![block_number, dummy_block_hash];
     let entry_point_call = CallEntryPoint {
         entry_point_selector: selector_from_name("test_get_block_hash"),
         calldata,

crates/blockifier/src/execution/syscalls/syscall_tests/get_class_hash_at.rs

@@ -25,11 +25,13 @@ fn test_get_class_hash_at(runnable_version: RunnableCairo1) {
     let mut state = test_state(chain_info, BALANCE, &[(test_contract, 1)]);
     let address = contract_address!("0x111");
     let class_hash = class_hash!("0x222");
+    let class_hash_of_undeployed_contract = felt!("0x0");
+    let non_existing_address = felt!("0x333");
     state.state.address_to_class_hash.insert(address, class_hash);
 
     // Test deployed contract.
     let positive_entry_point_call = CallEntryPoint {
-        calldata: calldata![address.into(), class_hash.0],
+        calldata: calldata![address.into(), class_hash.0, non_existing_address],
         entry_point_selector: selector_from_name("test_get_class_hash_at"),
         ..trivial_external_entry_point_new(test_contract)
     };
@@ -51,18 +53,20 @@ fn test_get_class_hash_at(runnable_version: RunnableCairo1) {
             l2_to_l1_messages: [],
             cairo_native: false,
             failed: false,
-            gas_consumed: 16460,
+            gas_consumed: 29070,
         }
     "#]]
     .assert_debug_eq(&positive_call_info.execution);
     assert!(!positive_call_info.execution.failed);
     assert_eq!(positive_call_info.execution.retdata, retdata![]);
-    // Test undeployed contract - should return class_hash = 0 and succeed.
-    let non_existing_address = felt!("0x333");
-    let class_hash_of_undeployed_contract = felt!("0x0");
 
+    // Test undeployed contract - should return class_hash = 0 and succeed.
     let negative_entry_point_call = CallEntryPoint {
-        calldata: calldata![non_existing_address, class_hash_of_undeployed_contract],
+        calldata: calldata![
+            non_existing_address,
+            class_hash_of_undeployed_contract,
+            non_existing_address
+        ],
         entry_point_selector: selector_from_name("test_get_class_hash_at"),
         ..trivial_external_entry_point_new(test_contract)
     };
@@ -71,7 +75,7 @@ fn test_get_class_hash_at(runnable_version: RunnableCairo1) {
     // Sanity check: giving the wrong expected class hash to the test should make it fail.
     let different_class_hash = class_hash!("0x444");
     let different_class_hash_entry_point_call = CallEntryPoint {
-        calldata: calldata![address.into(), different_class_hash.0],
+        calldata: calldata![address.into(), different_class_hash.0, non_existing_address],
         entry_point_selector: selector_from_name("test_get_class_hash_at"),
         ..trivial_external_entry_point_new(test_contract)
     };

crates/blockifier/src/execution/syscalls/syscall_tests/out_of_gas.rs

@@ -18,9 +18,9 @@ use crate::test_utils::trivial_external_entry_point_new;
 #[test_case(RunnableCairo1::Casm; "VM")]
 fn test_out_of_gas(runnable_version: RunnableCairo1) {
     let test_contract = FeatureContract::TestContract(CairoVersion::Cairo1(runnable_version));
-    let (mut state, block_number, _block_hash) = initialize_state(test_contract);
+    let (mut state, block_number, block_hash) = initialize_state(test_contract);
 
-    let calldata = calldata![block_number];
+    let calldata = calldata![block_number, block_hash];
     let entry_point_call = CallEntryPoint {
         entry_point_selector: selector_from_name("test_get_block_hash"),
         calldata,