test(blockifier): replace class revert

starkware-libs · Nov 20, 2024 · d364c56 · d364c56
1 parent 2beb6fd
commit d364c56
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/crates/blockifier/feature_contracts/cairo1/test_contract.cairo b/crates/blockifier/feature_contracts/cairo1/test_contract.cairo
@@ -624,6 +624,7 @@ mod TestContract {
         entry_point_selector: felt252,
         calldata: Array::<felt252>
     ) {
+        let class_hash_before_call = syscalls::get_class_hash_at_syscall(contract_address).unwrap_syscall();
          match syscalls::call_contract_syscall(
             contract_address, entry_point_selector, calldata.span())
         {
@@ -642,7 +643,8 @@ mod TestContract {
                 }
             },
         };
-        // TODO(Yoni, 1/12/2024): test replace class once get_class_hash_at syscall is supported.
+        let class_hash_after_call = syscalls::get_class_hash_at_syscall(contract_address).unwrap_syscall();
         assert(self.my_storage_var.read() == 0, 'values should not change.');
+        assert(class_hash_before_call == class_hash_after_call, 'class hash should not change if the tx reverted.');
     }
 }
diff --git a/crates/blockifier/src/execution/syscalls/syscall_tests/call_contract.rs b/crates/blockifier/src/execution/syscalls/syscall_tests/call_contract.rs
@@ -62,10 +62,6 @@ fn test_call_contract_that_panics() {
     );
     assert!(inner_call.execution.events.is_empty());
     assert!(inner_call.execution.l2_to_l1_messages.is_empty());
-    assert_eq!(
-        state.get_class_hash_at(inner_call.call.storage_address).unwrap(),
-        test_contract.get_class_hash()
-    );
 }
 
 #[cfg_attr(