Skip to content

test_bal_7002_request_from_contract()

Documentation for tests/amsterdam/eip7928_block_level_access_lists/test_block_access_lists_eip7002.py::test_bal_7002_request_from_contract@20373115.

Generate fixtures for these test cases for Amsterdam with:

fill -v tests/amsterdam/eip7928_block_level_access_lists/test_block_access_lists_eip7002.py::test_bal_7002_request_from_contract --fork Amsterdam

Ensure BAL captures withdrawal request from contract with correct source address.

Alice calls RelayContract which internally calls EIP-7002 system contract with withdrawal request. Withdrawal request should have source_address = RelayContract (not Alice).

Source code in tests/amsterdam/eip7928_block_level_access_lists/test_block_access_lists_eip7002.py
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
def test_bal_7002_request_from_contract(
    pre: Alloc,
    blockchain_test: BlockchainTestFiller,
) -> None:
    """
    Ensure BAL captures withdrawal request from contract with correct
    source address.

    Alice calls RelayContract which internally calls EIP-7002 system
    contract with withdrawal request. Withdrawal request should have
    source_address = RelayContract (not Alice).
    """
    fee = Spec7002.get_fee(0)

    # Create withdrawal request interaction using Prague helper
    interaction = WithdrawalRequestContract(
        requests=[
            WithdrawalRequest(
                validator_pubkey=0x01,
                amount=0,
                fee=fee,
            )
        ],
        contract_balance=fee,
    )

    # Set up pre-state using helper
    prepared = interaction.update_pre(pre)

    alice = prepared.sender_account
    relay_contract = prepared.contract_address

    # Build queue storage slots with contract as source
    queue_writes, queue_reads = _build_queue_storage_slots(
        [relay_contract], prepared.requests
    )

    block = Block(
        txs=prepared.transactions(),
        expected_block_access_list=BlockAccessListExpectation(
            account_expectations={
                alice: BalAccountExpectation(
                    nonce_changes=[
                        BalNonceChange(block_access_index=1, post_nonce=1)
                    ],
                ),
                relay_contract: BalAccountExpectation(
                    balance_changes=[
                        BalBalanceChange(
                            block_access_index=1,
                            post_balance=0,
                        )
                    ],
                ),
                Spec7002.WITHDRAWAL_REQUEST_PREDEPLOY_ADDRESS: BalAccountExpectation(  # noqa: E501
                    balance_changes=[
                        BalBalanceChange(
                            block_access_index=1,
                            post_balance=fee,
                        )
                    ],
                    storage_reads=[
                        Spec7002.EXCESS_WITHDRAWAL_REQUESTS_STORAGE_SLOT,
                        Spec7002.WITHDRAWAL_REQUEST_QUEUE_HEAD_STORAGE_SLOT,
                    ]
                    + queue_reads,
                    storage_changes=[
                        BalStorageSlot(
                            slot=Spec7002.WITHDRAWAL_REQUEST_COUNT_STORAGE_SLOT,
                            slot_changes=_build_incremental_changes(
                                1,
                                BalStorageChange,
                                "post_value",
                                lambda i: i,
                                reset_to=0,
                            ),
                        ),
                        BalStorageSlot(
                            slot=Spec7002.WITHDRAWAL_REQUEST_QUEUE_TAIL_STORAGE_SLOT,
                            slot_changes=_build_incremental_changes(
                                1,
                                BalStorageChange,
                                "post_value",
                                lambda i: i,
                                reset_to=0,
                            ),
                        ),
                    ]
                    + queue_writes,
                ),
            }
        ),
    )

    blockchain_test(
        pre=pre,
        blocks=[block],
        post={
            alice: Account(nonce=1),
            relay_contract: Account(balance=0),
            Spec7002.WITHDRAWAL_REQUEST_PREDEPLOY_ADDRESS: Account(
                balance=fee,
                storage=_extract_post_storage_from_queue_writes(queue_writes),
            ),
        },
    )

Parametrized Test Cases

This test generates 1 parametrized test case across 1 fork.