Adapt slash test method helper

osmosis-labs · Dec 4, 2023 · ebf1785 · ebf1785
1 parent 246c7bb
commit ebf1785
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 9 deletions.
diff --git a/contracts/provider/external-staking/src/multitest.rs b/contracts/provider/external-staking/src/multitest.rs
@@ -1359,7 +1359,7 @@ fn slashing() {
     // But now validators[0] slashing happens
     contract
         .test_methods_proxy()
-        .test_handle_slashing(validators[0].to_string())
+        .test_handle_slashing(validators[0].to_string(), Uint128::new(8))
         .call("test")
         .unwrap();
 
@@ -1444,7 +1444,7 @@ fn slashing_pending_tx_partial_unbond() {
     // Now validators[0] slashing happens
     contract
         .test_methods_proxy()
-        .test_handle_slashing(validators[0].to_string())
+        .test_handle_slashing(validators[0].to_string(), Uint128::new(20))
         .call("test")
         .unwrap();
 
@@ -1527,7 +1527,7 @@ fn slashing_pending_tx_full_unbond() {
     // Now validators[0] slashing happens
     contract
         .test_methods_proxy()
-        .test_handle_slashing(validators[0].to_string())
+        .test_handle_slashing(validators[0].to_string(), Uint128::new(20))
         .call("test")
         .unwrap();
 
@@ -1612,7 +1612,7 @@ fn slashing_pending_tx_full_unbond_rolled_back() {
     // Now validators[0] slashing happens
     contract
         .test_methods_proxy()
-        .test_handle_slashing(validators[0].to_string())
+        .test_handle_slashing(validators[0].to_string(), Uint128::new(20))
         .call("test")
         .unwrap();
 
@@ -1717,7 +1717,7 @@ fn slashing_pending_tx_bond() {
     // Now validators[0] slashing happens
     contract
         .test_methods_proxy()
-        .test_handle_slashing(validators[0].to_string())
+        .test_handle_slashing(validators[0].to_string(), Uint128::new(20))
         .call("test")
         .unwrap();
 
@@ -1804,7 +1804,7 @@ fn slashing_pending_tx_bond_rolled_back() {
     // Now validators[0] slashing happens
     contract
         .test_methods_proxy()
-        .test_handle_slashing(validators[0].to_string())
+        .test_handle_slashing(validators[0].to_string(), Uint128::new(20))
         .call("test")
         .unwrap();
 

diff --git a/contracts/provider/external-staking/src/test_methods.rs b/contracts/provider/external-staking/src/test_methods.rs
@@ -1,4 +1,4 @@
-use cosmwasm_std::{Coin, Response, StdError};
+use cosmwasm_std::{Coin, Response, StdError, Uint128};
 use mesh_apis::converter_api::RewardInfo;
 use mesh_apis::ibc::AddValidator;
 use sylvia::interface;
@@ -97,5 +97,6 @@ pub trait TestMethods {
         &self,
         ctx: ExecCtx,
         validator: String,
+        slash_amount: Uint128,
     ) -> Result<Response, Self::Error>;
 }
diff --git a/contracts/provider/external-staking/src/test_methods_impl.rs b/contracts/provider/external-staking/src/test_methods_impl.rs
@@ -2,7 +2,7 @@ use crate::contract::ExternalStakingContract;
 use crate::error::ContractError;
 use crate::test_methods::TestMethods;
 
-use cosmwasm_std::{Coin, Response};
+use cosmwasm_std::{Coin, Response, Uint128};
 use mesh_apis::converter_api::RewardInfo;
 use mesh_apis::ibc::AddValidator;
 use sylvia::contract;
@@ -225,6 +225,7 @@ impl TestMethods for ExternalStakingContract<'_> {
         &self,
         ctx: ExecCtx,
         validator: String,
+        slash_amount: Uint128,
     ) -> Result<Response, ContractError> {
         #[cfg(any(test, feature = "mt"))]
         {
@@ -235,6 +236,8 @@ impl TestMethods for ExternalStakingContract<'_> {
                 &cfg,
                 &validator,
                 cfg.slash_ratio.double_sign,
+                slash_amount,
+                ctx.env.block.time.seconds(),
             )?;
             match slash_msg {
                 Some(msg) => Ok(Response::new().add_message(msg)),
@@ -243,7 +246,7 @@ impl TestMethods for ExternalStakingContract<'_> {
         }
         #[cfg(not(any(test, feature = "mt")))]
         {
-            let _ = (ctx, validator);
+            let _ = (ctx, validator, slash_amount);
             Err(ContractError::Unauthorized {})
         }
     }