doido-storage 0.0.10

Attached-file storage — blobs, polymorphic attachments and pluggable services (disk, memory, S3, Cloudflare R2, Azure Blob) in ActiveStorage fashion for Doido.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

//! DiskService round-trips bytes under a tempdir and is idempotent on delete.

use doido_storage::{DiskService, Service};

async fn disk() -> (tempfile::TempDir, DiskService) {
    let dir = tempfile::tempdir().unwrap();
    let svc = DiskService::new("local", dir.path().to_path_buf());
    (dir, svc)
}

#[tokio::test]
async fn upload_download_roundtrip() {
    let (_dir, svc) = disk().await;
    svc.upload("abcd1234", b"hello disk".to_vec(), Some("text/plain"))
        .await
        .unwrap();

    assert!(svc.exists("abcd1234").await.unwrap());
    assert_eq!(svc.download("abcd1234").await.unwrap(), b"hello disk");
    assert_eq!(svc.size("abcd1234").await.unwrap(), 10);
}

#[tokio::test]
async fn delete_is_idempotent_and_download_missing_errs() {
    let (_dir, svc) = disk().await;
    // Deleting a missing key is fine.
    svc.delete("nope").await.unwrap();
    assert!(!svc.exists("nope").await.unwrap());
    assert!(svc.download("nope").await.is_err());

    svc.upload("k1", b"x".to_vec(), None).await.unwrap();
    svc.delete("k1").await.unwrap();
    assert!(!svc.exists("k1").await.unwrap());
}

#[tokio::test]
async fn rejects_keys_that_escape_the_root() {
    let (_dir, svc) = disk().await;
    assert!(svc.upload("../evil", b"x".to_vec(), None).await.is_err());
    assert!(svc.download("a/b").await.is_err());
}

#[tokio::test]
async fn public_flag_root_empty_key_and_short_shard() {
    let (dir, svc) = disk().await;
    let public = DiskService::new("local", dir.path()).public(true);
    assert!(doido_storage::Service::public(&public));

    assert_eq!(public.root(), dir.path());
    assert!(public.upload("", b"x".to_vec(), None).await.is_err());
    assert!(public.size("missing").await.is_err());

    public.upload("ab", b"short".to_vec(), None).await.unwrap();
    let path = dir.path().join("ab").join("__").join("ab");
    assert!(path.exists());
}