doido-storage 0.0.17

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134

//! has_one / has_many attachment semantics and dependent purge.

use doido_storage::{MemoryService, Signer, Storage};
use std::sync::Arc;

async fn storage() -> Storage {
    let conn = doido_model::sea_orm::Database::connect("sqlite::memory:")
        .await
        .unwrap();
    let s = Storage::new(
        conn,
        Arc::new(MemoryService::default()),
        Signer::new(b"test-secret".to_vec()),
    );
    s.ensure_tables().await.unwrap();
    s
}

#[tokio::test]
async fn has_one_replaces_previous() {
    let s = storage().await;
    let a = s
        .create_and_upload("a.txt", b"a".to_vec(), None)
        .await
        .unwrap();
    let b = s
        .create_and_upload("b.txt", b"b".to_vec(), None)
        .await
        .unwrap();

    // Attach then replace via detach+attach (has_one semantics).
    s.attach("User", "1", "avatar", &a.key).await.unwrap();
    assert_eq!(
        s.one("User", "1", "avatar").await.unwrap().unwrap().key,
        a.key
    );

    s.detach("User", "1", "avatar").await.unwrap();
    s.attach("User", "1", "avatar", &b.key).await.unwrap();
    assert_eq!(
        s.one("User", "1", "avatar").await.unwrap().unwrap().key,
        b.key
    );
}

#[tokio::test]
async fn has_many_keeps_all() {
    let s = storage().await;
    let one = s
        .create_and_upload("1.txt", b"1".to_vec(), None)
        .await
        .unwrap();
    let two = s
        .create_and_upload("2.txt", b"2".to_vec(), None)
        .await
        .unwrap();
    s.attach("Post", "9", "photos", &one.key).await.unwrap();
    s.attach("Post", "9", "photos", &two.key).await.unwrap();

    let photos = s.many("Post", "9", "photos").await.unwrap();
    assert_eq!(photos.len(), 2);
    assert_eq!(photos[0].key, one.key);
    assert_eq!(photos[1].key, two.key);
}

#[tokio::test]
async fn purge_for_record_purges_dependent_blobs() {
    let s = storage().await;
    let one = s
        .create_and_upload("1.txt", b"1".to_vec(), None)
        .await
        .unwrap();
    let two = s
        .create_and_upload("2.txt", b"2".to_vec(), None)
        .await
        .unwrap();
    s.attach("Post", "9", "photos", &one.key).await.unwrap();
    s.attach("Post", "9", "photos", &two.key).await.unwrap();

    s.purge_for_record("Post", "9").await.unwrap();

    assert!(s.many("Post", "9", "photos").await.unwrap().is_empty());
    assert!(s.find_blob(&one.key).await.unwrap().is_none());
    assert!(s.find_blob(&two.key).await.unwrap().is_none());
}

#[tokio::test]
async fn replace_one_swaps_attachment() {
    let s = storage().await;
    let a = s
        .create_and_upload("a.txt", b"a".to_vec(), None)
        .await
        .unwrap();
    let b = s
        .create_and_upload("b.txt", b"b".to_vec(), None)
        .await
        .unwrap();

    doido_storage::attachments::replace_one(s.conn(), "avatar", "User", "1", &a.key)
        .await
        .unwrap();
    doido_storage::attachments::replace_one(s.conn(), "avatar", "User", "1", &b.key)
        .await
        .unwrap();

    let keys = doido_storage::attachments::attached_keys(s.conn(), "avatar", "User", "1")
        .await
        .unwrap();
    assert_eq!(keys, vec![b.key]);
}

#[tokio::test]
async fn detach_blob_removes_one_of_many() {
    let s = storage().await;
    let one = s
        .create_and_upload("1.txt", b"1".to_vec(), None)
        .await
        .unwrap();
    let two = s
        .create_and_upload("2.txt", b"2".to_vec(), None)
        .await
        .unwrap();
    s.attach("Post", "1", "docs", &one.key).await.unwrap();
    s.attach("Post", "1", "docs", &two.key).await.unwrap();

    doido_storage::attachments::detach_blob(s.conn(), "docs", "Post", "1", &two.key)
        .await
        .unwrap();

    let keys = doido_storage::attachments::attached_keys(s.conn(), "docs", "Post", "1")
        .await
        .unwrap();
    assert_eq!(keys, vec![one.key]);
}