add stake config test

gemwalletcom · May 17, 2024 · ebc4fed · ebc4fed
1 parent 92c05e9
commit ebc4fed
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git 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
@@ -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
+        );
+    }
+}