test_balance_within_block()¶

def test_balance_within_block(
    blockchain_test: BlockchainTestFiller, pre: Alloc
) -> None:
    """
    Test withdrawal balance increase within the same block in a contract call.
    """
    save_balance_on_block_number = Op.SSTORE(
        Op.NUMBER,
        Op.BALANCE(Op.CALLDATALOAD(0)),
    )
    sender = pre.fund_eoa()
    recipient = pre.fund_eoa(ONE_GWEI)
    contract_address = pre.deploy_contract(save_balance_on_block_number)

    blocks = [
        Block(
            txs=[
                Transaction(
                    sender=sender,
                    to=contract_address,
                    data=Hash(recipient, left_padding=True),
                )
            ],
            withdrawals=[
                Withdrawal(
                    index=0,
                    validator_index=0,
                    address=recipient,
                    amount=1,
                )
            ],
        ),
        Block(
            txs=[
                Transaction(
                    sender=sender,
                    to=contract_address,
                    data=Hash(recipient, left_padding=True),
                )
            ]
        ),
    ]

    post = {
        contract_address: Account(
            storage={
                1: ONE_GWEI,
                2: 2 * ONE_GWEI,
            }
        )
    }

    blockchain_test(pre=pre, post=post, blocks=blocks)