feat: db port&pg adapter: get verification otp
This commit is contained in:
parent
1565e56737
commit
1be035be98
3 changed files with 112 additions and 0 deletions
|
@ -0,0 +1,67 @@
|
||||||
|
// SPDX-FileCopyrightText: 2024 Aravinth Manivannan <realaravinth@batsense.net>
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
|
||||||
|
use super::errors::*;
|
||||||
|
use super::DBOutPostgresAdapter;
|
||||||
|
use crate::identity::application::port::output::db::{errors::*, get_verification_secret::*};
|
||||||
|
|
||||||
|
#[async_trait::async_trait]
|
||||||
|
impl GetVerificationSecretOutDBPort for DBOutPostgresAdapter {
|
||||||
|
async fn get_verification_secret(&self, username: &str) -> OutDBPortResult<String> {
|
||||||
|
struct Secret {
|
||||||
|
secret: String,
|
||||||
|
}
|
||||||
|
let res = sqlx::query_as!(
|
||||||
|
Secret,
|
||||||
|
"SELECT
|
||||||
|
secret
|
||||||
|
FROM
|
||||||
|
verification_otp
|
||||||
|
WHERE
|
||||||
|
username = $1
|
||||||
|
AND
|
||||||
|
purpose = $2;",
|
||||||
|
username,
|
||||||
|
REGISTRATION_SECRET_PURPOSE,
|
||||||
|
)
|
||||||
|
.fetch_one(&self.pool)
|
||||||
|
.await
|
||||||
|
.map_err(|e| map_row_not_found_err(e, OutDBPortError::VerificationOTPSecretNotFound))?;
|
||||||
|
Ok(res.secret)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use crate::identity::application::port::output::db::create_verification_secret::*;
|
||||||
|
|
||||||
|
#[actix_rt::test]
|
||||||
|
async fn test_postgres_get_verification_secret() {
|
||||||
|
let username = "batman";
|
||||||
|
let secret = "bsdasdf";
|
||||||
|
let settings = crate::settings::tests::get_settings().await;
|
||||||
|
let db = super::DBOutPostgresAdapter::new(
|
||||||
|
sqlx::postgres::PgPool::connect(&settings.database.url)
|
||||||
|
.await
|
||||||
|
.unwrap(),
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
db.get_verification_secret(username).await.err(),
|
||||||
|
Some(OutDBPortError::VerificationOTPSecretNotFound)
|
||||||
|
);
|
||||||
|
|
||||||
|
let create_msg = CreateSecretMsgBuilder::default()
|
||||||
|
.secret(secret.into())
|
||||||
|
.username(username.into())
|
||||||
|
.build()
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
db.create_verification_secret(create_msg).await.unwrap();
|
||||||
|
|
||||||
|
assert_eq!(db.get_verification_secret(username).await.unwrap(), secret);
|
||||||
|
|
||||||
|
settings.drop_db().await;
|
||||||
|
}
|
||||||
|
}
|
|
@ -13,6 +13,7 @@ pub mod create_verification_secret;
|
||||||
pub mod delete_verification_secret;
|
pub mod delete_verification_secret;
|
||||||
pub mod email_exists;
|
pub mod email_exists;
|
||||||
mod errors;
|
mod errors;
|
||||||
|
pub mod get_verification_secret;
|
||||||
pub mod username_exists;
|
pub mod username_exists;
|
||||||
pub mod verification_secret_exists;
|
pub mod verification_secret_exists;
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,44 @@
|
||||||
|
// SPDX-FileCopyrightText: 2024 Aravinth Manivannan <realaravinth@batsense.net>
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
||||||
|
|
||||||
|
use mockall::predicate::*;
|
||||||
|
use mockall::*;
|
||||||
|
|
||||||
|
pub use super::create_verification_secret::REGISTRATION_SECRET_PURPOSE;
|
||||||
|
use super::errors::*;
|
||||||
|
#[cfg(test)]
|
||||||
|
#[allow(unused_imports)]
|
||||||
|
pub use tests::*;
|
||||||
|
|
||||||
|
#[automock]
|
||||||
|
#[async_trait::async_trait]
|
||||||
|
pub trait GetVerificationSecretOutDBPort: Send + Sync {
|
||||||
|
async fn get_verification_secret(&self, username: &str) -> OutDBPortResult<String>;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub type GetVerificationSecretOutDBPortObj = std::sync::Arc<dyn GetVerificationSecretOutDBPort>;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
pub mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
pub fn mock_get_verification_secret_db_port(
|
||||||
|
times: Option<usize>,
|
||||||
|
returning_secret: String,
|
||||||
|
) -> GetVerificationSecretOutDBPortObj {
|
||||||
|
let mut m = MockGetVerificationSecretOutDBPort::new();
|
||||||
|
if let Some(times) = times {
|
||||||
|
m.expect_get_verification_secret()
|
||||||
|
.times(times)
|
||||||
|
.returning(move |_| Ok(returning_secret.clone()));
|
||||||
|
} else {
|
||||||
|
m.expect_get_verification_secret()
|
||||||
|
.returning(move |_| Ok(returning_secret.clone()));
|
||||||
|
}
|
||||||
|
|
||||||
|
Arc::new(m)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue