fix: return username to store in sessions when getting password

This commit is contained in:
realaravinth
2022-05-11 18:53:18 +05:30
parent d9b36179d1
commit 5bcf7beddc
3 changed files with 36 additions and 18 deletions

View File

@@ -89,6 +89,15 @@ pub enum Login<'a> {
Email(&'a str),
}
#[derive(Clone, Debug, Deserialize, Serialize, PartialEq)]
/// type encapsulating username and hashed password of a user
pub struct NameHash {
/// username
pub username: String,
/// hashed password
pub hash: String,
}
#[async_trait]
/// mCaptcha's database requirements. To implement support for $Database, kindly implement this
/// trait.
@@ -112,7 +121,7 @@ pub trait MCDatabase: std::marker::Send + std::marker::Sync + CloneSPDatabase {
async fn update_email(&self, p: &UpdateEmail) -> DBResult<()>;
/// get a user's password
async fn get_password(&self, l: &Login) -> DBResult<String>;
async fn get_password(&self, l: &Login) -> DBResult<NameHash>;
}
/// Trait to clone MCDatabase

View File

@@ -29,20 +29,23 @@ pub async fn database_works<'a, T: MCDatabase>(db: &T, p: &Register<'a>) {
}
db.register(p).await.unwrap();
assert_eq!(
db.get_password(&Login::Username(p.username)).await.unwrap(),
p.hash,
"user password matches"
);
// testing get_password
assert_eq!(
db.get_password(&Login::Email(p.email.as_ref().unwrap()))
.await
.unwrap(),
p.hash,
"user password matches"
);
// with username
let name_hash = db.get_password(&Login::Username(p.username)).await.unwrap();
assert_eq!(name_hash.hash, p.hash, "user password matches");
assert_eq!(name_hash.username, p.username, "username matches");
// with email
let name_hash = db
.get_password(&Login::Email(p.email.as_ref().unwrap()))
.await
.unwrap();
assert_eq!(name_hash.hash, p.hash, "user password matches");
assert_eq!(name_hash.username, p.username, "username matches");
// testing email exists
assert!(
db.email_exists(p.email.as_ref().unwrap()).await.unwrap(),
"user is registered so email should exsit"
@@ -70,11 +73,11 @@ pub async fn database_works<'a, T: MCDatabase>(db: &T, p: &Register<'a>) {
"user registration with email is deleted; so email shouldn't exsit"
);
// testing update email
let update_email = UpdateEmail {
username: p.username,
new_email: p.email.as_ref().unwrap(),
};
db.update_email(&update_email).await.unwrap();
println!(
"null user email: {}",