new(tests): added more cases (two contract req, one EOA req) in test_valid_deposit_withdrawal_consolidation_requests

Author: felix314159

tests/prague/eip7685_general_purpose_el_requests/test_deposits_withdrawals_consolidations.py

@@ -5,7 +5,7 @@
 """  # noqa: E501
 
 from itertools import permutations
-from typing import Any, Dict, Generator, List, Tuple
+from typing import Dict, Generator, List, Tuple
 
 import pytest
 
@@ -26,6 +26,7 @@
     Transaction,
 )
 from ethereum_test_tools import Opcodes as Op
+from ethereum_test_tools.utility.pytest import ParameterSet
 
 from ..eip6110_deposits.helpers import DepositContract, DepositRequest, DepositTransaction
 from ..eip6110_deposits.spec import Spec as Spec_EIP6110
@@ -99,7 +100,13 @@ def single_consolidation_from_contract(i: int) -> ConsolidationRequestContract:
     return ConsolidationRequestContract(requests=[single_consolidation(i)])
 
 
-def get_permutations(n: int = 3) -> Generator[Any, None, None]:
+def get_permutations(
+    n: int = 3,
+) -> Generator[
+    ParameterSet,
+    None,
+    None,
+]:
     """Return possible permutations of the requests from an EOA."""
     requests = [
         (
@@ -119,7 +126,7 @@ def get_permutations(n: int = 3) -> Generator[Any, None, None]:
         yield pytest.param([p[1] for p in perm], id="+".join([p[0] for p in perm]))
 
 
-def get_eoa_permutations(n: int = 3) -> Generator[Any, None, None]:
+def get_eoa_permutations(n: int = 3) -> Generator[ParameterSet, None, None]:
     """Return possible permutations of the requests from an EOA."""
     requests = [
         (
@@ -139,7 +146,7 @@ def get_eoa_permutations(n: int = 3) -> Generator[Any, None, None]:
         yield pytest.param([p[1] for p in perm], id="+".join([p[0] for p in perm]))
 
 
-def get_contract_permutations(n: int = 3) -> Generator[Any, None, None]:
+def get_contract_permutations(n: int = 3) -> Generator[ParameterSet, None, None]:
     """Return possible permutations of the requests from a contract."""
     requests = [
         (
@@ -172,14 +179,6 @@ def get_contract_permutations(n: int = 3) -> Generator[Any, None, None]:
             ],
             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),
@@ -214,10 +213,127 @@ def get_contract_permutations(n: int = 3) -> Generator[Any, None, None]:
             ],
             id="withdrawal_from_eoa+consolidation_from_eoa+withdrawal_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(
             [],
             id="empty_requests",
         ),
+        ###################### contract: consolidation + withdrawal
+        pytest.param(
+            [
+                single_withdrawal_from_eoa(0),
+                single_consolidation_from_contract(0),
+                single_withdrawal_from_contract(1),
+            ],
+            id="withdrawal_from_eoa+consolidation_from_contract+withdrawal_from_contract",
+        ),
+        pytest.param(
+            [
+                single_deposit_from_eoa(0),
+                single_consolidation_from_contract(0),
+                single_withdrawal_from_contract(0),
+            ],
+            id="deposit_from_eoa+consolidation_from_contract+withdrawal_from_contract",
+        ),
+        pytest.param(
+            [
+                single_consolidation_from_eoa(0),
+                single_consolidation_from_contract(1),
+                single_withdrawal_from_contract(0),
+            ],
+            id="consolidation_from_eoa+consolidation_from_contract+withdrawal_from_contract",
+        ),
+        ###################### contract: consolidation + deposit
+        pytest.param(
+            [
+                single_withdrawal_from_eoa(0),
+                single_consolidation_from_contract(0),
+                single_deposit_from_contract(0),
+            ],
+            id="withdrawal_from_eoa+consolidation_from_contract+deposit_from_contract",
+        ),
+        pytest.param(
+            [
+                single_deposit_from_eoa(0),
+                single_consolidation_from_contract(0),
+                single_deposit_from_contract(1),
+            ],
+            id="deposit_from_eoa+consolidation_from_contract+deposit_from_contract",
+        ),
+        pytest.param(
+            [
+                single_consolidation_from_eoa(0),
+                single_consolidation_from_contract(1),
+                single_deposit_from_contract(0),
+            ],
+            id="consolidation_from_eoa+consolidation_from_contract+deposit_from_contract",
+        ),
+        ###################### contract: withdrawal + deposit
+        pytest.param(
+            [
+                single_withdrawal_from_eoa(0),
+                single_withdrawal_from_contract(1),
+                single_deposit_from_contract(0),
+            ],
+            id="withdrawal_from_eoa+withdrawal_from_contract+deposit_from_contract",
+        ),
+        pytest.param(
+            [
+                single_deposit_from_eoa(0),
+                single_withdrawal_from_contract(0),
+                single_deposit_from_contract(1),
+            ],
+            id="deposit_from_eoa+withdrawal_from_contract+deposit_from_contract",
+        ),
+        pytest.param(
+            [
+                single_consolidation_from_eoa(0),
+                single_withdrawal_from_contract(0),
+                single_deposit_from_contract(0),
+            ],
+            id="consolidation_from_eoa+withdrawal_from_contract+deposit_from_contract",
+        ),
+        # testing upper limits of each request type per slot if it exists
+        pytest.param(
+            [
+                single_consolidation_from_contract(0),
+                single_consolidation_from_contract(1),
+                # single_consolidation_from_contract(2),  # i=2 is not allowed cuz only 2 MAX CONSOLIDATIONS PER BLOCK (EIP-7251), but why does the error message not inform about this? only reports hash mismatch  # noqa: E501
+                #
+                # the following performs single_withdrawal_from_contract(0) to (16)
+                *[
+                    single_withdrawal_from_contract(i)
+                    for i in range(
+                        0,
+                        16,  # ensure that the hard limit of deposits per epoch has been removed in all clients (EIP-6610), prev limit was 16 deposits/slot or 512 deposits/epoch
+                    )  # when using larger numbers: it fails around 1000 due to: UnixHTTPConnectionPool(host='localhost', port=None): Read timed out. (read timeout=20), workaround: in transition_tool.py use SLOW_REQUEST_TIMEOUT (60 sec), so slow=True
+                ],
+                # single_withdrawal_from_contract(16), # i=16 not allowed cuz only 16 MAX WITHDRAWALS PER BLOCK (EIP-7002)
+                #
+                # the following performs single_deposit_from_contract(0) to (18)
+                *[
+                    single_deposit_from_contract(i)
+                    for i in range(
+                        0,
+                        18,  # ensure that the hard limit of deposits per epoch has been removed in all clients (EIP-6610), prev limit was 16 deposits/slot or 512 deposits/epoch
+                    )  # when using larger numbers: it fails around 1000 due to: UnixHTTPConnectionPool(host='localhost', port=None): Read timed out. (read timeout=20), workaround: in transition_tool.py use SLOW_REQUEST_TIMEOUT (60 sec), so slow=True
+                ],
+            ],
+            # following ID not possible to due filename length limitations
+            # id=(
+            #     ("consolidation_from_contract+" * 2)
+            #     + ("withdrawal_from_contract+" * 16)
+            #     + ("deposit_from_contract+" * 18)
+            # )[:-1],  # remove last '+'
+            id="max_withdrawals_per_slot+max_consolidations_per_slot+unlimited_deposits_per_slot",
+        ),
     ],
 )
 def test_valid_deposit_withdrawal_consolidation_requests(
@@ -324,7 +440,7 @@ def test_valid_deposit_withdrawal_consolidation_request_from_same_tx(
     )
 
 
-def invalid_requests_block_combinations(fork: Fork) -> List[Any]:
+def invalid_requests_block_combinations(fork: Fork) -> List[ParameterSet]:
     """
     Return a list of invalid request combinations for the given fork.