refactor: update worst opcode scenario

Author: LouisTsai-Csie

tests/benchmark/test_worst_opcode.py

@@ -7,18 +7,15 @@
 
 import pytest
 
-from ethereum_test_forks import Fork
+from ethereum_test_benchmark.benchmark_code_generator import JumpLoopGenerator
 from ethereum_test_tools import (
     Alloc,
+    BenchmarkTestFiller,
     Bytecode,
-    StateTestFiller,
-    Transaction,
 )
 from ethereum_test_tools.vm.opcode import Opcodes as Op
 from ethereum_test_vm.opcode import Opcode
 
-from .helpers import code_loop_precompile_call
-
 
 @pytest.mark.parametrize(
     "opcode",
@@ -43,30 +40,26 @@
 )
 @pytest.mark.parametrize("fixed_offset", [True, False])
 def test_worst_log_opcodes(
-    state_test: StateTestFiller,
+    benchmark_test: BenchmarkTestFiller,
     pre: Alloc,
-    fork: Fork,
     opcode: Opcode,
     zeros_topic: bool,
     size: int,
     fixed_offset: bool,
     non_zero_data: bool,
-    gas_benchmark_value: int,
 ):
     """Test running a block with as many LOG opcodes as possible."""
-    max_code_size = fork.max_code_size()
-
-    calldata = Bytecode()
+    setup = Bytecode()
 
     # For non-zero data, load  into memory.
     if non_zero_data:
-        calldata += Op.CODECOPY(dest_offset=0, offset=0, size=Op.CODESIZE)
+        setup += Op.CODECOPY(dest_offset=0, offset=0, size=Op.CODESIZE)
 
     # Push the size value onto the stack and access it using the DUP opcode.
-    calldata += Op.PUSH3(size)
+    setup += Op.PUSH3(size)
 
     # For non-zeros topic, push a non-zero value for topic.
-    calldata += Op.PUSH0 if zeros_topic else Op.PUSH32(2**256 - 1)
+    setup += Op.PUSH0 if zeros_topic else Op.PUSH32(2**256 - 1)
 
     topic_count = len(opcode.kwargs or []) - 2
     offset = Op.PUSH0 if fixed_offset else Op.MOD(Op.GAS, 7)
@@ -75,21 +68,10 @@ def test_worst_log_opcodes(
     # 0 topics -> DUP1, 1 topic -> DUP2, N topics -> DUP(N+1)
     size_op = getattr(Op, f"DUP{topic_count + 2}")
 
-    code_sequence = Op.DUP1 * topic_count + size_op + offset + opcode
-
-    code = code_loop_precompile_call(calldata, code_sequence, fork)
-    assert len(code) <= max_code_size
-
-    code_address = pre.deploy_contract(code=code)
-
-    tx = Transaction(
-        to=code_address,
-        gas_limit=gas_benchmark_value,
-        sender=pre.fund_eoa(),
-    )
+    attack_block = Op.DUP1 * topic_count + size_op + offset + opcode
 
-    state_test(
+    benchmark_test(
         pre=pre,
         post={},
-        tx=tx,
+        code_generator=JumpLoopGenerator(setup=setup, attack_block=attack_block),
     )