Skip to content

test_set_code_address_and_authority_warm_state_call_types()

Documentation for tests/prague/eip7702_set_code_tx/test_set_code_txs.py::test_set_code_address_and_authority_warm_state_call_types@892e6d1e.

Generate fixtures for these test cases for Amsterdam with:

fill -v tests/prague/eip7702_set_code_tx/test_set_code_txs.py::test_set_code_address_and_authority_warm_state_call_types --fork Amsterdam

Test set to code address and authority warm status after a call to authority address, or vice-versa, using all available call opcodes without using GAS opcode.

Source code in tests/prague/eip7702_set_code_tx/test_set_code_txs.py
1574
1575
1576
1577
1578
1579
1580
1581
1582
1583
1584
1585
1586
1587
1588
1589
1590
1591
1592
1593
1594
1595
1596
1597
1598
1599
1600
1601
1602
1603
1604
1605
1606
1607
1608
1609
1610
1611
1612
1613
1614
1615
1616
1617
1618
1619
1620
1621
1622
1623
1624
1625
1626
1627
1628
1629
1630
1631
1632
1633
1634
1635
1636
1637
1638
1639
1640
1641
1642
1643
1644
1645
1646
1647
1648
@pytest.mark.with_all_call_opcodes()
@pytest.mark.parametrize(
    "set_code_address_first",
    [
        pytest.param(True, id="call_set_code_address_first_then_authority"),
        pytest.param(False, id="call_authority_first_then_set_code_address"),
    ],
)
def test_set_code_address_and_authority_warm_state_call_types(
    state_test: StateTestFiller,
    pre: Alloc,
    call_opcode: Op,
    set_code_address_first: bool,
) -> None:
    """
    Test set to code address and authority warm status after a call to
    authority address, or vice-versa, using all available call opcodes without
    using `GAS` opcode.
    """
    auth_signer = pre.fund_eoa(auth_account_start_balance)

    slot = count(1)
    slot_call_return_code = next(slot)
    slot_call_success = next(slot)

    set_code = Op.STOP
    set_code_to_address = pre.deploy_contract(set_code)

    call_set_code_to_address = Op.SSTORE(
        slot_call_return_code, call_opcode(address=set_code_to_address)
    )
    call_authority_address = Op.SSTORE(
        slot_call_return_code, call_opcode(address=auth_signer)
    )

    callee_code = Bytecode()
    if set_code_address_first:
        callee_code += call_set_code_to_address + call_authority_address
    else:
        callee_code += call_authority_address + call_set_code_to_address
    callee_code += Op.SSTORE(slot_call_success, 1) + Op.STOP

    callee_address = pre.deploy_contract(callee_code)
    callee_storage = Storage()
    callee_storage[slot_call_return_code] = call_return_code(
        opcode=call_opcode, success=True
    )
    callee_storage[slot_call_success] = 1

    tx = Transaction(
        gas_limit=1_000_000,
        to=callee_address,
        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={
            callee_address: Account(storage=callee_storage),
            auth_signer: Account(
                nonce=1,
                code=Spec.delegation_designation(set_code_to_address),
                balance=auth_account_start_balance,
            ),
        },
    )

Parametrized Test Cases

This test generates 8 parametrized test cases across 3 forks.