add token supply init param

multiversx · Dec 14, 2022 · 09ac7eb · 09ac7eb
1 parent 5ad0d52
commit 09ac7eb
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 0 deletions.
diff --git a/farm-staking/farm-staking-proxy/tests/staking_farm_with_lp_staking_contract_setup/mod.rs b/farm-staking/farm-staking-proxy/tests/staking_farm_with_lp_staking_contract_setup/mod.rs
@@ -43,6 +43,7 @@ where
                 max_apr,
                 UNBOND_EPOCHS,
                 0,
+                managed_biguint!(0),
                 ManagedAddress::<DebugApi>::zero(),
                 MultiValueEncoded::new(),
             );

diff --git a/farm-staking/farm-staking/src/lib.rs b/farm-staking/farm-staking/src/lib.rs
@@ -56,6 +56,7 @@ pub trait FarmStaking:
         max_apr: BigUint,
         min_unbond_epochs: u64,
         upgrade_block: Nonce,
+        new_farm_supply: BigUint,
         owner: ManagedAddress,
         admins: MultiValueEncoded<ManagedAddress>,
     ) {
@@ -89,6 +90,10 @@ pub trait FarmStaking:
         });
         self.reward_reserve()
             .update(|r| *r += accumulated_rewards_before);
+
+        if new_farm_supply > 0 {
+            self.farm_token_supply().set(new_farm_supply);
+        }
     }
 
     #[payable("*")]

diff --git a/farm-staking/farm-staking/tests/farm_staking_test.rs b/farm-staking/farm-staking/tests/farm_staking_test.rs
@@ -69,6 +69,7 @@ where
                 managed_biguint!(MAX_APR),
                 MIN_UNBOND_EPOCHS,
                 0,
+                managed_biguint!(0),
                 ManagedAddress::<DebugApi>::zero(),
                 MultiValueEncoded::new(),
             );