Oxicloud/src/infrastructure/services/thumbnail_service_test.rs

use std::sync::Arc;
use std::time::Duration;

use bytes::Bytes;

use super::thumbnail_service::{ThumbnailService, ThumbnailSize};

/// Minimal valid 1x1 red PNG (68 bytes).
fn tiny_png() -> Vec<u8> {
    // Generated from a real 1×1 PNG — smallest valid RGBA image.
    let mut img = image::RgbaImage::new(1, 1);
    img.put_pixel(0, 0, image::Rgba([255, 0, 0, 255]));
    let mut buf = Vec::new();
    img.write_to(&mut std::io::Cursor::new(&mut buf), image::ImageFormat::Png)
        .expect("encode test PNG");
    buf
}

/// Regression test: thumbnail generation must work when the source file lives
/// at a blob-style path (`.blobs/ab/ab1234…`) rather than a logical path
/// (`folder/image.png`).  This broke after the blob storage migration
/// (commit 3c7c16f) because the handler passed the logical path — which
/// doesn't exist on disk — to the thumbnail service.
#[tokio::test]
async fn generate_thumbnail_from_blob_path() {
    let tmp = tempfile::tempdir().expect("create temp dir");
    let storage_root = tmp.path();

    // Simulate a blob-store layout: .blobs/ab/<hash>.blob
    let blob_dir = storage_root.join(".blobs").join("ab");
    std::fs::create_dir_all(&blob_dir).expect("create blob dir");
    let blob_path = blob_dir.join("ab1234567890.blob");
    std::fs::write(&blob_path, tiny_png()).expect("write test blob");

    let svc = Arc::new(ThumbnailService::new(
        storage_root,
        100,
        10 * 1024 * 1024,
        Some(Duration::from_secs(30)),
    ));
    svc.initialize().await.expect("init thumbnail dirs");

    // The key assertion: the service can read from a blob path (not a logical path)
    let result = svc
        .get_thumbnail(
            "test-file-id",
            "ab1234567890",
            ThumbnailSize::Icon,
            &blob_path,
        )
        .await;

    let thumb_bytes = result.expect("thumbnail generation should succeed from blob path");
    assert!(!thumb_bytes.is_empty(), "thumbnail bytes must not be empty");

    // Verify it's valid JPEG (starts with SOI marker 0xFF 0xD8)
    assert!(
        thumb_bytes.len() > 2 && thumb_bytes[0] == 0xFF && thumb_bytes[1] == 0xD8,
        "output should be JPEG format"
    );
}

/// Verify that a non-existent path produces an error, not a panic.
#[tokio::test]
async fn generate_thumbnail_nonexistent_path_returns_error() {
    let tmp = tempfile::tempdir().expect("create temp dir");
    let svc = Arc::new(ThumbnailService::new(
        tmp.path(),
        100,
        10 * 1024 * 1024,
        Some(Duration::from_secs(30)),
    ));
    svc.initialize().await.expect("init thumbnail dirs");

    let bad_path = tmp.path().join("does-not-exist.png");
    let result = svc
        .get_thumbnail(
            "missing-id",
            "nonexistent-hash",
            ThumbnailSize::Icon,
            &bad_path,
        )
        .await;

    assert!(result.is_err(), "should fail for nonexistent file");
}

/// Regression test: thumbnail generation must also work when the image source
/// is reconstructed from blob storage bytes instead of a single local file.
#[tokio::test]
async fn generate_thumbnail_from_blob_bytes() {
    let tmp = tempfile::tempdir().expect("create temp dir");
    let storage_root = tmp.path();

    let svc = Arc::new(ThumbnailService::new(
        storage_root,
        100,
        10 * 1024 * 1024,
        Some(Duration::from_secs(30)),
    ));
    svc.initialize().await.expect("init thumbnail dirs");

    let result = svc
        .get_thumbnail_from_bytes(
            "bytes-file-id",
            "bytes-hash-123",
            ThumbnailSize::Preview,
            Bytes::from(tiny_png()),
        )
        .await;

    let thumb_bytes = result.expect("thumbnail generation should succeed from raw bytes");
    assert!(!thumb_bytes.is_empty(), "thumbnail bytes must not be empty");
    assert!(
        thumb_bytes.len() > 2 && thumb_bytes[0] == 0xFF && thumb_bytes[1] == 0xD8,
        "output should be JPEG format"
    );
}