From 863ed888ad8317255b68bdd05e7962c10b8a1f08 Mon Sep 17 00:00:00 2001 From: jonesmarvin8 <83104039+jonesmarvin8@users.noreply.github.com> Date: Thu, 27 Nov 2025 06:07:28 -0500 Subject: [PATCH] minor fix to remove_liquidity Fixed line 2767 to use delta_lp in liquidity_pool_cap --- nssa/src/state.rs | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/nssa/src/state.rs b/nssa/src/state.rs index 8a1aadf..b7d535b 100644 --- a/nssa/src/state.rs +++ b/nssa/src/state.rs @@ -2754,9 +2754,8 @@ pub mod tests { let user_b_post = state.get_account_by_address(&user_b_holding_address); let user_lp_post = state.get_account_by_address(&user_lp_holding_address); - //TODO: temp - // let delta_lp : u128 = (pool_def_data.liquidity_pool_cap*user_lp_amt)/pool_def_data.liquidity_pool_cap; - let delta_lp : u128 = (init_balance_a*init_balance_a)/init_balance_a; + //TODO: this accounts for the initial balance for User_LP + let delta_lp : u128 = (init_balance_a*init_balance_a + temp_amt)/init_balance_a; let expected_pool = Account { program_owner: Program::amm().id(), @@ -2768,7 +2767,7 @@ pub mod tests { vault_a_addr: vault_a_address, vault_b_addr: vault_b_address, liquidity_pool_id: token_lp_definition_address, - liquidity_pool_cap: PoolDefinition::parse(&pool_post.data).unwrap().liquidity_pool_cap, //TODOinit_balance_a - delta_lp, + liquidity_pool_cap: init_balance_a - delta_lp, //TODO: not 0 due to temp_amt; results in wrapping arithmetic. reserve_a: 0, reserve_b: 0, token_program_id: Program::token().id(),