ForgeFlux/ForgeFlux
2
0
Fork 0
Code Issues 1 Pull requests Packages Projects Releases Wiki Activity

feat: rm update_oauth_access_token and implement delete_oauth_access_token

Browse source
This commit is contained in:
Aravinth Manivannan 2024-05-12 21:20:13 +05:30
parent ed2425699c
commit 9437d50831
Signed by: realaravinth
GPG key ID: F8F50389936984FF
4 changed files with 105 additions and 132 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
src/auth/adapter/out/db/postgres/delete_oauth_access_token.rs Normal file
View file

@ -0,0 +1,70 @@
use super::DBOutPostgresAdapter;
use crate::auth::application::port::out::db::{
delete_oauth_access_token::DeleteOAuthAccessToken, errors::OutDBPortResult,
};
#[async_trait::async_trait]
impl DeleteOAuthAccessToken for DBOutPostgresAdapter {
async fn delete_oauth_access_token(
&self,
username: &str,
oauth_provider: &str,
) -> OutDBPortResult<()> {
sqlx::query!(
"
DELETE FROM
oauth_access_token
WHERE
username = $1
AND
oauth_provider = $2;",
username,
oauth_provider
)
.execute(&self.pool)
.await?;
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::auth::application::port::out::db::{
get_oauth_access_token::GetOAuthAccessToken, save_oauth_access_token::SaveOAuthAccessToken,
};
use crate::auth::domain::OAuthAccessToken;
#[actix_rt::test]
async fn test_postgres_delete_oauth_access_token() {
let oauth_provider = "oauthprovitestpostgres";
let username = "test_postgres_save_oauth_access_token";
let old_access_token = OAuthAccessToken {
refresh_token: "asdfasdfarefresh".into(),
..Default::default()
};
let settings = crate::settings::tests::get_settings().await;
let db = super::DBOutPostgresAdapter::new(
sqlx::postgres::PgPool::connect(&settings.database.url)
.await
.unwrap(),
);
db.save_oauth_access_token(username, oauth_provider, &old_access_token)
.await
.unwrap();
db.delete_oauth_access_token(username, oauth_provider)
.await
.unwrap();
assert!(db
.get_oauth_access_token(username, oauth_provider)
.await
.is_err());
settings.drop_db().await;
}
}

95
src/auth/adapter/out/db/postgres/update_oauth_access_token.rs
View file

@ -1,95 +0,0 @@
use super::DBOutPostgresAdapter;
use crate::auth::application::port::out::db::{
errors::OutDBPortResult, update_oauth_access_token::UpdateOAuthAccessToken,
};
use crate::auth::domain::OAuthAccessToken;
#[async_trait::async_trait]
impl UpdateOAuthAccessToken for DBOutPostgresAdapter {
async fn update_oauth_access_token(
&self,
old_oauth_access_token: &OAuthAccessToken,
new_oauth_access_token: &OAuthAccessToken,
) -> OutDBPortResult<()> {
sqlx::query!(
"
UPDATE
oauth_access_token
SET
access_token = $1,
token_type = $2,
expires_in = $3,
refresh_token = $4,
created_at = $5
WHERE
access_token = $6
AND
token_type = $7
AND
expires_in = $8
AND
refresh_token = $9;",
&new_oauth_access_token.access_token,
&new_oauth_access_token.token_type,
new_oauth_access_token.expires_in as i64,
&new_oauth_access_token.refresh_token,
&new_oauth_access_token.created_at,
&old_oauth_access_token.access_token,
&old_oauth_access_token.token_type,
old_oauth_access_token.expires_in as i64,
&old_oauth_access_token.refresh_token,
)
.execute(&self.pool)
.await?;
Ok(())
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::auth::application::port::out::db::{
get_oauth_access_token::GetOAuthAccessToken, save_oauth_access_token::SaveOAuthAccessToken,
};
#[actix_rt::test]
async fn test_postgres_update_oauth_access_token() {
let oauth_provider = "oauthprovitestpostgres";
let username = "test_postgres_save_oauth_access_token";
let old_access_token = OAuthAccessToken {
refresh_token: "asdfasdfarefresh".into(),
..Default::default()
};
let settings = crate::settings::tests::get_settings().await;
let db = super::DBOutPostgresAdapter::new(
sqlx::postgres::PgPool::connect(&settings.database.url)
.await
.unwrap(),
);
db.save_oauth_access_token(username, oauth_provider, &old_access_token)
.await
.unwrap();
let new_access_token = OAuthAccessToken {
refresh_token: "newrefreshtoken".into(),
access_token: "newreaccesstoken".into(),
..Default::default()
};
db.update_oauth_access_token(&old_access_token, &new_access_token)
.await
.unwrap();
assert_eq!(
db.get_oauth_access_token(username, oauth_provider)
.await
.unwrap(),
new_access_token
);
settings.drop_db().await;
}
}

35
src/auth/application/port/out/db/delete_oauth_access_token.rs Normal file
View file

@ -0,0 +1,35 @@
use mockall::predicate::*;
use mockall::*;
use super::errors::*;
#[automock]
#[async_trait::async_trait]
pub trait DeleteOAuthAccessToken: Send + Sync {
async fn delete_oauth_access_token(
&self,
username: &str,
oauth_provider: &str,
) -> OutDBPortResult<()>;
}
#[cfg(test)]
pub mod tests {
use super::*;
use std::sync::Arc;
pub fn mock_delete_oauth_access_token(times: Option<usize>) -> Arc<dyn DeleteOAuthAccessToken> {
let mut m = MockDeleteOAuthAccessToken::default();
if let Some(times) = times {
m.expect_delete_oauth_access_token()
.times(times)
.returning(move |_, _| Ok(()));
} else {
m.expect_delete_oauth_access_token()
.returning(move |_, _| Ok(()));
}
Arc::new(m)
}
}

37
src/auth/application/port/out/db/update_oauth_access_token.rs
View file

@ -1,37 +0,0 @@
use mockall::predicate::*;
use mockall::*;
use super::errors::*;
use crate::auth::domain::OAuthAccessToken;
#[automock]
#[async_trait::async_trait]
pub trait UpdateOAuthAccessToken: Send + Sync {
async fn update_oauth_access_token(
&self,
old_oauth_access_token: &OAuthAccessToken,
new_oauth_access_token: &OAuthAccessToken,
) -> OutDBPortResult<()>;
}
#[cfg(test)]
pub mod tests {
use super::*;
use std::sync::Arc;
pub fn mock_update_oauth_access_token(times: Option<usize>) -> Arc<dyn UpdateOAuthAccessToken> {
let mut m = MockUpdateOAuthAccessToken::default();
if let Some(times) = times {
m.expect_update_oauth_access_token()
.times(times)
.returning(move |_, _| Ok(()));
} else {
m.expect_update_oauth_access_token()
.returning(move |_, _| Ok(()));
}
Arc::new(m)
}
}