Engagement half-life

contracts/tg4-engagement/src/contract.rs

@@ -1,8 +1,8 @@
 #[cfg(not(feature = "library"))]
 use cosmwasm_std::entry_point;
 use cosmwasm_std::{
-    coin, to_binary, Addr, BankMsg, Binary, Coin, CustomQuery, Decimal, Deps, DepsMut, Empty, Env,
-    Event, MessageInfo, Order, StdResult, Timestamp, Uint128,
+    coin, to_binary, Addr, BankMsg, Binary, Coin, CustomQuery, Decimal, Deps, DepsMut, Env, Event,
+    MessageInfo, Order, StdResult, Timestamp, Uint128,
 };
 use cw2::set_contract_version;
 use cw_storage_plus::Bound;
@@ -14,8 +14,8 @@ use tg4::{
 
 use crate::error::ContractError;
 use crate::msg::{
-    DelegatedResponse, ExecuteMsg, HalflifeInfo, HalflifeResponse, InstantiateMsg, PreauthResponse,
-    QueryMsg, RewardsResponse, SudoMsg,
+    DelegatedResponse, ExecuteMsg, HalflifeInfo, HalflifeResponse, InstantiateMsg, MigrateMsg,
+    PreauthResponse, QueryMsg, RewardsResponse, SudoMsg,
 };
 use crate::state::{
     Distribution, Halflife, WithdrawAdjustment, DISTRIBUTION, HALFLIFE, PREAUTH_SLASHING,
@@ -913,8 +913,26 @@ fn list_members_by_points<Q: CustomQuery>(
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]
-pub fn migrate(deps: DepsMut, _env: Env, _msg: Empty) -> Result<Response, ContractError> {
+pub fn migrate(
+    deps: DepsMut<TgradeQuery>,
+    _env: Env,
+    msg: MigrateMsg,
+) -> Result<Response, ContractError> {
     ensure_from_older_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
+    if let Some(duration) = msg.halflife {
+        // Update half life's duration
+        // Zero duration means no / remove half life
+        HALFLIFE.update(deps.storage, |hf| -> StdResult<_> {
+            Ok(Halflife {
+                halflife: if duration.seconds() > 0 {
+                    Some(duration)
+                } else {
+                    None
+                },
+                last_applied: hf.last_applied,
+            })
+        })?;
+    };
     Ok(Response::new())
 }
 

contracts/tg4-engagement/src/msg.rs

@@ -183,6 +183,12 @@ pub struct ListSlashersResponse {
     pub slashers: Vec<String>,
 }
 
+#[derive(Serialize, Deserialize, Clone, PartialEq, Eq, JsonSchema, Debug)]
+#[serde(rename_all = "snake_case")]
+pub struct MigrateMsg {
+    pub halflife: Option<Duration>,
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

contracts/tg4-engagement/src/multitest.rs

@@ -921,3 +921,54 @@ mod slashing {
         assert_eq!(suite.token_balance(members[2]).unwrap(), 0);
     }
 }
+
+mod migration {
+    use super::*;
+    use crate::msg::MigrateMsg;
+
+    #[test]
+    fn migration_can_alter_cfg() {
+        let mut suite = SuiteBuilder::new()
+            .with_halflife(Duration::new(100))
+            .build();
+        let admin = suite.admin().to_string();
+
+        let cfg = suite.halflife().unwrap();
+        assert_eq!(cfg.halflife_info.unwrap().halflife, Duration::new(100));
+
+        suite
+            .migrate(
+                &admin,
+                &MigrateMsg {
+                    halflife: Some(Duration::new(200)),
+                },
+            )
+            .unwrap();
+
+        let cfg = suite.halflife().unwrap();
+        assert_eq!(cfg.halflife_info.unwrap().halflife.seconds(), 200);
+    }
+
+    #[test]
+    fn migration_can_remove_halflife() {
+        let mut suite = SuiteBuilder::new()
+            .with_halflife(Duration::new(100))
+            .build();
+        let admin = suite.admin().to_string();
+
+        let cfg = suite.halflife().unwrap();
+        assert_eq!(cfg.halflife_info.unwrap().halflife, Duration::new(100));
+
+        suite
+            .migrate(
+                &admin,
+                &MigrateMsg {
+                    halflife: Some(Duration::new(0)),
+                },
+            )
+            .unwrap();
+
+        let cfg = suite.halflife().unwrap();
+        assert!(cfg.halflife_info.is_none());
+    }
+}

contracts/tg4-engagement/src/multitest/suite.rs

@@ -15,6 +15,7 @@ fn contract_engagement() -> Box<dyn Contract<TgradeMsg, TgradeQuery>> {
         crate::contract::instantiate,
         crate::contract::query,
     )
+    .with_migrate(crate::contract::migrate)
     .with_sudo(crate::contract::sudo);
 
     Box::new(contract)
@@ -133,6 +134,7 @@ impl SuiteBuilder {
 
         Suite {
             app,
+            code_id: contract_id,
             contract,
             owner,
             denom,
@@ -145,9 +147,11 @@ impl SuiteBuilder {
 pub struct Suite {
     #[derivative(Debug = "ignore")]
     pub app: TgradeApp,
+    /// The code id of the engagement contract
+    code_id: u64,
     /// Engagement contract address
     pub contract: Addr,
-    /// Mixer contract address
+    /// Admin of engagement contract
     pub owner: Addr,
     /// Denom of tokens which might be distributed by this contract
     pub denom: String,
@@ -170,6 +174,10 @@ impl Suite {
         )
     }
 
+    pub fn admin(&self) -> &str {
+        self.owner.as_str()
+    }
+
     pub fn withdraw_funds<'s>(
         &mut self,
         executor: &str,
@@ -340,4 +348,22 @@ impl Suite {
         )?;
         Ok(resp.members)
     }
+
+    /// Queries engagement contract for its halflife
+    pub fn halflife(&self) -> StdResult<HalflifeResponse> {
+        self.app
+            .wrap()
+            .query_wasm_smart(&self.contract, &QueryMsg::Halflife {})
+    }
+
+    /// Migrates the contract to the same version (same code id), but possibly changing
+    /// some cfg values via MigrateMsg.
+    pub fn migrate(&mut self, addr: &str, msg: &MigrateMsg) -> AnyResult<AppResponse> {
+        self.app.migrate_contract(
+            Addr::unchecked(addr),
+            self.contract.clone(),
+            msg,
+            self.code_id,
+        )
+    }
 }