|
|
@@ -80,27 +80,9 @@ pub fn encode_base10(amount: u64, decimal_places: usize) -> String {
|
|
|
String::from_iter(&s).trim_end_matches('0').trim_end_matches('.').to_string()
|
|
|
}
|
|
|
|
|
|
-pub fn truncate(amount: u64, decimals: u16, token_decimals: u16) -> Result<u64> {
|
|
|
- let mut amount: Vec<char> = amount.to_string().chars().collect();
|
|
|
-
|
|
|
- if token_decimals > decimals {
|
|
|
- if amount.len() <= (token_decimals - decimals) as usize {
|
|
|
- return Ok(0)
|
|
|
- }
|
|
|
- amount.truncate(amount.len() - (token_decimals - decimals) as usize);
|
|
|
- }
|
|
|
-
|
|
|
- if token_decimals < decimals {
|
|
|
- amount.resize(amount.len() + (decimals - token_decimals) as usize, '0');
|
|
|
- }
|
|
|
-
|
|
|
- let amount = u64::from_str(&String::from_iter(amount))?;
|
|
|
- Ok(amount)
|
|
|
-}
|
|
|
-
|
|
|
#[cfg(test)]
|
|
|
mod tests {
|
|
|
- use super::{decode_base10, encode_base10, truncate};
|
|
|
+ use super::{decode_base10, encode_base10};
|
|
|
|
|
|
#[test]
|
|
|
fn test_decode_base10() {
|
|
|
@@ -119,55 +101,4 @@ mod tests {
|
|
|
assert_eq!("2343211", &encode_base10(2343211, 0));
|
|
|
assert_eq!("0.00002343", &encode_base10(2343, 8));
|
|
|
}
|
|
|
-
|
|
|
- #[test]
|
|
|
- fn test_truncate() {
|
|
|
- // Token decimals is equal to 8
|
|
|
- assert_eq!(100, truncate(100, 8, 8).unwrap());
|
|
|
- assert_eq!(12, truncate(12, 8, 8).unwrap());
|
|
|
-
|
|
|
- // Token decimals is bigger than 8
|
|
|
- assert_eq!(100000000, truncate(1000000000, 8, 9).unwrap());
|
|
|
- assert_eq!(10, truncate(100, 8, 9).unwrap());
|
|
|
- assert_eq!(1, truncate(12, 8, 9).unwrap());
|
|
|
- assert_eq!(10, truncate(102, 8, 9).unwrap());
|
|
|
- assert_eq!(0, truncate(1, 8, 9).unwrap());
|
|
|
- assert_eq!(1, truncate(100000000, 8, 16).unwrap());
|
|
|
- assert_eq!(10, truncate(100000000, 8, 15).unwrap());
|
|
|
- assert_eq!(0, truncate(100000000, 8, 17).unwrap());
|
|
|
- assert_eq!(0, truncate(10, 8, 16).unwrap());
|
|
|
-
|
|
|
- // Token decimals is less than 8
|
|
|
- assert_eq!(1000, truncate(100, 8, 7).unwrap());
|
|
|
- assert_eq!(12000, truncate(120, 8, 6).unwrap());
|
|
|
- assert_eq!(1000000, truncate(100, 8, 4).unwrap());
|
|
|
-
|
|
|
- // token decimals is 0
|
|
|
- assert_eq!(00000000, truncate(0, 8, 0).unwrap());
|
|
|
- assert_eq!(100000000, truncate(1, 8, 0).unwrap());
|
|
|
-
|
|
|
- //
|
|
|
- // reverse truncate
|
|
|
- //
|
|
|
-
|
|
|
- // Token decimals is less than decimals
|
|
|
- assert_eq!(1000000000, truncate(100000000, 9, 8).unwrap());
|
|
|
- assert_eq!(100000000, truncate(10000000, 9, 8).unwrap());
|
|
|
- assert_eq!(100, truncate(10, 9, 8).unwrap());
|
|
|
- assert_eq!(10, truncate(1, 9, 8).unwrap());
|
|
|
- assert_eq!(100, truncate(10, 9, 8).unwrap());
|
|
|
- assert_eq!(0, truncate(0, 9, 8).unwrap());
|
|
|
- assert_eq!(100000000, truncate(1, 16, 8).unwrap());
|
|
|
- assert_eq!(100000000, truncate(10, 15, 8).unwrap());
|
|
|
- assert_eq!(0, truncate(0, 17, 8).unwrap());
|
|
|
-
|
|
|
- // Token decimals is bigger than decimals
|
|
|
- assert_eq!(100, truncate(1000, 7, 8).unwrap());
|
|
|
- assert_eq!(120, truncate(12000, 6, 8).unwrap());
|
|
|
- assert_eq!(100, truncate(1000000, 4, 8).unwrap());
|
|
|
-
|
|
|
- // token decimals is 0
|
|
|
- assert_eq!(0, truncate(00000000, 0, 8).unwrap());
|
|
|
- assert_eq!(1, truncate(100000000, 0, 8).unwrap());
|
|
|
- }
|
|
|
}
|