Exhaustive check fixes

src/ethereum_test_forks/forks/forks.py

@@ -404,7 +404,9 @@ def call_opcodes(
         """
         At Homestead, DELEGATECALL opcode was introduced.
         """
-        return [(Opcodes.DELEGATECALL, EVMCodeType.LEGACY),] + super(
+        return [
+            (Opcodes.DELEGATECALL, EVMCodeType.LEGACY),
+        ] + super(
             Homestead, cls
         ).call_opcodes(block_number, timestamp)
 
@@ -449,7 +451,9 @@ def call_opcodes(
         """
         At Byzantium, STATICCALL opcode was introduced.
         """
-        return [(Opcodes.STATICCALL, EVMCodeType.LEGACY),] + super(
+        return [
+            (Opcodes.STATICCALL, EVMCodeType.LEGACY),
+        ] + super(
             Byzantium, cls
         ).call_opcodes(block_number, timestamp)
 
@@ -483,7 +487,9 @@ def create_opcodes(
         """
         At Constantinople, `CREATE2` opcode is added.
         """
-        return [(Opcodes.CREATE2, EVMCodeType.LEGACY),] + super(
+        return [
+            (Opcodes.CREATE2, EVMCodeType.LEGACY),
+        ] + super(
             Constantinople, cls
         ).create_opcodes(block_number, timestamp)
 
@@ -974,7 +980,7 @@ def pre_allocation_blockchain(cls) -> Mapping:
         type tests.
         """
         new_allocation = {
-            0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFE: {
+            Address(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFE): {
                 "nonce": 1,
                 "code": (
                     "0x60203611603157600143035f35116029575f35612000014311602957612000"
@@ -1020,7 +1026,10 @@ def evm_code_types(cls, block_number: int = 0, timestamp: int = 0) -> List[EVMCo
         """
         EOF V1 is supported starting from this fork.
         """
-        return super(CancunEIP7692, cls,).evm_code_types(  # noqa: SC200
+        return super(
+            CancunEIP7692,
+            cls,
+        ).evm_code_types(  # noqa: SC200
             block_number,
             timestamp,
         ) + [EVMCodeType.EOF_V1]
@@ -1049,7 +1058,9 @@ def create_opcodes(
         """
         EOF V1 introduces `EOFCREATE`.
         """
-        return [(Opcodes.EOFCREATE, EVMCodeType.EOF_V1),] + super(
+        return [
+            (Opcodes.EOFCREATE, EVMCodeType.EOF_V1),
+        ] + super(
             CancunEIP7692, cls  # noqa: SC200
         ).create_opcodes(block_number, timestamp)
 
@@ -1084,7 +1095,10 @@ def evm_code_types(cls, block_number: int = 0, timestamp: int = 0) -> List[EVMCo
         """
         EOF V1 is supported starting from this fork.
         """
-        return super(PragueEIP7692, cls,).evm_code_types(  # noqa: SC200
+        return super(
+            PragueEIP7692,
+            cls,
+        ).evm_code_types(  # noqa: SC200
             block_number,
             timestamp,
         ) + [EVMCodeType.EOF_V1]

tests/verkle/eip4762_verkle_gas_witness/test_balance.py

@@ -14,7 +14,6 @@
     Block,
     BlockchainTestFiller,
     Environment,
-    Hash,
     TestAddress,
     TestAddress2,
     Transaction,
@@ -23,16 +22,14 @@
 from ethereum_test_tools.vm.opcode import Opcodes as Op
 from ethereum_test_types.verkle.helpers import chunkify_code
 
-# TODO(verkle): Update reference spec version
 REFERENCE_SPEC_GIT_PATH = "EIPS/eip-4762.md"
 REFERENCE_SPEC_VERSION = "2f8299df31bb8173618901a03a8366a3183479b0"
 
 precompile_address = Address("0x04")
-system_contract_address = Address("0x000F3df6D732807Ef1319fB7B8bB8522d0Beac02")
+system_contract_address = Address("0xfffffffffffffffffffffffffffffffffffffffe")
 example_address = Address("0xd94f5374fce5edbc8e2a8697c15331677e6ebf0c")
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.parametrize(
     "target",
@@ -50,7 +47,6 @@ def test_balance(blockchain_test: BlockchainTestFiller, fork: str, target, warm)
     _balance(blockchain_test, fork, target, [target], warm=warm)
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.parametrize("target", [example_address, precompile_address])
 def test_balance_insufficient_gas(blockchain_test: BlockchainTestFiller, fork: str, target):

tests/verkle/eip4762_verkle_gas_witness/test_calls.py

@@ -7,6 +7,7 @@
 
 import pytest
 
+from ethereum_test_forks import Verkle
 from ethereum_test_tools import (
     Account,
     Address,
@@ -21,17 +22,16 @@
     WitnessCheck,
 )
 from ethereum_test_tools.vm.opcode import Opcodes as Op
+from ethereum_test_types.verkle.helpers import chunkify_code
 
-# TODO(verkle): Update reference spec version
 REFERENCE_SPEC_GIT_PATH = "EIPS/eip-4762.md"
 REFERENCE_SPEC_VERSION = "2f8299df31bb8173618901a03a8366a3183479b0"
 
 caller_address = Address("0xd94f5374fce5edbc8e2a8697c15331677e6ebf0c")
-system_contract_address = Address("0x000F3df6D732807Ef1319fB7B8bB8522d0Beac02")
+system_contract_address = Address("0xfffffffffffffffffffffffffffffffffffffffe")
 precompile_address = Address("0x04")
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.parametrize(
     "call_instruction, value",
@@ -48,9 +48,9 @@
 @pytest.mark.parametrize(
     "target",
     [
-        TestAddress2,
+        # TestAddress2,
         precompile_address,
-        system_contract_address,
+        # system_contract_address,
     ],
 )
 def test_calls(
@@ -66,7 +66,6 @@ def test_calls(
     _generic_call(blockchain_test, call_instruction, target, value)
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.parametrize(
     "call_instruction",
@@ -84,7 +83,6 @@ def test_calls_warm(blockchain_test: BlockchainTestFiller, fork: str, call_instr
     _generic_call(blockchain_test, call_instruction, TestAddress2, 0, warm=True)
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.skip("Pending TBD gas limits")
 @pytest.mark.parametrize(
@@ -174,17 +172,20 @@ def _generic_call(
         value=tx_value,
     )
 
-    witness_check = WitnessCheck()
+    witness_check = WitnessCheck(fork=Verkle)
     for address in [TestAddress, caller_address, env.fee_recipient]:
-        witness_check.add_account_full(
-            address=address,
-            account=(None if address == env.fee_recipient else pre[address]),
-        )
-    if target != precompile_address and enough_gas_read_witness:
-        witness_check.add_account_basic_data(
-            address=target,
-            account=pre[target],
-        )
+        witness_check.add_account_full(address=address, account=pre.get(address))
+    if enough_gas_read_witness:
+        if value > 0 or (target != precompile_address and target != precompile_address):
+            witness_check.add_account_basic_data(address=target, account=pre[target])
+
+    code_chunks = chunkify_code(pre[caller_address].code)
+    for i, chunk in enumerate(code_chunks, start=0):
+        witness_check.add_code_chunk(address=caller_address, chunk_number=i, value=chunk)
+    code_chunks = chunkify_code(pre[target].code)
+    if target != precompile_address and target != system_contract_address:
+        for i, chunk in enumerate(code_chunks, start=0):
+            witness_check.add_code_chunk(address=target, chunk_number=i, value=chunk)
 
     blocks = [
         Block(
@@ -213,7 +214,6 @@ def _generic_call(
     )
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.parametrize(
     "call_instruction, gas_limit, enough_gas_account_creation",
@@ -260,7 +260,7 @@ def test_call_non_existent_account(
         gas_price=10,
     )
 
-    witness_check = WitnessCheck()
+    witness_check = WitnessCheck(fork=Verkle)
     for address in [TestAddress, caller_address, env.fee_recipient]:
         witness_check.add_account_full(
             address=address,

tests/verkle/eip4762_verkle_gas_witness/test_codecopy_ext_precompile.py

@@ -7,28 +7,30 @@
 
 import pytest
 
+from ethereum_test_forks import Verkle
 from ethereum_test_tools import (
     Account,
     Address,
     Block,
     BlockchainTestFiller,
     Environment,
-    Initcode,
     TestAddress,
+    TestAddress2,
     Transaction,
     WitnessCheck,
 )
 from ethereum_test_tools.vm.opcode import Opcodes as Op
+from ethereum_test_types.verkle.helpers import chunkify_code
+from ethereum_test_forks import Fork
+
 
-# TODO(verkle): Update reference spec version
 REFERENCE_SPEC_GIT_PATH = "EIPS/eip-4762.md"
 REFERENCE_SPEC_VERSION = "2f8299df31bb8173618901a03a8366a3183479b0"
 
 precompile_address = Address("0x04")
-system_contract_address = Address("0x000F3df6D732807Ef1319fB7B8bB8522d0Beac02")
+system_contract_address = Address("0xfffffffffffffffffffffffffffffffffffffffe")
 
 
-# TODO(verkle): update to Osaka when t8n supports the fork.
 @pytest.mark.valid_from("Verkle")
 @pytest.mark.parametrize(
     "target",
@@ -37,7 +39,7 @@
         system_contract_address,
     ],
 )
-def test_extcodecopy_precompile(blockchain_test: BlockchainTestFiller, fork: str, target):
+def test_extcodecopy_precompile(blockchain_test: BlockchainTestFiller, fork: Fork, target):
     """
     Test EXTCODECOPY targeting a precompile or system contract.
     """
@@ -50,23 +52,32 @@ def test_extcodecopy_precompile(blockchain_test: BlockchainTestFiller, fork: str
     )
     pre = {
         TestAddress: Account(balance=1000000000000000000000),
+        TestAddress2: Account(
+            balance=1000000000000000000000, code=Op.EXTCODECOPY(target, 0, 0, 10)
+        ),
     }
 
     tx = Transaction(
         ty=0x0,
         chain_id=0x01,
         nonce=0,
-        to=None,
+        to=TestAddress2,
         gas_limit=1_000_000,
         gas_price=10,
-        data=Initcode(deploy_code=Op.EXTCODECOPY(target, 0, 0, 100)),
     )
 
-    witness_check = WitnessCheck()
-    for address in [env.fee_recipient, TestAddress]:
-        witness_check.add_account_full(
-            address=address,
-            account=(None if address == env.fee_recipient else pre[address]),
+    witness_check = WitnessCheck(fork=Verkle)
+    for address in [env.fee_recipient, TestAddress, TestAddress2]:
+        witness_check.add_account_full(address=address, account=pre.get(address))
+    code_chunks = chunkify_code(pre[TestAddress2].code)
+    for i, chunk in enumerate(code_chunks, start=0):
+        witness_check.add_code_chunk(address=TestAddress2, chunk_number=i, value=chunk)
+
+    if target == system_contract_address:
+        code = Account(**fork.pre_allocation_blockchain()[system_contract_address]).code
+        code_chunks = chunkify_code(code)
+        witness_check.add_code_chunk(
+            address=system_contract_address, chunk_number=0, value=code_chunks[0]
         )
 
     blocks = [