feat: add setter utils to blockenv (#116)

Author: mattsse

crates/evm/src/env.rs

@@ -1,5 +1,6 @@
 //! Configuration types for EVM environment.
 
+use alloy_primitives::U256;
 use revm::{
     context::{BlockEnv, CfgEnv},
     primitives::hardfork::SpecId,
@@ -44,6 +45,81 @@ impl<Spec> EvmEnv<Spec> {
     pub const fn spec_id(&self) -> &Spec {
         &self.cfg_env.spec
     }
+
+    /// Overrides the configured block number
+    pub fn with_block_number(mut self, number: U256) -> Self {
+        self.block_env.number = number;
+        self
+    }
+
+    /// Convenience function that overrides the configured block number with the given
+    /// `Some(number)`.
+    ///
+    /// This is intended for block overrides.
+    pub fn with_block_number_opt(mut self, number: Option<U256>) -> Self {
+        if let Some(number) = number {
+            self.block_env.number = number;
+        }
+        self
+    }
+
+    /// Sets the block number if provided.
+    pub fn set_block_number_opt(&mut self, number: Option<U256>) -> &mut Self {
+        if let Some(number) = number {
+            self.block_env.number = number;
+        }
+        self
+    }
+
+    /// Overrides the configured block timestamp.
+    pub fn with_timestamp(mut self, timestamp: U256) -> Self {
+        self.block_env.timestamp = timestamp;
+        self
+    }
+
+    /// Convenience function that overrides the configured block timestamp with the given
+    /// `Some(timestamp)`.
+    ///
+    /// This is intended for block overrides.
+    pub fn with_timestamp_opt(mut self, timestamp: Option<U256>) -> Self {
+        if let Some(timestamp) = timestamp {
+            self.block_env.timestamp = timestamp;
+        }
+        self
+    }
+
+    /// Sets the block timestamp if provided.
+    pub fn set_timestamp_opt(&mut self, timestamp: Option<U256>) -> &mut Self {
+        if let Some(timestamp) = timestamp {
+            self.block_env.timestamp = timestamp;
+        }
+        self
+    }
+
+    /// Overrides the configured block base fee.
+    pub fn with_base_fee(mut self, base_fee: u64) -> Self {
+        self.block_env.basefee = base_fee;
+        self
+    }
+
+    /// Convenience function that overrides the configured block base fee with the given
+    /// `Some(base_fee)`.
+    ///
+    /// This is intended for block overrides.
+    pub fn with_base_fee_opt(mut self, base_fee: Option<u64>) -> Self {
+        if let Some(base_fee) = base_fee {
+            self.block_env.basefee = base_fee;
+        }
+        self
+    }
+
+    /// Sets the block base fee if provided.
+    pub fn set_base_fee_opt(&mut self, base_fee: Option<u64>) -> &mut Self {
+        if let Some(base_fee) = base_fee {
+            self.block_env.basefee = base_fee;
+        }
+        self
+    }
 }
 
 impl<Spec> From<(CfgEnv<Spec>, BlockEnv)> for EvmEnv<Spec> {