feat(forks): Add transaction_data_floor_cost_calculator

Author: marioevz

src/ethereum_test_forks/base_fork.py

@@ -51,6 +51,18 @@ def __call__(self, *, data: BytesConvertible, floor: bool = False) -> int:
         pass
 
 
+class TransactionDataFloorCostCalculator(Protocol):
+    """
+    A protocol to calculate the transaction floor cost due to its calldata for a given fork.
+    """
+
+    def __call__(self, *, data: BytesConvertible) -> int:
+        """
+        Returns the transaction gas cost of calldata given its contents.
+        """
+        pass
+
+
 class TransactionIntrinsicCostCalculator(Protocol):
     """
     A protocol to calculate the intrinsic gas cost of a transaction for a given fork.
@@ -247,6 +259,16 @@ def calldata_gas_calculator(
         """
         pass
 
+    @classmethod
+    @abstractmethod
+    def transaction_data_floor_cost_calculator(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> TransactionDataFloorCostCalculator:
+        """
+        Returns a callable that calculates the transaction floor cost due to its calldata.
+        """
+        pass
+
     @classmethod
     @abstractmethod
     def transaction_intrinsic_cost_calculator(

src/ethereum_test_forks/forks/forks.py

@@ -18,6 +18,7 @@
     BaseFork,
     CalldataGasCalculator,
     MemoryExpansionGasCalculator,
+    TransactionDataFloorCostCalculator,
     TransactionIntrinsicCostCalculator,
 )
 from ..gas_costs import GasCosts
@@ -197,6 +198,19 @@ def fn(*, data: BytesConvertible, floor: bool = False) -> int:
 
         return fn
 
+    @classmethod
+    def transaction_data_floor_cost_calculator(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> TransactionDataFloorCostCalculator:
+        """
+        At frontier, the transaction data floor cost is a constant zero.
+        """
+
+        def fn(*, data: BytesConvertible) -> int:
+            return 0
+
+        return fn
+
     @classmethod
     def transaction_intrinsic_cost_calculator(
         cls, block_number: int = 0, timestamp: int = 0
@@ -1185,6 +1199,21 @@ def fn(*, data: BytesConvertible, floor: bool = False) -> int:
 
         return fn
 
+    @classmethod
+    def transaction_data_floor_cost_calculator(
+        cls, block_number: int = 0, timestamp: int = 0
+    ) -> TransactionDataFloorCostCalculator:
+        """
+        Starting in Prague, due to EIP-7623, the transaction data floor cost is introduced.
+        """
+        calldata_gas_calculator = cls.calldata_gas_calculator(block_number, timestamp)
+        gas_costs = cls.gas_costs(block_number, timestamp)
+
+        def fn(*, data: BytesConvertible) -> int:
+            return calldata_gas_calculator(data=data, floor=True) + gas_costs.G_TRANSACTION
+
+        return fn
+
     @classmethod
     def transaction_intrinsic_cost_calculator(
         cls, block_number: int = 0, timestamp: int = 0
@@ -1196,6 +1225,9 @@ def transaction_intrinsic_cost_calculator(
             block_number, timestamp
         )
         gas_costs = cls.gas_costs(block_number, timestamp)
+        transaction_data_floor_cost_calculator = cls.transaction_data_floor_cost_calculator(
+            block_number, timestamp
+        )
 
         def fn(
             *,
@@ -1219,8 +1251,8 @@ def fn(
             if return_cost_deducted_prior_execution:
                 return intrinsic_cost
 
-            floor_data_cost = cls.calldata_gas_calculator()(data=calldata, floor=True)
-            return max(intrinsic_cost, floor_data_cost + gas_costs.G_TRANSACTION)
+            transaction_floor_data_cost = transaction_data_floor_cost_calculator(data=calldata)
+            return max(intrinsic_cost, transaction_floor_data_cost)
 
         return fn