Implement worst case scenario for price algorithm v1

crates/fuel-gas-price-algorithm/src/lib.rs

@@ -4,3 +4,24 @@
 
 pub mod v0;
 pub mod v1;
+
+#[allow(clippy::cast_possible_truncation)]
+pub(crate) fn cumulative_percentage_change(
+    new_exec_price: u64,
+    for_height: u32,
+    percentage: u64,
+    height: u32,
+) -> u64 {
+    let blocks = height.saturating_sub(for_height) as f64;
+    let percentage_as_decimal = percentage as f64 / 100.0;
+    let multiple = (1.0f64 + percentage_as_decimal).powf(blocks);
+    let mut approx = new_exec_price as f64 * multiple;
+    // account for rounding errors and take a slightly higher value
+    const ARB_CUTOFF: f64 = 16948547188989277.0;
+    if approx > ARB_CUTOFF {
+        const ARB_ADDITION: f64 = 2000.0;
+        approx += ARB_ADDITION;
+    }
+    // `f64` over `u64::MAX` are cast to `u64::MAX`
+    approx.ceil() as u64
+}

crates/fuel-gas-price-algorithm/src/v0.rs

@@ -3,6 +3,8 @@ use std::{
     num::NonZeroU64,
 };
 
+use crate::cumulative_percentage_change;
+
 #[cfg(test)]
 mod tests;
 
@@ -27,22 +29,13 @@ impl AlgorithmV0 {
         self.new_exec_price
     }
 
-    #[allow(clippy::cast_possible_truncation)]
     pub fn worst_case(&self, height: u32) -> u64 {
-        let price = self.new_exec_price as f64;
-        let blocks = height.saturating_sub(self.for_height) as f64;
-        let percentage = self.percentage as f64;
-        let percentage_as_decimal = percentage / 100.0;
-        let multiple = (1.0f64 + percentage_as_decimal).powf(blocks);
-        let mut approx = price * multiple;
-        // account for rounding errors and take a slightly higher value
-        const ARB_CUTOFF: f64 = 16948547188989277.0;
-        if approx > ARB_CUTOFF {
-            const ARB_ADDITION: f64 = 2000.0;
-            approx += ARB_ADDITION;
-        }
-        // `f64` over `u64::MAX` are cast to `u64::MAX`
-        approx.ceil() as u64
+        cumulative_percentage_change(
+            self.new_exec_price,
+            self.for_height,
+            self.percentage,
+            height,
+        )
     }
 }
 

crates/fuel-gas-price-algorithm/src/v1.rs

@@ -4,6 +4,8 @@ use std::{
     ops::Div,
 };
 
+use crate::cumulative_percentage_change;
+
 #[cfg(test)]
 mod tests;
 
