feat: updates for payload attributes v4.

Author: spencer-tb

src/ethereum_test_forks/base_fork.py

@@ -270,6 +270,14 @@ def target_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
         """
         pass
 
+    @classmethod
+    @abstractmethod
+    def max_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
+        """
+        Returns the max blobs per block for a given fork.
+        """
+        pass
+
     @classmethod
     @abstractmethod
     def get_reward(cls, block_number: int = 0, timestamp: int = 0) -> int:
@@ -384,6 +392,26 @@ def engine_new_payload_target_blobs_per_block(
         """
         pass
 
+    @classmethod
+    @abstractmethod
+    def engine_payload_attribute_target_blobs_per_block(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> bool:
+        """
+        Returns true if the payload attributes include the target blobs per block.
+        """
+        pass
+
+    @classmethod
+    @abstractmethod
+    def engine_payload_attribute_max_blobs_per_block(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> bool:
+        """
+        Returns true if the payload attributes include the max blobs per block.
+        """
+        pass
+
     @classmethod
     @abstractmethod
     def engine_forkchoice_updated_version(

src/ethereum_test_forks/forks/forks.py

@@ -239,6 +239,13 @@ def target_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
         """
         return 0
 
+    @classmethod
+    def max_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
+        """
+        Returns the max number of blobs per block for a given fork.
+        """
+        return 0
+
     @classmethod
     def header_requests_required(cls, block_number: int, timestamp: int) -> bool:
         """
@@ -305,6 +312,24 @@ def engine_new_payload_target_blobs_per_block(
         """
         return False
 
+    @classmethod
+    def engine_payload_attribute_target_blobs_per_block(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> bool:
+        """
+        At genesis, payload attributes do not include the target blobs per block.
+        """
+        return False
+
+    @classmethod
+    def engine_payload_attribute_max_blobs_per_block(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> bool:
+        """
+        At genesis, payload attributes do not include the max blobs per block.
+        """
+        return False
+
     @classmethod
     def engine_forkchoice_updated_version(
         cls, block_number: int = 0, timestamp: int = 0
@@ -978,6 +1003,13 @@ def target_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
         """
         return 3
 
+    @classmethod
+    def max_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
+        """
+        Blobs are enabled starting from Cancun, with a static max of 6 blobs.
+        """
+        return 6
+
     @classmethod
     def tx_types(cls, block_number: int = 0, timestamp: int = 0) -> List[int]:
         """
@@ -1212,6 +1244,20 @@ def header_requests_required(cls, block_number: int, timestamp: int) -> bool:
         """
         return True
 
+    @classmethod
+    def target_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
+        """
+        Blobs are decoupled from EL at Prague, and gets a static target of 6 blobs from the CL.
+        """
+        return 6
+
+    @classmethod
+    def max_blobs_per_block(cls, block_number: int, timestamp: int) -> int:
+        """
+        Blobs are decoupled from EL at Prague, and gets static max of 9 blobs from the CL.
+        """
+        return 9
+
     @classmethod
     def header_target_blobs_per_block_required(
         cls,
@@ -1249,6 +1295,24 @@ def engine_new_payload_version(
         """
         return 4
 
+    @classmethod
+    def engine_payload_attribute_target_blobs_per_block(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> bool:
+        """
+        Starting at Prague, payload attributes include the target blobs per block.
+        """
+        return True
+
+    @classmethod
+    def engine_payload_attribute_max_blobs_per_block(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> bool:
+        """
+        Starting at Prague, payload attributes include the max blobs per block.
+        """
+        return True
+
 
 class CancunEIP7692(  # noqa: SC200
     Cancun,

src/ethereum_test_rpc/types.py

@@ -111,6 +111,8 @@ class PayloadAttributes(CamelModel):
     suggested_fee_recipient: Address
     withdrawals: List[Withdrawal] | None = None
     parent_beacon_block_root: Hash | None = None
+    target_blobs_per_block: HexNumber | None = None
+    max_blobs_per_block: HexNumber | None = None
 
 
 class BlobsBundle(CamelModel):

src/pytest_plugins/execute/rpc/hive.py

@@ -656,6 +656,16 @@ def generate_block(self: "EthRPC"):
             suggested_fee_recipient=Address(0),
             withdrawals=[] if self.fork.header_withdrawals_required() else None,
             parent_beacon_block_root=parent_beacon_block_root,
+            target_blobs_per_block=(
+                self.fork.target_blobs_per_block(0, 0)
+                if self.fork.engine_payload_attribute_target_blobs_per_block(0, 0)
+                else None
+            ),
+            max_blobs_per_block=(
+                self.fork.max_blobs_per_block(0, 0)
+                if self.fork.engine_payload_attribute_max_blobs_per_block(0, 0)
+                else None
+            ),
         )
         forkchoice_updated_version = self.fork.engine_forkchoice_updated_version()
         assert (