Fix tests

osmosis-labs · Sep 5, 2024 · 8b6f4ee · 8b6f4ee
1 parent 9d82cc7
commit 8b6f4ee
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/contracts/consumer/virtual-staking/src/contract.rs b/contracts/consumer/virtual-staking/src/contract.rs
@@ -534,7 +534,7 @@ impl VirtualStakingApi for VirtualStakingContract<'_> {
                 delegator,
                 validator: validator.clone(),
             },
-            VirtualStakeMsg::Unbond { amount, validator }
+            VirtualStakeMsg::Unbond { amount, validator },
         ];
         Ok(Response::new().add_messages(msgs))
     }
@@ -579,7 +579,7 @@ impl VirtualStakingApi for VirtualStakingContract<'_> {
             let all_delegations = TokenQuerier::new(&deps.querier)
                 .all_delegations(env.contract.address.to_string(), config.max_retrieve)?;
             if all_delegations.delegations.len() == 0 {
-                return Ok(resp.add_message(VirtualStakeMsg::DeleteAllScheduledTasks{}));
+                return Ok(resp.add_message(VirtualStakeMsg::DeleteAllScheduledTasks {}));
             }
             let mut msgs = vec![];
             for delegation in all_delegations.delegations {
@@ -1737,9 +1737,9 @@ mod tests {
 
         #[track_caller]
         fn assert_no_bonding(&self) -> &Self {
-            if !self.virtual_stake_msgs.is_empty() {
+            if self.virtual_stake_msgs.len() > 1 {
                 panic!(
-                    "hit_epoch result was expected to be a noop, but has these: {:?}",
+                    "hit_epoch result was expected to only contain DeleteAllScheduledTasks, but has these: {:?}",
                     self.virtual_stake_msgs
                 );
             }

diff --git a/packages/bindings/src/msg.rs b/packages/bindings/src/msg.rs
@@ -39,7 +39,7 @@ pub enum VirtualStakeMsg {
         validator: String,
     },
     /// Delete all scheduled tasks after zero max cap and unbond all delegations
-    DeleteAllScheduledTasks {}
+    DeleteAllScheduledTasks {},
 }
 
 impl VirtualStakeMsg {

diff --git a/packages/virtual-staking-mock/src/lib.rs b/packages/virtual-staking-mock/src/lib.rs
@@ -136,6 +136,9 @@ impl Module for VirtualStakingModule {
                 }
             }
             mesh_bindings::VirtualStakeMsg::UpdateDelegation { .. } => Ok(AppResponse::default()),
+            mesh_bindings::VirtualStakeMsg::DeleteAllScheduledTasks { .. } => {
+                Ok(AppResponse::default())
+            }
         }
     }