Add better tests for state, code and balance overrides

Author: grandizzy

crates/cast/src/cmd/call.rs

@@ -56,6 +56,7 @@ static OVERRIDE_PATTERN: LazyLock<Regex> =
 ///   --override-nonce 0x123:1 \
 ///   --override-code 0x123:0x1234 \
 ///   --override-state 0x123:0x1:0x1234
+///   --override-state-diff 0x123:0x1:0x1234
 /// ```
 #[derive(Debug, Parser)]
 pub struct CallArgs {

crates/cast/tests/cli/main.rs

@@ -2258,34 +2258,110 @@ forgetest_async!(cast_call_custom_chain_id, |_prj, cmd| {
 });
 
 // https://github.com/foundry-rs/foundry/issues/10189
-forgetest_async!(cast_call_custom_override, |_prj, cmd| {
-    let (_api, handle) = anvil::spawn(
-        NodeConfig::test()
-            .with_auto_impersonate(true)
-            .with_eth_rpc_url(Some("https://sepolia.base.org")),
+forgetest_async!(cast_call_custom_override, |prj, cmd| {
+    let (_, handle) = anvil::spawn(NodeConfig::test()).await;
+
+    foundry_test_utils::util::initialize(prj.root());
+    prj.add_source(
+        "Counter",
+        r#"
+contract Counter {
+    uint256 public number;
+
+    function getBalance(address target) public returns (uint256) {
+        return target.balance;
+    }
+}
+   "#,
     )
-    .await;
+    .unwrap();
 
-    let http_endpoint = handle.http_endpoint();
+    // Deploy counter contract.
+    cmd.args([
+        "script",
+        "--private-key",
+        "0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80",
+        "--rpc-url",
+        &handle.http_endpoint(),
+        "--broadcast",
+        "CounterScript",
+    ])
+    .assert_success();
 
+    // Override state, `number()` should return overridden value.
     cmd.cast_fuse()
         .args([
             "call",
-            "5FbDB2315678afecb367f032d93F642f64180aa3",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3",
             "--rpc-url",
-            &http_endpoint,
+            &handle.http_endpoint(),
+            "--override-state",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3:0x0:0x1234",
+            "number()(uint256)",
+        ])
+        .assert_success()
+        .stdout_eq(str![[r#"
+4660
+
+"#]]);
+
+    // Override balance, `getBalance()` should return overridden value.
+    cmd.cast_fuse()
+        .args([
+            "call",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3",
+            "--rpc-url",
+            &handle.http_endpoint(),
             "--override-balance",
-            "5FbDB2315678afecb367f032d93F642f64180aa3:1234",
-            "--override-nonce",
-            "5FbDB2315678afecb367f032d93F642f64180aa3:5",
-            // "--override-code",
-            // "5FbDB2315678afecb367f032d93F642f64180aa3:0x1234",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3:0x1111",
+            "getBalance(address)(uint256)",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3",
+        ])
+        .assert_success()
+        .stdout_eq(str![[r#"
+4369
+
+"#]]);
+
+    // Override code with
+    // contract Counter {
+    //     uint256 public number1;
+    // }
+    // Calling `number()` should fail.
+    cmd.cast_fuse()
+        .args([
+            "call",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3",
+            "--rpc-url",
+            &handle.http_endpoint(),
+            "--override-code",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3:0x6080604052348015600e575f5ffd5b50600436106026575f3560e01c8063c223a39e14602a575b5f5ffd5b60306044565b604051603b9190605f565b60405180910390f35b5f5481565b5f819050919050565b6059816049565b82525050565b5f60208201905060705f8301846052565b9291505056fea26469706673582212202a0acfb9083efed3e0e9f27177b090731d4392cf196d58e27e05088f59008d0964736f6c634300081d0033",
+            "number()(uint256)",
+        ])
+        .assert_failure()
+        .stderr_eq(str![[r#"
+Error: server returned an error response: error code 3: execution reverted, data: "0x"
+
+"#]]);
+
+    // Calling `number1()` with overridden state should return new value.
+    cmd.cast_fuse()
+        .args([
+            "call",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3",
+            "--rpc-url",
+            &handle.http_endpoint(),
+            "--override-code",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3:0x6080604052348015600e575f5ffd5b50600436106026575f3560e01c8063c223a39e14602a575b5f5ffd5b60306044565b604051603b9190605f565b60405180910390f35b5f5481565b5f819050919050565b6059816049565b82525050565b5f60208201905060705f8301846052565b9291505056fea26469706673582212202a0acfb9083efed3e0e9f27177b090731d4392cf196d58e27e05088f59008d0964736f6c634300081d0033",
             "--override-state",
-            "5FbDB2315678afecb367f032d93F642f64180aa3:0x1:1234",
-            // "--override-state-diff",
-            // "5FbDB2315678afecb367f032d93F642f64180aa3:0x2:1234",
+            "0x5FbDB2315678afecb367f032d93F642f64180aa3:0x0:0x2222",
+            "number1()(uint256)",
         ])
-        .assert_success();
+        .assert_success()
+        .stdout_eq(str![[r#"
+8738
+
+"#]]);
 });
 
 // https://github.com/foundry-rs/foundry/issues/9541