Stake: using tag v1.0.0 impl

contracts/stake/src/contract.rs

@@ -762,6 +762,27 @@ impl Staking {
 
         env.deployer().update_current_contract_wasm(new_wasm_hash);
     }
+
+    #[allow(dead_code)]
+    pub fn migrate_distributions(env: Env) {
+        let distributions = get_distributions(&env);
+
+        distributions.iter().for_each(|distribution_addr| {
+            save_distribution(
+                &env,
+                &distribution_addr,
+                &Distribution {
+                    shares_per_point: 1u128,
+                    shares_leftover: 0u64,
+                    distributed_total: 0u128,
+                    withdrawable_total: 0u128,
+                    max_bonus_bps: 0u64,
+                    bonus_per_day_bps: 0u64,
+                },
+            );
+            save_reward_curve(&env, distribution_addr, &Curve::Constant(0));
+        })
+    }
 }
 
 // Function to remove a stake from the vector

contracts/stake/src/tests/setup.rs

@@ -283,10 +283,12 @@ mod tests {
         let new_stake_wasm = install_stake_wasm(&env);
 
         old_stake_client.update(&new_stake_wasm);
-        //old_stake_client.update(&new_stake_wasm);
 
         let latest_stake_client = latest_stake::Client::new(&env, &stake_addr);
 
+        // now we migrate the distributions
+        latest_stake_client.migrate_distributions();
+
         // check the rewards again, this time with the old deprecated method
         assert_eq!(
             latest_stake_client.query_withdrawable_rewards_dep(&user_1),
@@ -450,15 +452,29 @@ mod tests {
 
         lp_token_client.mint(&new_user, &10_000_000_000_000);
 
-        soroban_sdk::testutils::arbitrary::std::dbg!("BEFORE");
         latest_stake_client.bond(&new_user, &10_000_000_000); // new_user also bonds 1,000 tokens
-        soroban_sdk::testutils::arbitrary::std::dbg!("AFTER");
 
         // two months pass by
         env.ledger()
             .with_mut(|li| li.timestamp += 60 * DAY_AS_SECONDS);
 
         // distribute and take the rewards
         latest_stake_client.distribute_rewards();
+
+        assert_eq!(
+            latest_stake_client.query_withdrawable_rewards(&new_user),
+            latest_stake::WithdrawableRewardsResponse {
+                rewards: vec![
+                    &env,
+                    latest_stake::WithdrawableReward {
+                        reward_address: reward_token_addr.clone(),
+                        reward_amount: 5_000_000,
+                    }
+                ]
+            }
+        );
+
+        latest_stake_client.withdraw_rewards(&new_user);
+        assert_eq!(reward_token_client.balance(&new_user), 5_000_000);
     }
 }