diff --git a/crates/primitives/src/chain_stake.rs b/crates/primitives/src/chain_stake.rs index 9fd4691f..4c871db1 100644 --- a/crates/primitives/src/chain_stake.rs +++ b/crates/primitives/src/chain_stake.rs @@ -6,6 +6,7 @@ use typeshare::typeshare; #[derive(Copy, Clone, Debug, Serialize, Deserialize, EnumIter, AsRefStr, EnumString)] #[typeshare(swift = "Equatable, Codable, CaseIterable")] #[serde(rename_all = "lowercase")] +#[strum(serialize_all = "lowercase")] pub enum StakeChain { Cosmos, Osmosis, diff --git a/gemstone/src/config/stake.rs b/gemstone/src/config/stake.rs index 3adf7f58..440cbdd6 100644 --- a/gemstone/src/config/stake.rs +++ b/gemstone/src/config/stake.rs @@ -14,3 +14,23 @@ pub fn get_min_stake_amount(chain: &str) -> u64 { Err(_) => 0, } } + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_get_stake_lock_time() { + assert_eq!(get_stake_lock_time("sui"), 86400); + assert_eq!(get_stake_lock_time("smartchain"), 604800); + } + + #[test] + fn test_get_min_stake_amount() { + assert_eq!(get_min_stake_amount("sui"), 1_000_000_000); + assert_eq!( + get_min_stake_amount("smartchain"), + 1_000_000_000_000_000_000 + ); + } +}