chore: tests

Author: Evalir

testdata/cheats/TryFfi.sol

@@ -0,0 +1,33 @@
+// SPDX-License-Identifier: Unlicense
+pragma solidity >=0.8.18;
+
+import "ds-test/test.sol";
+import "./Vm.sol";
+
+contract TryFfiTest is DSTest {
+    Vm constant vm = Vm(HEVM_ADDRESS);
+
+    function testTryFfi() public {
+        string[] memory inputs = new string[](3);
+        inputs[0] = "bash";
+        inputs[1] = "-c";
+        inputs[2] =
+            "echo -n 0x0000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000000966666920776f726b730000000000000000000000000000000000000000000000";
+
+        Vm.FfiResult memory f = vm.tryFfi(inputs);
+        (string memory output) = abi.decode(f.stdout, (string));
+        assertEq(output, "ffi works", "ffi failed");
+        assertEq(f.exit_code, 0, "ffi failed");
+    }
+
+    function testTryFfiFail() public {
+        string[] memory inputs = new string[](3);
+        inputs[0] = "bash";
+        inputs[1] = "-c";
+        inputs[2] = "quikmafs";
+
+        Vm.FfiResult memory f = vm.tryFfi(inputs);
+        assert(f.exit_code != 0);
+        assertEq(string(f.stderr), string("bash: quikmafs: command not found\n"));
+    }
+}

testdata/cheats/Vm.sol

@@ -52,6 +52,12 @@ interface Vm {
         uint256 privateKey;
     }
 
+    struct FfiResult {
+        int32 exit_code;
+        bytes stdout;
+        bytes stderr;
+    }
+
     // Set block.timestamp (newTimestamp)
     function warp(uint256) external;
 
@@ -116,6 +122,9 @@ interface Vm {
     // Performs a foreign function call via terminal, (stringInputs) => (result)
     function ffi(string[] calldata) external returns (bytes memory);
 
+    // Performs a foreign function call via terminal and returns the exit code, stdout, and stderr
+    function tryFfi(string[] calldata) external returns (FfiResult memory);
+
     // Set environment variables, (name, value)
     function setEnv(string calldata, string calldata) external;