Adds benchmark to user-test Program

OffchainLabs · Dec 11, 2024 · ecc4e1f · ecc4e1f
1 parent 228f963
commit ecc4e1f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/arbitrator/wasm-libraries/user-test/src/program.rs b/arbitrator/wasm-libraries/user-test/src/program.rs
@@ -8,6 +8,7 @@ use arbutil::{
         user::UserOutcomeKind,
         EvmData,
     },
+    benchmark::Benchmark,
     Bytes20, Bytes32, Color,
 };
 use caller_env::{static_caller::STATIC_MEM, GuestPtr, MemAccess};
@@ -28,6 +29,7 @@ impl From<MemoryBoundsError> for eyre::ErrReport {
 /// Mock type representing a `user_host::Program`
 pub struct Program {
     evm_api: MockEvmApi,
+    benchmark: Option<Benchmark>,
 }
 
 #[allow(clippy::unit_arg)]
@@ -52,6 +54,10 @@ impl UserHost<VecReader> for Program {
         &EVM_DATA
     }
 
+    fn benchmark(&mut self) -> &mut Option<Benchmark> {
+        &mut self.benchmark
+    }
+
     fn evm_return_data_len(&mut self) -> &mut u32 {
         unimplemented!()
     }
@@ -91,6 +97,7 @@ impl Program {
     pub fn current() -> Self {
         Self {
             evm_api: MockEvmApi,
+            benchmark: None,
         }
     }