jax-daemon 0.1.13

//! Integration tests for FUSE mount management
//!
//! These tests verify the mount configuration and lifecycle management without
//! requiring actual FUSE mounting (which requires privileges).

#![cfg(feature = "fuse")]

use tempfile::TempDir;
use uuid::Uuid;

use jax_daemon::{Database, FuseMount, MountStatus};

/// Create an in-memory test database
async fn setup_test_db() -> Database {
    let db_url = url::Url::parse("sqlite::memory:").unwrap();
    Database::connect(&db_url).await.unwrap()
}

#[tokio::test]
async fn test_create_and_get_mount() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();
    let mount_point = temp_dir.path().join("mount").to_string_lossy().to_string();
    std::fs::create_dir_all(&mount_point).unwrap();

    let bucket_id = Uuid::new_v4();

    let mount = FuseMount::create(
        bucket_id,
        &mount_point,
        false,
        false,
        Some(50),
        Some(30),
        &db,
    )
    .await
    .unwrap();

    assert_eq!(*mount.bucket_id, bucket_id);
    assert_eq!(mount.mount_point, mount_point);
    assert!(!*mount.auto_mount);
    assert!(!*mount.read_only);
    assert_eq!(mount.cache_size_mb, 50);
    assert_eq!(mount.cache_ttl_secs, 30);
    assert_eq!(mount.status, MountStatus::Stopped);
    assert!(*mount.enabled);

    // Get the mount by ID
    let retrieved = FuseMount::get(*mount.mount_id, &db).await.unwrap().unwrap();
    assert_eq!(*retrieved.mount_id, *mount.mount_id);
    assert_eq!(*retrieved.bucket_id, bucket_id);
}

#[tokio::test]
async fn test_list_mounts() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();

    // Create multiple mounts
    for i in 0..3 {
        let mount_point = temp_dir
            .path()
            .join(format!("mount{}", i))
            .to_string_lossy()
            .to_string();
        std::fs::create_dir_all(&mount_point).unwrap();

        FuseMount::create(
            Uuid::new_v4(),
            &mount_point,
            i == 1, // Only middle one is auto-mount
            false,
            None,
            None,
            &db,
        )
        .await
        .unwrap();
    }

    let mounts = FuseMount::list(&db).await.unwrap();
    assert_eq!(mounts.len(), 3);
}

#[tokio::test]
async fn test_update_mount() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();
    let mount_point = temp_dir.path().join("mount").to_string_lossy().to_string();
    std::fs::create_dir_all(&mount_point).unwrap();

    let mount = FuseMount::create(Uuid::new_v4(), &mount_point, false, false, None, None, &db)
        .await
        .unwrap();
    assert!(!*mount.auto_mount);
    assert!(!*mount.read_only);

    // Update the mount
    let updated = FuseMount::update(
        *mount.mount_id,
        None,
        Some(false),
        Some(true),
        Some(true),
        Some(200),
        Some(120),
        &db,
    )
    .await
    .unwrap()
    .unwrap();
    assert!(!*updated.enabled);
    assert!(*updated.auto_mount);
    assert!(*updated.read_only);
    assert_eq!(updated.cache_size_mb, 200);
    assert_eq!(updated.cache_ttl_secs, 120);
}

#[tokio::test]
async fn test_delete_mount() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();
    let mount_point = temp_dir.path().join("mount").to_string_lossy().to_string();
    std::fs::create_dir_all(&mount_point).unwrap();

    let mount = FuseMount::create(Uuid::new_v4(), &mount_point, false, false, None, None, &db)
        .await
        .unwrap();

    // Verify mount exists
    assert!(FuseMount::get(*mount.mount_id, &db)
        .await
        .unwrap()
        .is_some());

    // Delete the mount
    let deleted = FuseMount::delete(*mount.mount_id, &db).await.unwrap();
    assert!(deleted);

    // Verify mount is gone
    assert!(FuseMount::get(*mount.mount_id, &db)
        .await
        .unwrap()
        .is_none());

    // Deleting again should return false
    let deleted_again = FuseMount::delete(*mount.mount_id, &db).await.unwrap();
    assert!(!deleted_again);
}