@@ -21,13 +23,37 @@ pub enum Error {
 
 #[derive(Debug, Clone, PartialEq)]
 pub struct AlgorithmV1 {
-    /// the combined Execution and DA gas prices
-    combined_gas_price: u64,
+    /// The gas price for to cover the execution of the next block
+    new_exec_price: u64,
+    /// The change percentage per block
+    exec_price_percentage: u64,
+    /// The gas price for to cover DA commitment
+    new_da_gas_price: u64,
+    /// The change percentage per block
+    da_gas_price_percentage: u64,
+    /// The block height of the next L2 block
+    for_height: u32,
 }
 
 impl AlgorithmV1 {
     pub fn calculate(&self) -> u64 {
-        self.combined_gas_price
+        self.new_exec_price.saturating_add(self.new_da_gas_price)
+    }
+
+    pub fn worst_case(&self, height: u32) -> u64 {
+        let exec = cumulative_percentage_change(
+            self.new_exec_price,
+            self.for_height,
+            self.exec_price_percentage,
+            height,
+        );
+        let da = cumulative_percentage_change(
+            self.new_da_gas_price,
+            self.for_height,
+            self.da_gas_price_percentage,
+            height,
+        );
+        exec.saturating_add(da)
     }
 }
 
@@ -370,10 +396,13 @@ impl AlgorithmUpdaterV1 {
     }
 
     pub fn algorithm(&self) -> AlgorithmV1 {
-        let combined_gas_price = self
-            .descaled_exec_price()
-            .saturating_add(self.descaled_da_price());
-        AlgorithmV1 { combined_gas_price }
+        AlgorithmV1 {
+            new_exec_price: self.descaled_exec_price(),
+            exec_price_percentage: self.exec_gas_price_change_percent as u64,
+            new_da_gas_price: self.descaled_da_price(),
+            da_gas_price_percentage: self.max_da_gas_price_change_percent as u64,
+            for_height: self.l2_block_height,
+        }
     }
 
     // We only need to track the difference between the rewards and costs after we have true DA data

crates/fuel-gas-price-algorithm/src/v1/tests/algorithm_v1_tests.rs

@@ -1,4 +1,8 @@
-use crate::v1::tests::UpdaterBuilder;
+use crate::v1::{
+    tests::UpdaterBuilder,
+    AlgorithmV1,
+};
+use proptest::prelude::*;
 
 #[test]
 fn calculate__returns_sum_of_da_and_exec_gas_prices() {
@@ -28,3 +32,60 @@ fn calculate__returns_sum_of_da_and_exec_gas_prices() {
     let expected = starting_exec_gas_price + starting_da_gas_price;
     assert_eq!(expected, actual);
 }
+
+fn _worst_case__correctly_calculates_value(
+    new_exec_price: u64,
+    new_da_gas_price: u64,
+    for_height: u32,
+    block_horizon: u32,
+    exec_price_percentage: u64,
+    da_gas_price_percentage: u64,
+) {
+    // given
+    let algorithm = AlgorithmV1 {
+        new_exec_price,
+        exec_price_percentage,
+        new_da_gas_price,
+        da_gas_price_percentage,
+        for_height,
+    };
+
+    // when
+    let target_height = for_height.saturating_add(block_horizon);
+    let actual = algorithm.worst_case(target_height);
+
+    // then
+    let mut expected_exec_price = new_exec_price;
+    let mut expected_da_gas_price = new_da_gas_price;
+
+    for _ in 0..block_horizon {
+        let change_amount = expected_exec_price
+            .saturating_mul(exec_price_percentage)
+            .saturating_div(100);
+        expected_exec_price = expected_exec_price.saturating_add(change_amount);
+
+        let change_amount = expected_da_gas_price
+            .saturating_mul(da_gas_price_percentage)
+            .saturating_div(100);
+        expected_da_gas_price = expected_da_gas_price.saturating_add(change_amount);
+    }
+
+    let expected = expected_exec_price.saturating_add(expected_da_gas_price);
+
+    dbg!(actual, expected);
+    assert!(actual >= expected);
+}
+
+proptest! {
+    #[test]
+    fn worst_case__correctly_calculates_value(
+        exec_price: u64,
+        da_price: u64,
+        starting_height: u32,
+        block_horizon in 0..10_000u32,
+        exec_percentage: u64,
+        da_percentage: u64,
+    ) {
+        _worst_case__correctly_calculates_value(exec_price, da_price, starting_height, block_horizon, exec_percentage, da_percentage);
+    }
+}

crates/services/gas_price_service/src/fuel_gas_price_updater.rs

@@ -350,8 +350,7 @@ impl GasPriceAlgorithm for Algorithm {
     fn worst_case_gas_price(&self, height: BlockHeight) -> u64 {
         match self {
             Algorithm::V0(v0) => v0.worst_case(height.into()),
-            // TODO: Add worst_case calculation https://github.com/FuelLabs/fuel-core/issues/2203
-            Algorithm::V1(v1) => v1.calculate(),
+            Algorithm::V1(v1) => v1.worst_case(height.into()),
         }
     }
 }