feat: add eip7702 test case for clz opcode

Author: LouisTsai-Csie

tests/osaka/eip7939_count_leading_zeros/test_count_leading_zeros.py

@@ -9,14 +9,18 @@
 from ethereum_test_tools import (
     Account,
     Alloc,
+    AuthorizationTuple,
     Block,
     BlockchainTestFiller,
     CodeGasMeasure,
+    Environment,
     StateTestFiller,
+    Storage,
     Transaction,
 )
 from ethereum_test_tools.vm.opcode import Opcodes as Op
 
+from ...prague.eip7702_set_code_tx.spec import Spec as Spec7702
 from .spec import Spec, ref_spec_7939
 
 REFERENCE_SPEC_GIT_PATH = ref_spec_7939.git_path
@@ -222,3 +226,50 @@ def test_clz_fork_transition(blockchain_test: BlockchainTestFiller, pre: Alloc):
             ),
         },
     )
+
+
+auth_account_start_balance = 0
+
+
+@pytest.mark.valid_from("Osaka")
+@pytest.mark.parametrize("bits", [0, 1, 128, 255])
+def test_clz_from_set_code(
+    state_test: StateTestFiller,
+    pre: Alloc,
+    bits: int,
+):
+    """Test the address opcode in a set-code transaction."""
+    storage = Storage()
+    auth_signer = pre.fund_eoa(auth_account_start_balance)
+
+    expected_clz = 255 - bits
+    set_code = Op.SSTORE(storage.store_next(expected_clz), Op.CLZ(1 << bits)) + Op.STOP
+    set_code_to_address = pre.deploy_contract(set_code)
+
+    tx = Transaction(
+        gas_limit=10_000_000,
+        to=auth_signer,
+        value=0,
+        authorization_list=[
+            AuthorizationTuple(
+                address=set_code_to_address,
+                nonce=0,
+                signer=auth_signer,
+            ),
+        ],
+        sender=pre.fund_eoa(),
+    )
+
+    state_test(
+        env=Environment(),
+        pre=pre,
+        tx=tx,
+        post={
+            set_code_to_address: Account(storage={}),
+            auth_signer: Account(
+                nonce=1,
+                code=Spec7702.delegation_designation(set_code_to_address),
+                storage=storage,
+            ),
+        },
+    )