Additional pause

contracts/lsm-share-bond-provider/src/contract.rs

@@ -10,6 +10,7 @@ use drop_helpers::answer::{attr_coin, response};
 use drop_helpers::get_contracts;
 use drop_helpers::ibc_client_state::query_client_state;
 use drop_helpers::ibc_fee::query_ibc_fee;
+use drop_helpers::pause::PauseError;
 use drop_puppeteer_base::peripheral_hook::{
     IBCTransferReason, ReceiverExecuteMsg, ResponseAnswer, ResponseHookErrorMsg, ResponseHookMsg,
     ResponseHookSuccessMsg, Transaction,
@@ -24,7 +25,7 @@ use drop_staking_base::msg::lsm_share_bond_provider::{
 use drop_staking_base::state::core::LAST_PUPPETEER_RESPONSE;
 use drop_staking_base::state::lsm_share_bond_provider::{
     Config, ConfigOptional, ReplyMsg, TxState, TxStateStatus, CONFIG, LAST_LSM_REDEEM,
-    LSM_SHARES_TO_REDEEM, PENDING_LSM_SHARES, TOTAL_LSM_SHARES_REAL_AMOUNT, TX_STATE,
+    LSM_SHARES_TO_REDEEM, PAUSE, PENDING_LSM_SHARES, TOTAL_LSM_SHARES_REAL_AMOUNT, TX_STATE,
 };
 use neutron_sdk::bindings::msg::NeutronMsg;
 use neutron_sdk::bindings::query::NeutronQuery;
@@ -58,7 +59,7 @@ pub fn instantiate(
         lsm_redeem_maximum_interval: msg.lsm_redeem_maximum_interval,
     };
     CONFIG.save(deps.storage, config)?;
-
+    PAUSE.save(deps.storage, &false)?;
     TOTAL_LSM_SHARES_REAL_AMOUNT.save(deps.storage, &0)?;
     LAST_LSM_REDEEM.save(deps.storage, &env.block.time.seconds())?;
     TX_STATE.save(deps.storage, &TxState::default())?;
@@ -104,6 +105,7 @@ pub fn query(deps: Deps<NeutronQuery>, env: Env, msg: QueryMsg) -> ContractResul
         QueryMsg::AsyncTokensAmount {} => {
             to_json_binary(&TOTAL_LSM_SHARES_REAL_AMOUNT.load(deps.storage)?).map_err(From::from)
         }
+        QueryMsg::Pause {} => Ok(to_json_binary(&PAUSE.load(deps.storage)?)?),
         QueryMsg::CanBeRemoved {} => query_can_be_removed(deps, env),
     }
 }
@@ -225,14 +227,35 @@ pub fn execute(
         ExecuteMsg::Bond {} => execute_bond(deps, info),
         ExecuteMsg::ProcessOnIdle {} => execute_process_on_idle(deps, env, info),
         ExecuteMsg::PeripheralHook(msg) => execute_puppeteer_hook(deps, env, info, *msg),
+        ExecuteMsg::SetPause(pause) => execute_set_pause(deps.into_empty(), info, pause),
     }
 }
 
