lints

contracts/factory/src/tests/setup.rs

@@ -17,18 +17,21 @@ pub mod old_factory {
 }
 
 #[allow(clippy::too_many_arguments)]
+#[cfg(feature = "upgrade")]
 pub fn old_lp_wasm(env: &Env) -> BytesN<32> {
     soroban_sdk::contractimport!(file = "../../.artifacts/old_phoenix_pool.wasm");
     env.deployer().upload_contract_wasm(WASM)
 }
 
 #[allow(clippy::too_many_arguments)]
+#[cfg(feature = "upgrade")]
 pub fn old_stake_wasm(env: &Env) -> BytesN<32> {
     soroban_sdk::contractimport!(file = "../../.artifacts/old_phoenix_stake.wasm");
     env.deployer().upload_contract_wasm(WASM)
 }
 
 #[allow(clippy::too_many_arguments)]
+#[cfg(feature = "upgrade")]
 pub fn install_latest_factory(env: &Env) -> BytesN<32> {
     soroban_sdk::contractimport!(
         file = "../../target/wasm32-unknown-unknown/release/phoenix_factory.wasm"

contracts/pool/src/tests/setup.rs

@@ -30,6 +30,7 @@ pub mod latest_liquidity_pool {
     );
 }
 
+#[cfg(feature = "upgrade")]
 pub fn install_old_token_wasm(env: &Env) -> BytesN<32> {
     soroban_sdk::contractimport!(file = "../../.artifacts/old_soroban_token_contract.wasm");
     env.deployer().upload_contract_wasm(WASM)
@@ -51,6 +52,7 @@ pub fn install_stake_wasm(env: &Env) -> BytesN<32> {
 }
 
 #[allow(clippy::too_many_arguments)]
+#[cfg(feature = "upgrade")]
 pub fn install_new_lp_wasm(env: &Env) -> BytesN<32> {
     soroban_sdk::contractimport!(
         file = "../../target/wasm32-unknown-unknown/release/phoenix_pool.wasm"

contracts/stake/src/tests/setup.rs

@@ -3,7 +3,6 @@ use crate::{
     token_contract,
 };
 use soroban_sdk::{testutils::Address as _, Address, BytesN, Env};
-use soroban_sdk::{testutils::Ledger, vec};
 
 pub fn deploy_token_contract<'a>(env: &Env, admin: &Address) -> token_contract::Client<'a> {
     token_contract::Client::new(
@@ -63,6 +62,8 @@ pub fn deploy_staking_contract<'a>(
 #[allow(deprecated)]
 #[cfg(feature = "upgrade")]
 fn upgrade_stake_contract() {
+    use soroban_sdk::{testutils::Ledger, vec};
+
     let env = Env::default();
     env.mock_all_auths();
     env.cost_estimate().budget().reset_unlimited();

contracts/stake_rewards/Cargo.toml

@@ -11,6 +11,7 @@ crate-type = ["cdylib"]
 
 [features]
 testutils = ["soroban-sdk/testutils"]
+upgrade = []
 
 [dependencies]
 soroban-decimal = { workspace = true }

contracts/stake_rewards/src/tests/setup.rs

@@ -1,10 +1,30 @@
-use soroban_sdk::{Address, Env};
+use soroban_sdk::{Address, BytesN, Env};
 
 use crate::{
     contract::{StakingRewards, StakingRewardsClient},
     token_contract,
 };
 
+#[allow(clippy::too_many_arguments)]
+pub mod old_stake_rewards {
+    soroban_sdk::contractimport!(file = "../../.artifacts/old_phoenix_stake_rewards.wasm");
+}
+
+#[allow(clippy::too_many_arguments)]
+pub mod latest_stake_rewards {
+    soroban_sdk::contractimport!(
+        file = "../../target/wasm32-unknown-unknown/release/phoenix_stake_rewards.wasm"
+    );
+}
+
+#[allow(clippy::too_many_arguments)]
+pub fn install_stake_rewards_wasm(env: &Env) -> BytesN<32> {
+    soroban_sdk::contractimport!(
+        file = "../../target/wasm32-unknown-unknown/release/phoenix_stake_rewards.wasm"
+    );
+    env.deployer().upload_contract_wasm(WASM)
+}
+
 pub fn deploy_token_contract<'a>(env: &Env, admin: &Address) -> token_contract::Client<'a> {
     token_contract::Client::new(
         env,
@@ -35,3 +55,62 @@ pub fn deploy_staking_rewards_contract<'a>(
     );
     staking_rewards
 }
+
+#[test]
+#[allow(deprecated)]
+#[cfg(feature = "upgrade")]
+fn updapte_stake_rewards() {
+    use soroban_sdk::testutils::Address as _;
+
+    let env = Env::default();
+    env.mock_all_auths();
+    env.cost_estimate().budget().reset_unlimited();
+
+    let admin = Address::generate(&env);
+    let staking_contract = Address::generate(&env);
+    let reward_token = Address::generate(&env);
+
+    let old_stake_rewards_addr = env.register_contract_wasm(None, old_stake_rewards::WASM);
+
+    let old_stake_rewards_client = old_stake_rewards::Client::new(&env, &old_stake_rewards_addr);
+
+    old_stake_rewards_client.initialize(
+        &admin,
+        &staking_contract,
+        &reward_token,
+        &MAX_COMPLEXITY,
+        &MIN_REWARD,
+        &MIN_BOND,
+    );
+
+    assert_eq!(old_stake_rewards_client.query_admin(), admin);
+    assert_eq!(
+        old_stake_rewards_client.query_config().config,
+        old_stake_rewards::Config {
+            max_complexity: MAX_COMPLEXITY,
+            min_bond: MIN_BOND,
+            min_reward: MIN_REWARD,
+            reward_token: reward_token.clone(),
+            staking_contract: staking_contract.clone(),
+        }
+    );
+
+    let latest_stake_rewards_wasm = install_stake_rewards_wasm(&env);
+
+    old_stake_rewards_client.update(&latest_stake_rewards_wasm);
+
+    let latest_stake_rewards_client =
+        latest_stake_rewards::Client::new(&env, &old_stake_rewards_addr);
+
+    assert_eq!(latest_stake_rewards_client.query_admin(), admin);
+    assert_eq!(
+        latest_stake_rewards_client.query_config().config,
+        latest_stake_rewards::Config {
+            max_complexity: MAX_COMPLEXITY,
+            min_bond: MIN_BOND,
+            min_reward: MIN_REWARD,
+            reward_token,
+            staking_contract
+        }
+    );
+}