somnia 0.8.0

Type-safe SurrealDB ORM for Rust: typed query builder, #[derive(SurrealRecord)], schema generation, and Diesel-style migrations.
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

//! Live-query tests against the in-memory engine: a `LIVE SELECT` stream should
//! surface create/update/delete notifications with the deserialized record.

use futures::StreamExt;
use somnia::{Action, SomniaClient, SurrealRecord, Thing};

#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, SurrealRecord)]
#[table("widget", schemaless)]
struct Widget {
    #[field(thing)]
    id: Thing<Widget>,
    name: String,
}

#[tokio::test]
async fn live_select_streams_create_update_delete() {
    let client = SomniaClient::connect_anonymous("mem://", "t", "t")
        .await
        .expect("connect");
    client
        .raw("DEFINE TABLE widget SCHEMALESS;")
        .await
        .expect("define table");

    let mut stream = client
        .live_select::<Widget>()
        .await
        .expect("start live query");

    // Drive a create, an update, and a delete on the watched table.
    client
        .raw("CREATE widget:one SET name = 'first';")
        .await
        .expect("create");
    client
        .raw("UPDATE widget:one SET name = 'renamed';")
        .await
        .expect("update");
    client.raw("DELETE widget:one;").await.expect("delete");

    // Collect the three notifications (with a timeout so a hang fails fast).
    let mut actions = Vec::new();
    let mut names = Vec::new();
    for _ in 0..3 {
        let item = tokio::time::timeout(std::time::Duration::from_secs(5), stream.next())
            .await
            .expect("notification within timeout")
            .expect("stream not ended")
            .expect("notification ok");
        actions.push(item.action);
        names.push(item.data.name.clone());
        assert!(!item.query_id.is_nil(), "query_id should be set");
    }

    assert_eq!(
        actions,
        vec![Action::Create, Action::Update, Action::Delete],
        "expected create/update/delete in order, got {actions:?}"
    );
    // Create + update carry the live value; delete carries the last-known record.
    assert_eq!(names[0], "first");
    assert_eq!(names[1], "renamed");

    // Dropping the stream issues KILL; nothing should panic.
    drop(stream);
}