#[tokio::test]
async fn test_update_mount_status() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();
    let mount_point = temp_dir.path().join("mount").to_string_lossy().to_string();
    std::fs::create_dir_all(&mount_point).unwrap();

    let mount = FuseMount::create(Uuid::new_v4(), &mount_point, false, false, None, None, &db)
        .await
        .unwrap();
    assert_eq!(mount.status, MountStatus::Stopped);

    // Update to starting
    FuseMount::update_status(*mount.mount_id, MountStatus::Starting, None, &db)
        .await
        .unwrap();
    let mount = FuseMount::get(*mount.mount_id, &db).await.unwrap().unwrap();
    assert_eq!(mount.status, MountStatus::Starting);
    assert!(mount.error_message.is_none());

    // Update to running
    FuseMount::update_status(*mount.mount_id, MountStatus::Running, None, &db)
        .await
        .unwrap();
    let mount = FuseMount::get(*mount.mount_id, &db).await.unwrap().unwrap();
    assert_eq!(mount.status, MountStatus::Running);

    // Update to error with message
    FuseMount::update_status(*mount.mount_id, MountStatus::Error, Some("Test error"), &db)
        .await
        .unwrap();
    let mount = FuseMount::get(*mount.mount_id, &db).await.unwrap().unwrap();
    assert_eq!(mount.status, MountStatus::Error);
    assert_eq!(mount.error_message.as_deref(), Some("Test error"));
}

#[tokio::test]
async fn test_get_auto_mount_list() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();

    // Create mounts with different auto_mount settings
    for (i, (auto_mount, enabled)) in [(true, true), (true, false), (false, true), (false, false)]
        .iter()
        .enumerate()
    {
        let mount_point = temp_dir
            .path()
            .join(format!("mount{}", i))
            .to_string_lossy()
            .to_string();
        std::fs::create_dir_all(&mount_point).unwrap();

        let mount = FuseMount::create(
            Uuid::new_v4(),
            &mount_point,
            *auto_mount,
            false,
            None,
            None,
            &db,
        )
        .await
        .unwrap();

        // Disable the mount if needed
        if !enabled {
            FuseMount::update(
                *mount.mount_id,
                None,
                Some(false),
                None,
                None,
                None,
                None,
                &db,
            )
            .await
            .unwrap();
        }
    }

    // Only mounts with auto_mount=true AND enabled=true should be returned
    let auto_mounts = FuseMount::auto_list(&db).await.unwrap();
    assert_eq!(auto_mounts.len(), 1);
    assert!(*auto_mounts[0].auto_mount);
    assert!(*auto_mounts[0].enabled);
}

#[tokio::test]
async fn test_get_mounts_by_bucket() {
    let db = setup_test_db().await;
    let temp_dir = TempDir::new().unwrap();
    let bucket_id = Uuid::new_v4();

    // Create multiple mounts for the same bucket
    for i in 0..2 {
        let mount_point = temp_dir
            .path()
            .join(format!("mount_b1_{}", i))
            .to_string_lossy()
            .to_string();
        std::fs::create_dir_all(&mount_point).unwrap();

        FuseMount::create(bucket_id, &mount_point, false, false, None, None, &db)
            .await
            .unwrap();
    }

    // Create a mount for a different bucket
    let other_bucket = Uuid::new_v4();
    let other_mount_point = temp_dir
        .path()
        .join("mount_other")
        .to_string_lossy()
        .to_string();
    std::fs::create_dir_all(&other_mount_point).unwrap();

    FuseMount::create(
        other_bucket,
        &other_mount_point,
        false,
        false,
        None,
        None,
        &db,
    )
    .await
    .unwrap();

    // Get mounts by bucket
    let bucket_mounts = FuseMount::by_bucket(bucket_id, &db).await.unwrap();
    assert_eq!(bucket_mounts.len(), 2);
    for mount in &bucket_mounts {
        assert_eq!(*mount.bucket_id, bucket_id);
    }

    let other_mounts = FuseMount::by_bucket(other_bucket, &db).await.unwrap();
    assert_eq!(other_mounts.len(), 1);
    assert_eq!(*other_mounts[0].bucket_id, other_bucket);
}