feat: reduce yield if not enough rewards (#514)

* go

* code is good

* Checkpoint

* test: add test

* better test

* test

* revert this file

* clippy

* go

* go

* undo changes to stability

* cleanup

staking/integration-tests/src/integrity_pool/helper_functions.rs

@@ -19,6 +19,7 @@ pub fn initialize_pool_reward_custody(
     svm: &mut litesvm::LiteSVM,
     payer: &Keypair,
     pyth_token_mint: &Keypair,
+    reward_amount_override: Option<u64>,
 ) {
     let pool_config_pubkey = get_pool_config_address();
 
@@ -30,6 +31,6 @@ pub fn initialize_pool_reward_custody(
         payer,
         get_pool_reward_custody_address(pyth_token_mint.pubkey()),
         pyth_token_mint,
-        1_000_000 * FRAC_64_MULTIPLIER,
+        reward_amount_override.unwrap_or(1_000_000 * FRAC_64_MULTIPLIER),
     );
 }

staking/integration-tests/src/integrity_pool/instructions.rs

@@ -49,13 +49,19 @@ use {
 
 pub fn advance(svm: &mut LiteSVM, payer: &Keypair, publisher_caps: Pubkey) -> TransactionResult {
     let pool_config = get_pool_config_address();
-    let pool_data = fetch_account_data::<PoolConfig>(svm, &pool_config).pool_data;
+    let PoolConfig {
+        pool_data,
+        pyth_token_mint,
+        ..
+    } = fetch_account_data::<PoolConfig>(svm, &pool_config);
+    let pool_reward_custody = get_pool_reward_custody_address(pyth_token_mint);
 
     let accounts = integrity_pool::accounts::Advance {
         signer: payer.pubkey(),
         pool_config,
         publisher_caps,
         pool_data,
+        pool_reward_custody,
     };
 
     let instruction_data = integrity_pool::instruction::Advance {};

staking/integration-tests/src/setup.rs

@@ -40,11 +40,12 @@ pub struct SetupResult {
 }
 
 pub struct SetupProps {
-    pub init_config:     bool,
-    pub init_target:     bool,
-    pub init_mint:       bool,
-    pub init_pool_data:  bool,
-    pub init_publishers: bool,
+    pub init_config:            bool,
+    pub init_target:            bool,
+    pub init_mint:              bool,
+    pub init_pool_data:         bool,
+    pub init_publishers:        bool,
+    pub reward_amount_override: Option<u64>,
 }
 
 pub fn setup(props: SetupProps) -> SetupResult {
@@ -54,6 +55,7 @@ pub fn setup(props: SetupProps) -> SetupResult {
         init_mint,
         init_pool_data,
         init_publishers,
+        reward_amount_override,
     } = props;
 
     let pyth_token_mint = Keypair::new();
@@ -106,7 +108,7 @@ pub fn setup(props: SetupProps) -> SetupResult {
 
     advance_n_epochs(&mut svm, &payer, 1);
     if init_publishers {
-        initialize_pool_reward_custody(&mut svm, &payer, &pyth_token_mint);
+        initialize_pool_reward_custody(&mut svm, &payer, &pyth_token_mint, reward_amount_override);
         let publisher_caps =
             post_dummy_publisher_caps(&mut svm, &payer, publisher_keypair.pubkey(), 100);
         advance(&mut svm, &payer, publisher_caps).unwrap();