impl mock evaluator

anna-carroll · anna-carroll · commit 3d2b025b7f92 · 2025-02-20T17:09:26.000-06:00
diff --git a/src/tasks/simulator.rs b/src/tasks/simulator.rs
@@ -139,7 +139,10 @@ where
                     }
                     Some(Ok(Some(candidate))) = futs.join_next() => {
                         println!("candidate used gas: {:?}", candidate.result.result.gas_used());
-                        if candidate.score > best.as_ref().map(|b| b.score).unwrap_or_default() {
+                        // TODO: think about equality statement here. 
+                        // if using ">" then no candidate will be returned if every score is zero
+                        // if using ">=" then the candidate will be replaced every time if every score is zero 
+                        if candidate.score >= best.as_ref().map(|b| b.score).unwrap_or_default() {
                             best = Some(candidate);
                         }
                     }
diff --git a/tests/simulator_test.rs b/tests/simulator_test.rs
@@ -12,7 +12,7 @@ use std::str::FromStr;
 use std::sync::Arc;
 use tokio::sync::mpsc;
 use tokio::time::{Duration, Instant};
-use trevm::revm::primitives::ResultAndState;
+use trevm::revm::primitives::{Account, ExecutionResult, ResultAndState};
 use trevm::{NoopBlock, NoopCfg};
 
 #[tokio::test(flavor = "multi_thread")]
@@ -56,11 +56,23 @@ async fn test_spawn() {
 
     // Check the result
     assert!(best.is_some());
-    assert_eq!(best.unwrap().score, U256::from(1));
+    assert_eq!(best.unwrap().score, U256::from(0));
 }
 
-fn mock_evaluator(_state: &ResultAndState) -> U256 {
-    U256::from(1)
+fn mock_evaluator(state: &ResultAndState) -> U256 {
+    // log the transaction results
+    match &state.result {
+        ExecutionResult::Success { .. } => println!("Execution was successful."),
+        ExecutionResult::Revert { .. } => println!("Execution reverted."),
+        ExecutionResult::Halt { .. } => println!("Execution halted."),
+    }
+
+    // return the target account balance
+    let target_addr = Address::from_str("0x0000000000000000000000000000000000000000").unwrap();
+    let default_account = Account::default();
+    let target_account = state.state.get(&target_addr).unwrap_or(&default_account);
+    println!("target account balance: {:?}", target_account.info.balance);
+    target_account.info.balance
 }
 
 // Returns a new signed test transaction with default values

Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,10 @@ where`
`139`	`139`	`}`
`140`	`140`	`Some(Ok(Some(candidate))) = futs.join_next() => {`
`141`	`141`	`println!("candidate used gas: {:?}", candidate.result.result.gas_used());`
`142`		`- if candidate.score > best.as_ref().map(\|b\| b.score).unwrap_or_default() {`
	`142`	`+ // TODO: think about equality statement here.`
	`143`	`+ // if using ">" then no candidate will be returned if every score is zero`
	`144`	`+ // if using ">=" then the candidate will be replaced every time if every score is zero`
	`145`	`+ if candidate.score >= best.as_ref().map(\|b\| b.score).unwrap_or_default() {`
`143`	`146`	`best = Some(candidate);`
`144`	`147`	`}`
`145`	`148`	`}`