Skip to content

Commit aee2fe9

Browse files
authored
chore: add test for checking pause control (#19)
1 parent 61ae8e2 commit aee2fe9

File tree

2 files changed

+38
-1
lines changed

2 files changed

+38
-1
lines changed

eth-connector/src/admin_controlled.rs

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,3 +84,40 @@ pub mod error {
8484
}
8585
}
8686
}
87+
88+
#[test]
89+
fn test_pause_control() {
90+
use crate::connector_impl::EthConnector;
91+
92+
let mut connector = EthConnector {
93+
prover_account: "prover".parse().unwrap(),
94+
eth_custodian_address: Default::default(),
95+
paused_mask: UNPAUSE_ALL,
96+
account_with_access_right: "aurora".parse().unwrap(),
97+
owner_id: "aurora".parse().unwrap(),
98+
};
99+
100+
assert!(connector.assert_not_paused(PAUSE_DEPOSIT).is_ok());
101+
assert!(connector.assert_not_paused(PAUSE_WITHDRAW).is_ok());
102+
103+
connector.set_paused_flags(PAUSE_DEPOSIT);
104+
105+
assert!(connector.assert_not_paused(PAUSE_DEPOSIT).is_err());
106+
assert!(connector.assert_not_paused(PAUSE_WITHDRAW).is_ok());
107+
108+
connector.set_paused_flags(UNPAUSE_ALL);
109+
connector.set_paused_flags(PAUSE_WITHDRAW);
110+
111+
assert!(connector.assert_not_paused(PAUSE_DEPOSIT).is_ok());
112+
assert!(connector.assert_not_paused(PAUSE_WITHDRAW).is_err());
113+
114+
connector.set_paused_flags(PAUSE_WITHDRAW | PAUSE_DEPOSIT);
115+
116+
assert!(connector.assert_not_paused(PAUSE_DEPOSIT).is_err());
117+
assert!(connector.assert_not_paused(PAUSE_WITHDRAW).is_err());
118+
119+
connector.set_paused_flags(UNPAUSE_ALL);
120+
121+
assert!(connector.assert_not_paused(PAUSE_DEPOSIT).is_ok());
122+
assert!(connector.assert_not_paused(PAUSE_WITHDRAW).is_ok());
123+
}

eth-connector/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -480,7 +480,7 @@ impl ConnectorWithdraw for EthConnectorContract {
480480
self.assert_access_right().sdk_unwrap();
481481
assert_one_yocto();
482482

483-
// Check is current flow paused. If it's owner just skip asserrion.
483+
// Check is current flow paused. If it's owner just skip assertion.
484484
self.assert_not_paused(PAUSE_WITHDRAW)
485485
.map_err(|_| "WithdrawErrorPaused")
486486
.sdk_unwrap();

0 commit comments

Comments
 (0)