feat(tests/zkevm): add BLOBHASH test (#1666)

* zkevm: add BLOBHASH benchs

Signed-off-by: Ignacio Hagopian <[email protected]>

* generalize params

Signed-off-by: Ignacio Hagopian <[email protected]>

* improvements

Signed-off-by: Ignacio Hagopian <[email protected]>

---------

Signed-off-by: Ignacio Hagopian <[email protected]>

Author: jsign

tests/zkevm/test_worst_compute.py

@@ -14,6 +14,7 @@
 
 from ethereum_test_base_types.base_types import Bytes
 from ethereum_test_forks import Fork
+from ethereum_test_specs.tests.test_fixtures import TransactionType
 from ethereum_test_tools import (
     Address,
     Alloc,
@@ -23,6 +24,7 @@
     Environment,
     StateTestFiller,
     Transaction,
+    add_kzg_version,
 )
 from ethereum_test_tools.code.generators import While
 from ethereum_test_tools.vm.opcode import Opcodes as Op
@@ -823,6 +825,62 @@ def select_shift_amount(shift_fn, v):
     )
 
 
+@pytest.mark.valid_from("Cancun")
+@pytest.mark.parametrize(
+    "blob_index, blobs_present",
+    [
+        pytest.param(0, 0, id="no blobs"),
+        pytest.param(0, 1, id="one blob and accessed"),
+        pytest.param(1, 1, id="one blob but access non-existent index"),
+        pytest.param(5, 6, id="six blobs, access latest"),
+    ],
+)
+def test_worst_blobhash(
+    fork: Fork,
+    state_test: StateTestFiller,
+    pre: Alloc,
+    blob_index: int,
+    blobs_present: bool,
+):
+    """Test running a block with as many BLOBHASH instructions as possible."""
+    env = Environment()
+
+    code_prefix = Op.PUSH1(blob_index) + Op.JUMPDEST
+    code_suffix = Op.JUMP(len(code_prefix) - 1)
+    loop_iter = Op.POP(Op.BLOBHASH(Op.DUP1))
+    code_body_len = (MAX_CODE_SIZE - len(code_prefix) - len(code_suffix)) // len(loop_iter)
+    code_body = loop_iter * code_body_len
+    code = code_prefix + code_body + code_suffix
+    assert len(code) <= MAX_CODE_SIZE
+
+    tx_type = TransactionType.LEGACY
+    blob_versioned_hashes = None
+    max_fee_per_blob_gas = None
+    if blobs_present > 0:
+        tx_type = TransactionType.BLOB_TRANSACTION
+        max_fee_per_blob_gas = fork.min_base_fee_per_blob_gas()
+        blob_versioned_hashes = add_kzg_version(
+            [i.to_bytes() * 32 for i in range(blobs_present)],
+            BlobsSpec.BLOB_COMMITMENT_VERSION_KZG,
+        )
+
+    tx = Transaction(
+        ty=tx_type,
+        to=pre.deploy_contract(code=code),
+        gas_limit=env.gas_limit,
+        max_fee_per_blob_gas=max_fee_per_blob_gas,
+        blob_versioned_hashes=blob_versioned_hashes,
+        sender=pre.fund_eoa(),
+    )
+
+    state_test(
+        env=env,
+        pre=pre,
+        post={},
+        tx=tx,
+    )
+
+
 @pytest.mark.valid_from("Cancun")
 @pytest.mark.parametrize("mod_bits", [255, 191, 127, 63])
 @pytest.mark.parametrize("op", [Op.MOD, Op.SMOD])