Add trailing zeros for discriminators with tests
This commit is contained in:
parent
74b0e6141a
commit
9e0371b6ac
1 changed files with 26 additions and 1 deletions
27
src/utils.rs
27
src/utils.rs
|
@ -22,7 +22,32 @@ impl User for serenity::User {
|
|||
if self.discriminator == 0 {
|
||||
return self.name.clone();
|
||||
}
|
||||
format!("{}#{}", self.name, self.discriminator)
|
||||
format!("{}#{:0>4}", self.name, self.discriminator)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use crate::serenity::User;
|
||||
use crate::utils::User as UserTrait;
|
||||
|
||||
fn test_user(name: &str, discriminator: u16) -> User {
|
||||
let mut user = User::default();
|
||||
user.name = name.to_owned();
|
||||
user.discriminator = discriminator;
|
||||
user
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_legacy_username() {
|
||||
let user = test_user("test", 123);
|
||||
assert_eq!(user.username(), "test#0123");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_new_username() {
|
||||
let user = test_user("test", 0);
|
||||
assert_eq!(user.username(), "test");
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue