From 9d5d1076f2dc8c787f4d356e46fc00257e623561 Mon Sep 17 00:00:00 2001
From: Dmitriy Khomitskiy <dmitriy@helius.xyz>
Date: Mon, 9 Dec 2024 11:13:59 -0600
Subject: [PATCH] RPC-470 moving the getPriorityFeeEstimate to use v2, for
 easier migration in CF workder

---
 src/rpc_server.rs | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/src/rpc_server.rs b/src/rpc_server.rs
index eb1f76a..a004460 100644
--- a/src/rpc_server.rs
+++ b/src/rpc_server.rs
@@ -145,11 +145,11 @@ pub trait AtlasPriorityFeeEstimatorRpc {
 
     // TODO: DKH - delete after all the users were notified about moving to strict parsing
     #[method(name = "getPriorityFeeEstimate")]
-    fn get_priority_fee_estimate_light(
+    fn get_priority_fee_estimate(
         &self,
-        get_priority_fee_estimate_request: GetPriorityFeeEstimateRequestLight,
+        get_priority_fee_estimate_request: GetPriorityFeeEstimateRequest,
     ) -> RpcResult<GetPriorityFeeEstimateResponse> {
-        self.get_priority_fee_estimate_v1(get_priority_fee_estimate_request)
+        self.get_priority_fee_estimate_v2(get_priority_fee_estimate_request)
     }
 
     #[method(name = "getPriorityFeeEstimateV1")]
@@ -537,9 +537,9 @@ mod tests {
             max_lookback_slots: 150,
         };
 
-        let result = server.get_priority_fee_estimate_light(GetPriorityFeeEstimateRequestLight {
+        let result = server.get_priority_fee_estimate(GetPriorityFeeEstimateRequest {
             account_keys: Some(vec![acc1.to_string(), acc2.to_string()]),
-            options: Some(GetPriorityFeeEstimateOptionsLight::default()),
+            options: Some(GetPriorityFeeEstimateOptions::default()),
             ..Default::default()
         });
         let resp = result.unwrap();
@@ -562,7 +562,7 @@ mod tests {
             max_lookback_slots: 150,
         };
 
-        let result = server.get_priority_fee_estimate_light(GetPriorityFeeEstimateRequestLight {
+        let result = server.get_priority_fee_estimate(GetPriorityFeeEstimateRequest {
             account_keys: Some(vec![acc1.to_string(), acc2.to_string()]),
             ..Default::default()
         });
@@ -587,9 +587,9 @@ mod tests {
             max_lookback_slots: 150,
         };
 
-        let result = server.get_priority_fee_estimate_light(GetPriorityFeeEstimateRequestLight {
+        let result = server.get_priority_fee_estimate(GetPriorityFeeEstimateRequest {
             account_keys: Some(vec![acc1.to_string(), acc2.to_string()]),
-            options: Some(GetPriorityFeeEstimateOptionsLight {
+            options: Some(GetPriorityFeeEstimateOptions {
                 include_all_priority_fee_levels: Some(true),
                 ..Default::default()
             }),
@@ -597,9 +597,9 @@ mod tests {
         });
         let resp = result.unwrap();
         let levels = resp.priority_fee_levels.unwrap();
-        assert_eq!(levels.min, 100.0);
-        assert_eq!(levels.low, 100.0);
-        assert_eq!(levels.medium, 150.0);
+        assert_eq!(levels.min, 200.0);
+        assert_eq!(levels.low, 200.0);
+        assert_eq!(levels.medium, 200.0);
         assert_eq!(levels.high, 200.0);
         assert_eq!(levels.very_high, 200.0);
         assert_eq!(levels.unsafe_max, 200.0);
@@ -621,9 +621,9 @@ mod tests {
             max_lookback_slots: 150,
         };
 
-        let result = server.get_priority_fee_estimate_light(GetPriorityFeeEstimateRequestLight {
+        let result = server.get_priority_fee_estimate(GetPriorityFeeEstimateRequest {
             account_keys: Some(vec![acc1.to_string(), acc2.to_string()]),
-            options: Some(GetPriorityFeeEstimateOptionsLight {
+            options: Some(GetPriorityFeeEstimateOptions {
                 recommended: Some(true),
                 ..Default::default()
             }),