cedros-login-server 0.0.39

//! Tests for session repository

use crate::repositories::{InMemorySessionRepository, SessionEntity, SessionRepository};
use chrono::{Duration, Utc};
use uuid::Uuid;

#[tokio::test]
async fn test_create_and_find_session() {
    let repo = InMemorySessionRepository::new();
    let user_id = Uuid::new_v4();
    let session = SessionEntity::new(
        user_id,
        "refresh_hash_123".to_string(),
        Utc::now() + Duration::days(30),
        Some("127.0.0.1".to_string()),
        Some("Mozilla/5.0".to_string()),
    );
    let session_id = session.id;

    repo.create(session).await.unwrap();

    let found = repo.find_by_id(session_id).await.unwrap();
    assert!(found.is_some());
    assert_eq!(found.unwrap().user_id, user_id);
}

#[tokio::test]
async fn test_find_by_refresh_token() {
    let repo = InMemorySessionRepository::new();
    let session = SessionEntity::new(
        Uuid::new_v4(),
        "unique_refresh_hash".to_string(),
        Utc::now() + Duration::days(30),
        None,
        None,
    );

    repo.create(session.clone()).await.unwrap();

    let found = repo
        .find_by_refresh_token("unique_refresh_hash")
        .await
        .unwrap();
    assert!(found.is_some());
    assert_eq!(found.unwrap().id, session.id);
}

#[tokio::test]
async fn test_revoke_session() {
    let repo = InMemorySessionRepository::new();
    let session = SessionEntity::new(
        Uuid::new_v4(),
        "hash".to_string(),
        Utc::now() + Duration::days(30),
        None,
        None,
    );
    let session_id = session.id;

    repo.create(session).await.unwrap();
    assert!(!repo.is_revoked(session_id).await.unwrap());

    repo.revoke(session_id).await.unwrap();
    assert!(repo.is_revoked(session_id).await.unwrap());
}

#[tokio::test]
async fn test_revoke_with_reason_sets_reason() {
    let repo = InMemorySessionRepository::new();
    let session = SessionEntity::new(
        Uuid::new_v4(),
        "hash".to_string(),
        Utc::now() + Duration::days(30),
        None,
        None,
    );
    let session_id = session.id;

    repo.create(session).await.unwrap();
    repo.revoke_with_reason(session_id, "logout").await.unwrap();

    let found = repo.find_by_id(session_id).await.unwrap().unwrap();
    assert_eq!(found.revoked_reason.as_deref(), Some("logout"));
}

#[tokio::test]
async fn test_session_validity() {
    let valid_session = SessionEntity::new(
        Uuid::new_v4(),
        "hash".to_string(),
        Utc::now() + Duration::days(30),
        None,
        None,
    );
    assert!(valid_session.is_valid());

    let expired_session = SessionEntity::new(
        Uuid::new_v4(),
        "hash".to_string(),
        Utc::now() - Duration::days(1),
        None,
        None,
    );
    assert!(!expired_session.is_valid());
}

#[tokio::test]
async fn test_find_active_by_user_id() {
    let repo = InMemorySessionRepository::new();
    let user_id = Uuid::new_v4();

    let active_session = SessionEntity::new(
        user_id,
        "active_hash".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    let revoked_session = SessionEntity::new(
        user_id,
        "revoked_hash".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    let expired_session = SessionEntity::new(
        user_id,
        "expired_hash".to_string(),
        Utc::now() - Duration::days(1),
        None,
        None,
    );

    let revoked_id = revoked_session.id;
    repo.create(active_session).await.unwrap();
    repo.create(revoked_session).await.unwrap();
    repo.create(expired_session).await.unwrap();
    repo.revoke(revoked_id).await.unwrap();

    let sessions = repo.find_active_by_user_id(user_id).await.unwrap();
    assert_eq!(sessions.len(), 1);
    assert!(sessions[0].is_valid());
}

#[tokio::test]
async fn test_find_active_by_user_id_paged() {
    let repo = InMemorySessionRepository::new();
    let user_id = Uuid::new_v4();

    let mut session1 = SessionEntity::new(
        user_id,
        "hash1".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    session1.created_at = Utc::now() - Duration::minutes(3);

    let mut session2 = SessionEntity::new(
        user_id,
        "hash2".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    session2.created_at = Utc::now() - Duration::minutes(2);

    let mut session3 = SessionEntity::new(
        user_id,
        "hash3".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    session3.created_at = Utc::now() - Duration::minutes(1);

    repo.create(session1.clone()).await.unwrap();
    repo.create(session2.clone()).await.unwrap();
    repo.create(session3.clone()).await.unwrap();

    let sessions = repo
        .find_active_by_user_id_paged(user_id, 2, 1)
        .await
        .unwrap();
    assert_eq!(sessions.len(), 2);
    assert_eq!(sessions[0].id, session2.id);
    assert_eq!(sessions[1].id, session1.id);
}

#[tokio::test]
async fn test_count_active_by_user_id() {
    let repo = InMemorySessionRepository::new();
    let user_id = Uuid::new_v4();

    let active_session = SessionEntity::new(
        user_id,
        "active".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    let expired_session = SessionEntity::new(
        user_id,
        "expired".to_string(),
        Utc::now() - Duration::days(1),
        None,
        None,
    );

    repo.create(active_session).await.unwrap();
    repo.create(expired_session).await.unwrap();

    let count = repo.count_active_by_user_id(user_id).await.unwrap();
    assert_eq!(count, 1);
}

#[tokio::test]
async fn test_find_recent_by_user_id() {
    let repo = InMemorySessionRepository::new();
    let user_id = Uuid::new_v4();

    let mut session1 = SessionEntity::new(
        user_id,
        "hash1".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    session1.created_at = Utc::now() - Duration::minutes(2);

    let mut session2 = SessionEntity::new(
        user_id,
        "hash2".to_string(),
        Utc::now() + Duration::days(1),
        None,
        None,
    );
    session2.created_at = Utc::now() - Duration::minutes(1);

    repo.create(session1.clone()).await.unwrap();
    repo.create(session2.clone()).await.unwrap();

    let sessions = repo.find_recent_by_user_id(user_id, 1).await.unwrap();
    assert_eq!(sessions.len(), 1);
    assert_eq!(sessions[0].id, session2.id);
}

#[tokio::test]
async fn test_revoke_all_except() {
    let repo = InMemorySessionRepository::new();
    let user_id = Uuid::new_v4();

    let keep = SessionEntity::new(
        user_id,
        "keep_hash".to_string(),
        Utc::now() + Duration::days(30),
        None,
        None,
    );
    let other = SessionEntity::new(
        user_id,
        "other_hash".to_string(),
        Utc::now() + Duration::days(30),
        None,
        None,
    );

    let keep_id = keep.id;
    let other_id = other.id;
    repo.create(keep).await.unwrap();
    repo.create(other).await.unwrap();

    let revoked = repo.revoke_all_except(user_id, keep_id).await.unwrap();
    assert_eq!(revoked, 1);
    assert!(!repo.is_revoked(keep_id).await.unwrap());
    assert!(repo.is_revoked(other_id).await.unwrap());
}