test: cascading in-memory store

This commit is contained in:
2026-10-25 22:19:24 +01:00
parent 51d92e08a7
commit 06eec2feae
72 changed files with 2439 additions and 1627 deletions

View File

@@ -0,0 +1,58 @@
use async_trait::async_trait;
use chrono::Utc;
use crate::{
errors::DomainError, models::RefreshSession, ports::RefreshSessionRepository,
value_objects::UserId,
};
use super::InMemoryStore;
#[async_trait]
impl RefreshSessionRepository for InMemoryStore {
/// `refresh_sessions.token` is UNIQUE.
async fn create(&self, session: &RefreshSession) -> Result<(), DomainError> {
let mut sessions = self.refresh_sessions.write().unwrap();
if sessions.iter().any(|s| s.token == session.token) {
return Err(DomainError::InfrastructureError(
"Database operation failed".into(),
));
}
sessions.push(session.clone());
Ok(())
}
async fn get_by_token(&self, token: &str) -> Result<Option<RefreshSession>, DomainError> {
Ok(self
.refresh_sessions
.read()
.unwrap()
.iter()
.find(|s| s.token == token)
.cloned())
}
async fn revoke(&self, token: &str) -> Result<(), DomainError> {
self.refresh_sessions
.write()
.unwrap()
.retain(|s| s.token != token);
Ok(())
}
async fn revoke_all_for_user(&self, user_id: &UserId) -> Result<(), DomainError> {
self.refresh_sessions
.write()
.unwrap()
.retain(|s| s.user_id != *user_id);
Ok(())
}
async fn delete_expired(&self) -> Result<u64, DomainError> {
let mut store = self.refresh_sessions.write().unwrap();
let before = store.len();
let now = Utc::now();
store.retain(|s| s.expires_at >= now);
Ok((before - store.len()) as u64)
}
}