Cheatcode for getting hash of the block

crates/cheatnet/src/runtime_extensions/forge_runtime_extension/cheatcodes/cheat_block_hash.rs

@@ -0,0 +1,66 @@
+use super::cheat_execution_info::{
+    BlockInfoMockOperations, CheatArguments, ExecutionInfoMockOperations, Operation,
+};
+use crate::state::CheatSpan;
+use crate::CheatnetState;
+use cairo_vm::Felt252;
+use starknet_api::core::ContractAddress;
+
+impl CheatnetState {
+    pub fn cheat_block_hash(
+        &mut self,
+        contract_address: ContractAddress,
+        block_hash: Felt252,
+        span: CheatSpan,
+    ) {
+        self.cheat_execution_info(ExecutionInfoMockOperations {
+            block_info: BlockInfoMockOperations {
+                block_hash: Operation::Start(CheatArguments {
+                    value: block_hash,
+                    span,
+                    target: contract_address,
+                }),
+                ..Default::default()
+            },
+            ..Default::default()
+        });
+    }
+
+    pub fn start_cheat_block_hash_global(&mut self, block_hash: Felt252) {
+        self.cheat_execution_info(ExecutionInfoMockOperations {
+            block_info: BlockInfoMockOperations {
+                block_hash: Operation::StartGlobal(block_hash),
+                ..Default::default()
+            },
+            ..Default::default()
+        });
+    }
+
+    pub fn start_cheat_block_hash(
+        &mut self,
+        contract_address: ContractAddress,
+        block_hash: Felt252,
+    ) {
+        self.cheat_block_hash(contract_address, block_hash, CheatSpan::Indefinite);
+    }
+
+    pub fn stop_cheat_block_hash(&mut self, contract_address: ContractAddress) {
+        self.cheat_execution_info(ExecutionInfoMockOperations {
+            block_info: BlockInfoMockOperations {
+                block_hash: Operation::Stop(contract_address),
+                ..Default::default()
+            },
+            ..Default::default()
+        });
+    }
+
+    pub fn stop_cheat_block_hash_global(&mut self) {
+        self.cheat_execution_info(ExecutionInfoMockOperations {
+            block_info: BlockInfoMockOperations {
+                block_hash: Operation::StopGlobal,
+                ..Default::default()
+            },
+            ..Default::default()
+        });
+    }
+}

crates/cheatnet/src/runtime_extensions/forge_runtime_extension/cheatcodes/cheat_execution_info.rs

@@ -51,6 +51,7 @@ pub struct TxInfoMock {
 pub struct BlockInfoMock {
     pub block_number: CheatStatus<u64>,
     pub block_timestamp: CheatStatus<u64>,
+    pub block_hash: CheatStatus<Felt252>,
     pub sequencer_address: CheatStatus<ContractAddress>,
 }
 
@@ -82,6 +83,7 @@ pub struct TxInfoMockOperations {
 pub struct BlockInfoMockOperations {
     pub block_number: Operation<u64>,
     pub block_timestamp: Operation<u64>,
+    pub block_hash: Operation<Felt252>,
     pub sequencer_address: Operation<ContractAddress>,
 }
 
@@ -98,6 +100,7 @@ macro_rules! for_all_fields {
 
         $macro!(block_info.block_number);
         $macro!(block_info.block_timestamp);
+        $macro!(block_info.block_hash);
         $macro!(block_info.sequencer_address);
 
         $macro!(tx_info.version);

crates/cheatnet/src/runtime_extensions/forge_runtime_extension/cheatcodes/mod.rs

@@ -3,6 +3,7 @@ use cairo_vm::vm::errors::hint_errors::HintError;
 use runtime::EnhancedHintError;
 use starknet_types_core::felt::Felt;
 
+pub mod cheat_block_hash;
 pub mod cheat_block_number;
 pub mod cheat_block_timestamp;
 pub mod cheat_caller_address;