feat: add test cases

Author: 0xgregthedev

src/test-cases/precompiles/CallInputs.sol

@@ -12,26 +12,41 @@ contract TestCallInputs is Assertion, Test {
     constructor() payable {}
 
     function testGetCallInputs() external view {
-        PhEvm.CallInputs[] memory callInputs = ph.getCallInputs(address(TARGET), Target.readStorage.selector);
+        PhEvm.CallInputs[] memory callInputs = ph.getCallInputs(
+            address(TARGET),
+            Target.readStorage.selector
+        );
         require(callInputs.length == 1, "callInputs.length != 1");
         PhEvm.CallInputs memory callInput = callInputs[0];
 
-        require(callInput.target_address == address(TARGET), "callInput.target_address != target");
+        require(
+            callInput.target_address == address(TARGET),
+            "callInput.target_address != target"
+        );
         require(callInput.input.length == 0, "callInput.input.length != 0");
         require(callInput.value == 0, "callInput.value != 0");
 
-        callInputs = ph.getCallInputs(address(TARGET), Target.writeStorage.selector);
+        callInputs = ph.getCallInputs(
+            address(TARGET),
+            Target.writeStorage.selector
+        );
         require(callInputs.length == 2, "callInputs.length != 2");
 
         callInput = callInputs[0];
-        require(callInput.target_address == address(TARGET), "callInput.target_address != target");
+        require(
+            callInput.target_address == address(TARGET),
+            "callInput.target_address != target"
+        );
         require(callInput.input.length == 32, "callInput.input.length != 32");
         uint256 param = abi.decode(callInput.input, (uint256));
         require(param == 1, "First writeStorage param should be 1");
         require(callInput.value == 0, "callInput.value != 0");
 
         callInput = callInputs[1];
-        require(callInput.target_address == address(TARGET), "callInput.target_address != target");
+        require(
+            callInput.target_address == address(TARGET),
+            "callInput.target_address != target"
+        );
         require(callInput.input.length == 32, "callInput.input.length != 32");
         param = abi.decode(callInput.input, (uint256));
         require(param == 2, "Second writeStorage param should be 2");
@@ -40,7 +55,21 @@ contract TestCallInputs is Assertion, Test {
         callInputs = ph.getCallInputs(address(TARGET), bytes4(0));
         require(callInputs.length == 0, "callInputs.length != 0");
     }
-    //TODO: validate same selector to multiple targets does not get included for wrong address
+    function testCallInputsWrongTarget() external view {
+        PhEvm.CallInputs[] memory callInputs = ph.getCallInputs(
+            address(0x000000000000000000000000000000000000dEaD),
+            Target.readStorage.selector
+        );
+        require(callInputs.length == 0, "callInputs.length != 0");
+    }
+
+    function testCallNoSelector() external view {
+        PhEvm.CallInputs[] memory callInputs = ph.getCallInputs(
+            address(TARGET),
+            bytes4(0)
+        );
+        require(callInputs.length == 0, "callInputs.length != 0");
+    }
 
     function triggers() external view override {
         registerCallTrigger(this.testGetCallInputs.selector);
@@ -52,5 +81,15 @@ contract TriggeringTx {
         TARGET.writeStorage(1);
         TARGET.writeStorage(2);
         TARGET.readStorage();
+
+        address fallbackTarget = address(0x000000000000000000000000000000000000dEaD);
+        (bool success, ) = fallbackTarget.call(
+            abi.encodeWithSelector(TARGET.readStorage.selector)
+        );
+        require(success, "call failed");
+
+        (success, ) = address(TARGET).call("");
+        require(success, "call failed");
     }
 }
+

src/test-cases/precompiles/Forking.sol

@@ -44,7 +44,13 @@ contract TestForking is Assertion, Test {
             "post state newTarget.code.length should not be 0"
         );
     }
-    //TODO: add balance check on switching
+    function testForkSwitchBalance() external {
+        require(address(TARGET).balance == 1000, "balance != 1000");
+        ph.forkPreState();
+        require(address(TARGET).balance == 0, "balance != 0");
+        ph.forkPostState();
+        require(address(TARGET).balance == 1000, "balance != 1000");
+    }
 
     function testPersistTargetContracts() external {
         require(someInitValue == 1, "someInitValue != 1");
@@ -74,4 +80,12 @@ contract TestForking is Assertion, Test {
         registerCallTrigger(this.testForkSwitchNewDeployedContract.selector);
         registerCallTrigger(this.testPersistTargetContracts.selector);
     }
-}
+}
+
+contract TriggeringTx {
+    constructor() payable {
+        TARGET.incrementStorage();
+        (bool success, ) = address(TARGET).call{value: 1000}("");
+        require(success, "call failed");
+    }
+}