Add more tests for inexact deposits (#834)

quasar-finance · Aug 30, 2024 · 896ef1b · 896ef1b
1 parent d75209f
commit 896ef1b
Show file tree

Hide file tree

Showing 4 changed files with 334 additions and 40 deletions.
diff --git a/smart-contracts/osmosis/contracts/cl-vault/src/test_helpers.rs b/smart-contracts/osmosis/contracts/cl-vault/src/test_helpers.rs
@@ -60,6 +60,10 @@ impl QuasarQuerier {
             bank: BankQuerier::new(balances),
         }
     }
+
+    pub fn update_balances(&mut self, balances: &[(&str, &[Coin])]) {
+        self.bank = BankQuerier::new(balances);
+    }
 }
 
 impl Querier for QuasarQuerier {
@@ -167,48 +171,68 @@ impl Querier for QuasarQuerier {
     }
 }
 
+fn get_quasar_querier_with_balances(
+    position_base_amount: u128,
+    position_quote_amount: u128,
+    current_tick: i64,
+    position_lower_tick: i64,
+    position_upper_tick: i64,
+    balances: &[(&str, &[Coin])],
+) -> QuasarQuerier {
+    QuasarQuerier::new_with_balances(
+        FullPositionBreakdown {
+            position: Some(OsmoPosition {
+                position_id: POSITION_ID,
+                address: MOCK_CONTRACT_ADDR.to_string(),
+                pool_id: POOL_ID,
+                lower_tick: position_lower_tick,
+                upper_tick: position_upper_tick,
+                join_time: None,
+                liquidity: "1000000.1".to_string(),
+            }),
+            asset0: Some(OsmoCoin {
+                denom: BASE_DENOM.to_string(),
+                amount: position_base_amount.to_string(),
+            }),
+            asset1: Some(OsmoCoin {
+                denom: QUOTE_DENOM.to_string(),
+                amount: position_quote_amount.to_string(),
+            }),
+            claimable_spread_rewards: vec![
+                OsmoCoin {
+                    denom: BASE_DENOM.to_string(),
+                    amount: "100".to_string(),
+                },
+                OsmoCoin {
+                    denom: QUOTE_DENOM.to_string(),
+                    amount: "100".to_string(),
+                },
+            ],
+            claimable_incentives: vec![],
+            forfeited_incentives: vec![],
+        },
+        current_tick,
+        balances,
+    )
+}
+
 pub fn mock_deps_with_querier_with_balance(
     position_base_amount: u128,
     position_quote_amount: u128,
     current_tick: i64,
+    position_lower_tick: i64,
+    position_upper_tick: i64,
     balances: &[(&str, &[Coin])],
 ) -> OwnedDeps<MockStorage, MockApi, QuasarQuerier, Empty> {
     OwnedDeps {
         storage: MockStorage::default(),
         api: MockApi::default(),
-        querier: QuasarQuerier::new_with_balances(
-            FullPositionBreakdown {
-                position: Some(OsmoPosition {
-                    position_id: POSITION_ID,
-                    address: MOCK_CONTRACT_ADDR.to_string(),
-                    pool_id: POOL_ID,
-                    lower_tick: 100,
-                    upper_tick: 1000,
-                    join_time: None,
-                    liquidity: "1000000.1".to_string(),
-                }),
-                asset0: Some(OsmoCoin {
-                    denom: BASE_DENOM.to_string(),
-                    amount: position_base_amount.to_string(),
-                }),
-                asset1: Some(OsmoCoin {
-                    denom: QUOTE_DENOM.to_string(),
-                    amount: position_quote_amount.to_string(),
-                }),
-                claimable_spread_rewards: vec![
-                    OsmoCoin {
-                        denom: BASE_DENOM.to_string(),
-                        amount: "100".to_string(),
-                    },
-                    OsmoCoin {
-                        denom: QUOTE_DENOM.to_string(),
-                        amount: "100".to_string(),
-                    },
-                ],
-                claimable_incentives: vec![],
-                forfeited_incentives: vec![],
-            },
+        querier: get_quasar_querier_with_balances(
+            position_base_amount,
+            position_quote_amount,
             current_tick,
+            position_lower_tick,
+            position_upper_tick,
             balances,
         ),
         custom_query_type: PhantomData,