added check whether RPC supports state overwrites

Author: simone1999

IceCreamSwapWeb3/Web3Advanced.py

@@ -76,6 +76,7 @@ def __init__(
         self.revert_reason_available: bool = self._check_revert_reason_available()
         if not self.revert_reason_available:
             print(f"RPC {self.node_url} does not return revert reasons")
+        self.overwrites_available: bool = self._check_overwrites_available()
 
         self.middleware_onion.inject(BatchRetryMiddleware, layer=0, name="batch_retry")  # split and retry batch requests
 
@@ -142,3 +143,26 @@ def _check_revert_reason_available(self):
             if not isinstance(e, ContractLogicError):
                 return False
             return e.message == "execution reverted: abc"
+
+    def _check_overwrites_available(self) -> bool:
+        with files("IceCreamSwapWeb3").joinpath("./abi/OverwriteTester.abi").open('r') as f:
+            overwrite_tester_abi = f.read()
+        with files("IceCreamSwapWeb3").joinpath("./bytecode/OverwriteTesterRuntime.bytecode").open('r') as f:
+            overwrite_tester_bytecode = f.read()
+
+        test_address = to_checksum_address("0x1234567800000000000000000000000000000001")
+        test_value = 1234
+        overwrite_tester_contract = self.eth.contract(abi=overwrite_tester_abi, address=test_address)
+        try:
+            response = overwrite_tester_contract.functions.getSlot0().call(state_override={
+                test_address: {
+                    "code": overwrite_tester_bytecode,
+                    "stateDiff": {
+                        "0x" + "00" * 32: "0x" + hex(test_value)[2:].rjust(64, "0")
+                    }
+                }
+            })
+        except Exception as e:
+            print(f"RPC does not support state overwrites, got: {repr(e)}")
+            return False
+        return response == test_value

IceCreamSwapWeb3/abi/OverwriteTester.abi

@@ -0,0 +1,15 @@
+[
+	{
+		"inputs": [],
+		"name": "getSlot0",
+		"outputs": [
+			{
+				"internalType": "uint256",
+				"name": "",
+				"type": "uint256"
+			}
+		],
+		"stateMutability": "view",
+		"type": "function"
+	}
+]

IceCreamSwapWeb3/bytecode/OverwriteTesterRuntime.bytecode

@@ -0,0 +1 @@
+0x6080604052348015600e575f80fd5b50600436106026575f3560e01c8063c9ae5e0c14602a575b5f80fd5b60306044565b604051603b91906062565b60405180910390f35b5f8054905090565b5f819050919050565b605c81604c565b82525050565b5f60208201905060735f8301846055565b9291505056fea2646970667358221220dc7c290802ba6d0d442ef59ce6a5d494b010188b015fe5eaba912a894e5b8f4c64736f6c634300081a0033