Test Deposits Withdrawals Consolidations

Documentation for tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py.

Generate fixtures for these test cases for Prague with:

Prague only:

fill -v tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py --fork=Prague --evm-bin=/path/to/evm-tool-dev-version

For all forks up to and including Prague:

fill -v tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py --until=Prague --evm-bin=/path/to/evm-tool-dev-version

Tests EIP-7685: General purpose execution layer requests

Cross testing for withdrawal and deposit request for EIP-7685: General purpose execution layer requests

test_valid_deposit_withdrawal_consolidation_requests(blockchain_test, pre, blocks)

Test making a deposit to the beacon chain deposit contract and a withdrawal in the same block.

Source code in tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py

@pytest.mark.parametrize(
    "requests",
    [
        *get_eoa_permutations(),
        *get_contract_permutations(),
        pytest.param(
            [
                single_deposit_from_eoa(0),
                single_withdrawal_from_eoa(0),
                single_deposit_from_contract(1),
            ],
            id="deposit_from_eoa+withdrawal_from_eoa+deposit_from_contract",
        ),
        pytest.param(
            [
                single_withdrawal_from_eoa(0),
                single_deposit_from_eoa(0),
                single_withdrawal_from_contract(1),
            ],
            id="withdrawal_from_eoa+deposit_from_eoa+withdrawal_from_contract",
        ),
        pytest.param(
            [
                single_deposit_from_eoa(0),
                single_consolidation_from_eoa(0),
                single_deposit_from_contract(1),
            ],
            id="deposit_from_eoa+consolidation_from_eoa+deposit_from_contract",
        ),
        pytest.param(
            [
                single_consolidation_from_eoa(0),
                single_deposit_from_eoa(0),
                single_consolidation_from_contract(1),
            ],
            marks=pytest.mark.skip("Only one consolidation request is allowed per block"),
            id="consolidation_from_eoa+deposit_from_eoa+consolidation_from_contract",
        ),
        pytest.param(
            [
                single_consolidation_from_eoa(0),
                single_withdrawal_from_eoa(0),
                single_consolidation_from_contract(1),
            ],
            marks=pytest.mark.skip("Only one consolidation request is allowed per block"),
            id="consolidation_from_eoa+withdrawal_from_eoa+consolidation_from_contract",
        ),
        pytest.param(
            [
                single_withdrawal_from_eoa(0),
                single_consolidation_from_eoa(0),
                single_withdrawal_from_contract(1),
            ],
            id="withdrawal_from_eoa+consolidation_from_eoa+withdrawal_from_contract",
        ),
    ],
)
def test_valid_deposit_withdrawal_consolidation_requests(
    blockchain_test: BlockchainTestFiller,
    pre: Alloc,
    blocks: List[Block],
):
    """
    Test making a deposit to the beacon chain deposit contract and a withdrawal in the same block.
    """
    blockchain_test(
        genesis_environment=Environment(),
        pre=pre,
        post={},
        blocks=blocks,
    )

test_valid_deposit_withdrawal_consolidation_request_from_same_tx(blockchain_test, pre, requests)

Test making a deposit to the beacon chain deposit contract and a withdrawal in the same tx.

Source code in tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py

@pytest.mark.parametrize("requests", [*get_permutations()])
def test_valid_deposit_withdrawal_consolidation_request_from_same_tx(
    blockchain_test: BlockchainTestFiller,
    pre: Alloc,
    requests: List[DepositRequest | WithdrawalRequest | ConsolidationRequest],
):
    """
    Test making a deposit to the beacon chain deposit contract and a withdrawal in the same tx.
    """
    withdrawal_request_fee = 1
    consolidation_request_fee = 1

    calldata = b""
    contract_code = Bytecode()
    total_value = 0
    storage = Storage()

    for request in requests:
        calldata_start = len(calldata)
        current_calldata = request.calldata
        calldata += current_calldata

        contract_code += Op.CALLDATACOPY(0, calldata_start, len(current_calldata))

        call_contract_address = 0
        value = 0
        if isinstance(request, DepositRequest):
            call_contract_address = Spec_EIP6110.DEPOSIT_CONTRACT_ADDRESS
            value = request.value
        elif isinstance(request, WithdrawalRequest):
            call_contract_address = Spec_EIP7002.WITHDRAWAL_REQUEST_PREDEPLOY_ADDRESS
            value = withdrawal_request_fee
        elif isinstance(request, ConsolidationRequest):
            call_contract_address = Spec_EIP7251.CONSOLIDATION_REQUEST_PREDEPLOY_ADDRESS
            value = consolidation_request_fee

        total_value += value

        contract_code += Op.SSTORE(
            storage.store_next(1),
            Op.CALL(
                address=call_contract_address,
                value=value,
                args_offset=0,
                args_size=len(current_calldata),
            ),
        )

    sender = pre.fund_eoa()
    contract_address = pre.deploy_contract(
        code=contract_code,
    )

    tx = Transaction(
        gas_limit=10_000_000,
        to=contract_address,
        value=total_value,
        data=calldata,
        sender=sender,
    )

    blockchain_test(
        genesis_environment=Environment(),
        pre=pre,
        post={
            contract_address: Account(
                storage=storage,
            )
        },
        blocks=[
            Block(
                txs=[tx],
                header_verify=Header(
                    requests_root=[
                        request.with_source_address(contract_address)
                        for request in sorted(requests, key=lambda r: r.type_byte())
                    ]
                ),
            )
        ],
    )

test_invalid_deposit_withdrawal_consolidation_requests(blockchain_test, pre, blocks)

Negative testing for deposits and withdrawals in the same block.

Source code in tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py

@pytest.mark.parametrize(
    "requests,block_body_override_requests,exception",
    [
        pytest.param(
            [
                single_withdrawal_from_eoa(0),
                single_deposit_from_eoa(0),
            ],
            [
                single_withdrawal(0).with_source_address(TestAddress),
                single_deposit(0),
            ],
            # TODO: on the Engine API, the issue should be detected as an invalid block hash
            BlockException.INVALID_REQUESTS,
            id="single_withdrawal_single_deposit_incorrect_order",
        ),
        pytest.param(
            [
                single_consolidation_from_eoa(0),
                single_deposit_from_eoa(0),
            ],
            [
                single_consolidation(0).with_source_address(TestAddress),
                single_deposit(0),
            ],
            # TODO: on the Engine API, the issue should be detected as an invalid block hash
            BlockException.INVALID_REQUESTS,
            id="single_consolidation_single_deposit_incorrect_order",
        ),
        pytest.param(
            [
                single_consolidation_from_eoa(0),
                single_withdrawal_from_eoa(0),
            ],
            [
                single_consolidation(0).with_source_address(TestAddress),
                single_withdrawal(0).with_source_address(TestAddress2),
            ],
            # TODO: on the Engine API, the issue should be detected as an invalid block hash
            BlockException.INVALID_REQUESTS,
            id="single_consolidation_single_withdrawal_incorrect_order",
        ),
    ],
)
def test_invalid_deposit_withdrawal_consolidation_requests(
    blockchain_test: BlockchainTestFiller,
    pre: Alloc,
    blocks: List[Block],
):
    """
    Negative testing for deposits and withdrawals in the same block.
    """
    blockchain_test(
        genesis_environment=Environment(),
        pre=pre,
        post={},
        blocks=blocks,
    )