matrix/authentication-service
1
0
Fork 0
mirror of https://github.com/matrix-org/matrix-authentication-service.git synced 2025-07-29 22:01:14 +03:00
Code Activity

storage: test compat {session, access token, refresh token} repositories

Browse Source
This commit is contained in:
Quentin Gliech
2023-01-16 19:27:43 +01:00
parent 2b0566ebe9
commit 62be962c4e
2 changed files with 322 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
crates/storage/src/upstream_oauth2/mod.rs
View File

@ -29,12 +29,12 @@ mod tests {
use sqlx::PgPool;
use super::*;
use crate::{Clock, PgRepository, Repository};
use crate::{user::UserRepository, Clock, PgRepository, Repository};
#[sqlx::test(migrator = "crate::MIGRATOR")]
async fn test_repository(pool: PgPool) -> Result<(), Box<dyn std::error::Error>> {
let mut rng = rand_chacha::ChaChaRng::seed_from_u64(42);
let clock = Clock::default();
let clock = Clock::mock();
let mut repo = PgRepository::from_pool(&pool).await?;
// The provider list should be empty at the start
@ -115,6 +115,12 @@ mod tests {
.upstream_oauth_session()
.complete_with_link(&clock, session, &link, None)
.await?;
// Reload the session
let session = repo
.upstream_oauth_session()
.lookup(session.id)
.await?
.expect("session to be found in the database");
assert!(session.is_completed());
assert!(!session.is_consumed());
assert_eq!(session.link_id(), Some(link.id));
@ -123,8 +129,29 @@ mod tests {
.upstream_oauth_session()
.consume(&clock, session)
.await?;
// Reload the session
let session = repo
.upstream_oauth_session()
.lookup(session.id)
.await?
.expect("session to be found in the database");
assert!(session.is_consumed());
let user = repo.user().add(&mut rng, &clock, "john".to_owned()).await?;
repo.upstream_oauth_link()
.associate_to_user(&link, &user)
.await?;
let links = repo
.upstream_oauth_link()
.list_paginated(&user, None, None, Some(10), None)
.await?;
assert!(!links.has_previous_page);
assert!(!links.has_next_page);
assert_eq!(links.edges.len(), 1);
assert_eq!(links.edges[0].id, link.id);
assert_eq!(links.edges[0].user_id, Some(user.id));
Ok(())
}
}