+fn execute_set_pause(
+    deps: DepsMut,
+    info: MessageInfo,
+    pause: bool,
+) -> ContractResult<Response<NeutronMsg>> {
+    cw_ownable::assert_owner(deps.storage, &info.sender)?;
+
+    PAUSE.save(deps.storage, &pause)?;
+
+    Ok(response(
+        "execute-set-pause",
+        CONTRACT_NAME,
+        [("process_on_idle", pause.to_string())],
+    ))
+}
+
 fn execute_process_on_idle(
     mut deps: DepsMut<NeutronQuery>,
     env: Env,
     info: MessageInfo,
 ) -> ContractResult<Response<NeutronMsg>> {
+    if PAUSE.load(deps.as_ref().storage)? {
+        Err(ContractError::PauseError(PauseError::Paused {}))?
+    }
+
     let config = CONFIG.load(deps.storage)?;
     let addrs = get_contracts!(deps, config.factory_contract, core_contract);
 

contracts/lsm-share-bond-provider/src/tests.rs

@@ -11,15 +11,16 @@ use cw_ownable::{Action, Ownership};
 use cw_utils::PaymentError;
 use drop_helpers::{
     ica::IcaState,
+    pause::PauseError,
     testing::{mock_dependencies, mock_state_query, WasmMockQuerier},
 };
 use drop_staking_base::{
     error::lsm_share_bond_provider::ContractError,
     msg::puppeteer::DelegationsResponse,
     state::{
         lsm_share_bond_provider::{
-            Config, ConfigOptional, ReplyMsg, TxState, CONFIG, LAST_LSM_REDEEM, PENDING_LSM_SHARES,
-            TOTAL_LSM_SHARES_REAL_AMOUNT, TX_STATE,
+            Config, ConfigOptional, ReplyMsg, TxState, CONFIG, LAST_LSM_REDEEM, PAUSE,
+            PENDING_LSM_SHARES, TOTAL_LSM_SHARES_REAL_AMOUNT, TX_STATE,
         },
         puppeteer::{Delegations, DropDelegation},
     },
@@ -344,9 +345,55 @@ fn test_update_ownership() {
     );
 }
 
+#[test]
+fn execute_set_pause() {
+    let mut deps = mock_dependencies(&[]);
+    let deps_mut = deps.as_mut();
+    cw_ownable::initialize_owner(deps_mut.storage, deps_mut.api, Some("core")).unwrap();
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
+    let res = crate::contract::execute(
+        deps.as_mut(),
+        mock_env(),
+        mock_info("core", &[]),
+        drop_staking_base::msg::lsm_share_bond_provider::ExecuteMsg::SetPause(true),
+    )
+    .unwrap();
+    assert_eq!(
+        res,
+        cosmwasm_std::Response::new().add_event(
+            Event::new("crates.io:drop-staking__drop-lsm-share-bond-provider-execute-set-pause")
+                .add_attributes(vec![attr("process_on_idle", "true")])
+        )
+    );
+    assert!(PAUSE.load(deps.as_ref().storage).unwrap());
+}
+
+#[test]
+fn process_on_idle_paused() {
+    let mut deps = mock_dependencies(&[]);
+    PAUSE.save(deps.as_mut().storage, &true).unwrap();
+
+    let error = crate::contract::execute(
+        deps.as_mut(),
+        mock_env(),
+        mock_info("not_core_contract", &[]),
+        drop_staking_base::msg::lsm_share_bond_provider::ExecuteMsg::ProcessOnIdle {},
+    )
+    .unwrap_err();
+
+    assert_eq!(
+        error,
+        drop_staking_base::error::lsm_share_bond_provider::ContractError::PauseError(
+            PauseError::Paused {}
+        )
+    );
+}
+
 #[test]
 fn process_on_idle_not_core_contract() {
     let mut deps = mock_dependencies(&[]);
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
     mock_state_query(&mut deps);
 
     CONFIG
@@ -372,7 +419,7 @@ fn test_process_on_idle_lsm_share_not_ready() {
     let mut deps = mock_dependencies(&[]);
     mock_state_query(&mut deps);
     let deps_mut = deps.as_mut();
-
+    PAUSE.save(deps_mut.storage, &false).unwrap();
     CONFIG
         .save(deps_mut.storage, &get_default_config(100u64, 200u64))
         .unwrap();
@@ -400,6 +447,7 @@ fn test_process_on_idle_lsm_share_not_ready() {
 #[test]
 fn test_process_on_idle_supported() {
     let mut deps = mock_dependencies(&[]);
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
     mock_state_query(&mut deps);
     deps.querier.add_custom_query_response(|_| {
         to_json_binary(&MinIbcFeeResponse {

contracts/native-bond-provider/src/contract.rs

@@ -9,6 +9,7 @@ use drop_helpers::answer::{attr_coin, response};
 use drop_helpers::get_contracts;
 use drop_helpers::ibc_client_state::query_client_state;
 use drop_helpers::ibc_fee::query_ibc_fee;
+use drop_helpers::pause::PauseError;
 use drop_puppeteer_base::peripheral_hook::{
     IBCTransferReason, ReceiverExecuteMsg, ResponseAnswer, ResponseHookErrorMsg, ResponseHookMsg,
     ResponseHookSuccessMsg, Transaction,
@@ -21,7 +22,7 @@ use drop_staking_base::msg::native_bond_provider::{
 };
 use drop_staking_base::state::native_bond_provider::{
     Config, ConfigOptional, ReplyMsg, TxState, TxStateStatus, CONFIG, LAST_PUPPETEER_RESPONSE,
-    NON_STAKED_BALANCE, TX_STATE,
+    NON_STAKED_BALANCE, PAUSE, TX_STATE,
 };
 use neutron_sdk::bindings::msg::NeutronMsg;
 use neutron_sdk::bindings::query::NeutronQuery;
@@ -54,7 +55,7 @@ pub fn instantiate(
         timeout: msg.timeout,
     };
     CONFIG.save(deps.storage, config)?;
-
+    PAUSE.save(deps.storage, &false)?;
     NON_STAKED_BALANCE.save(deps.storage, &Uint128::zero())?;
     TX_STATE.save(deps.storage, &TxState::default())?;
 
@@ -95,6 +96,7 @@ pub fn query(deps: Deps<NeutronQuery>, env: Env, msg: QueryMsg) -> ContractResul
         QueryMsg::LastPuppeteerResponse {} => Ok(to_json_binary(&LastPuppeteerResponse {
             response: LAST_PUPPETEER_RESPONSE.may_load(deps.storage)?,
         })?),
+        QueryMsg::Pause {} => Ok(to_json_binary(&PAUSE.load(deps.storage)?)?),
         QueryMsg::CanBeRemoved {} => query_can_be_removed(deps, env),
     }
 }
@@ -211,9 +213,26 @@ pub fn execute(
         ExecuteMsg::Bond {} => execute_bond(deps, info),
         ExecuteMsg::ProcessOnIdle {} => execute_process_on_idle(deps, env, info),
         ExecuteMsg::PeripheralHook(msg) => execute_puppeteer_hook(deps, env, info, *msg),
+        ExecuteMsg::SetPause(pause) => execute_set_pause(deps.into_empty(), info, pause),
     }
 }
 
+fn execute_set_pause(
+    deps: DepsMut,
+    info: MessageInfo,
+    pause: bool,
+) -> ContractResult<Response<NeutronMsg>> {
+    cw_ownable::assert_owner(deps.storage, &info.sender)?;
+
+    PAUSE.save(deps.storage, &pause)?;
+
+    Ok(response(
+        "execute-set-pause",
+        CONTRACT_NAME,
+        [("process_on_idle", pause.to_string())],
+    ))
+}
+
 fn execute_update_config(
     deps: DepsMut<NeutronQuery>,
     info: MessageInfo,
@@ -287,6 +306,10 @@ fn execute_process_on_idle(
     env: Env,
     info: MessageInfo,
 ) -> ContractResult<Response<NeutronMsg>> {
+    if PAUSE.load(deps.as_ref().storage)? {
+        Err(ContractError::PauseError(PauseError::Paused {}))?
+    }
+
     let config = CONFIG.load(deps.storage)?;
     let addrs = get_contracts!(deps, config.factory_contract, core_contract);
 

contracts/native-bond-provider/src/tests.rs

@@ -8,10 +8,11 @@ use cw_ownable::{Action, Ownership};
 use cw_utils::PaymentError;
 use drop_helpers::{
     ica::IcaState,
+    pause::PauseError,
     testing::{mock_dependencies, mock_state_query},
 };
 use drop_staking_base::state::native_bond_provider::{
-    Config, ConfigOptional, ReplyMsg, TxState, CONFIG, NON_STAKED_BALANCE, TX_STATE,
+    Config, ConfigOptional, ReplyMsg, TxState, CONFIG, NON_STAKED_BALANCE, PAUSE, TX_STATE,
 };
 use neutron_sdk::{
     bindings::msg::{IbcFee, NeutronMsg},
@@ -278,6 +279,8 @@ fn query_can_bond_false() {
 fn query_can_not_process_on_idle_not_in_idle_state() {
     let mut deps = mock_dependencies(&[]);
 
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
     CONFIG
         .save(deps.as_mut().storage, &get_default_config())
         .unwrap();
@@ -566,11 +569,60 @@ fn update_ownership() {
     );
 }
 
+#[test]
+fn execute_set_pause() {
+    let mut deps = mock_dependencies(&[]);
+    let deps_mut = deps.as_mut();
+    cw_ownable::initialize_owner(deps_mut.storage, deps_mut.api, Some("owner")).unwrap();
+
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
+    let res = crate::contract::execute(
+        deps.as_mut(),
+        mock_env(),
+        mock_info("owner", &[]),
+        drop_staking_base::msg::native_bond_provider::ExecuteMsg::SetPause(true),
+    )
+    .unwrap();
+
+    assert_eq!(
+        res,
+        Response::new().add_event(
+            Event::new("crates.io:drop-staking__drop-native-bond-provider-execute-set-pause")
+                .add_attributes(vec![attr("process_on_idle", "true")])
+        )
+    );
+    assert!(PAUSE.load(deps.as_ref().storage).unwrap())
+}
+
+#[test]
+fn execute_process_on_idle_paused() {
+    let mut deps = mock_dependencies(&[]);
+    PAUSE.save(deps.as_mut().storage, &true).unwrap();
+
+    let error = crate::contract::execute(
+        deps.as_mut(),
+        mock_env(),
+        mock_info("owner", &[]),
+        drop_staking_base::msg::native_bond_provider::ExecuteMsg::ProcessOnIdle {},
+    )
+    .unwrap_err();
+
+    assert_eq!(
+        error,
+        drop_staking_base::error::native_bond_provider::ContractError::PauseError(
+            PauseError::Paused {}
+        )
+    )
+}
+
 #[test]
 fn process_on_idle_not_in_idle_state() {
     let mut deps = mock_dependencies(&[]);
     mock_state_query(&mut deps);
 
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
     CONFIG
         .save(deps.as_mut().storage, &get_default_config())
         .unwrap();
@@ -612,6 +664,8 @@ fn process_on_idle_not_core_contract() {
     let mut deps = mock_dependencies(&[]);
     mock_state_query(&mut deps);
 
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
     CONFIG
         .save(deps.as_mut().storage, &get_default_config())
         .unwrap();
@@ -635,6 +689,8 @@ fn process_on_idle_delegation() {
     let mut deps = mock_dependencies(&[]);
     mock_state_query(&mut deps);
 
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
     CONFIG
         .save(deps.as_mut().storage, &get_default_config())
         .unwrap();
@@ -691,6 +747,8 @@ fn process_on_idle_ibc_transfer() {
     let mut deps = mock_dependencies(&[]);
     mock_state_query(&mut deps);
 
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
     CONFIG
         .save(deps.as_mut().storage, &get_default_config())
         .unwrap();
@@ -777,6 +835,8 @@ fn process_on_idle_not_allowed_if_no_funds() {
     let mut deps = mock_dependencies(&[]);
     mock_state_query(&mut deps);
 
+    PAUSE.save(deps.as_mut().storage, &false).unwrap();
+
     CONFIG
         .save(deps.as_mut().storage, &get_default_config())
         .unwrap();