nodedb 0.3.0

Local-first, real-time, edge-to-cloud hybrid database for multi-modal workloads
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

// SPDX-License-Identifier: BUSL-1.1

//! Engine surface tests for the Graph engine overlay.
//!
//! Graph is a cross-engine overlay — edges and nodes live inside any collection.
//! Covers: MATCH pattern queries, GRAPH ALGO (SSSP, PageRank), edge
//! insertion, and rejection of `WITH (engine='graph')`.

mod common;
use common::pgwire_harness::TestServer;

#[tokio::test]
async fn match_pattern_simple_path() {
    let srv = TestServer::start().await;
    srv.exec("CREATE COLLECTION graph_nodes WITH (engine='document_schemaless')")
        .await
        .unwrap();

    srv.exec("INSERT INTO graph_nodes { id: 'alice', name: 'Alice' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_nodes { id: 'bob', name: 'Bob' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_nodes { id: 'carol', name: 'Carol' }")
        .await
        .unwrap();

    srv.exec("INSERT INTO graph_nodes { id: 'e1', _from: 'alice', _to: 'bob', _type: 'knows' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_nodes { id: 'e2', _from: 'bob', _to: 'carol', _type: 'knows' }")
        .await
        .unwrap();

    let rows = srv
        .query_rows(
            "MATCH (a {id: 'alice'})-[:knows]->(b) IN 'graph_nodes' \
             RETURN b.id",
        )
        .await
        .unwrap();
    let ids: Vec<&str> = rows.iter().map(|r| r[0].as_str()).collect();
    assert!(
        ids.contains(&"bob"),
        "expected bob in MATCH result: {ids:?}"
    );
}

#[tokio::test]
async fn graph_algo_sssp() {
    let srv = TestServer::start().await;
    srv.exec("CREATE COLLECTION graph_sssp WITH (engine='document_schemaless')")
        .await
        .unwrap();

    srv.exec("INSERT INTO graph_sssp { id: 'n1', label: 'start' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_sssp { id: 'n2', label: 'mid' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_sssp { id: 'n3', label: 'end' }")
        .await
        .unwrap();

    srv.exec(
        "INSERT INTO graph_sssp { id: 'e1', _from: 'n1', _to: 'n2', _type: 'road', weight: 1.0 }",
    )
    .await
    .unwrap();
    srv.exec(
        "INSERT INTO graph_sssp { id: 'e2', _from: 'n2', _to: 'n3', _type: 'road', weight: 1.0 }",
    )
    .await
    .unwrap();

    let rows = srv
        .query_rows("GRAPH ALGO SSSP ON graph_sssp SOURCE 'n1'")
        .await
        .unwrap();
    assert!(!rows.is_empty(), "SSSP returned no rows");
}

#[tokio::test]
async fn graph_algo_pagerank() {
    let srv = TestServer::start().await;
    srv.exec("CREATE COLLECTION graph_pr WITH (engine='document_schemaless')")
        .await
        .unwrap();

    srv.exec("INSERT INTO graph_pr { id: 'a', label: 'A' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_pr { id: 'b', label: 'B' }")
        .await
        .unwrap();
    srv.exec("INSERT INTO graph_pr { id: 'e1', _from: 'a', _to: 'b', _type: 'link' }")
        .await
        .unwrap();

    let rows = srv
        .query_rows("GRAPH ALGO PAGERANK ON graph_pr")
        .await
        .unwrap();
    assert!(!rows.is_empty(), "PageRank returned no rows");
}

#[tokio::test]
async fn engine_graph_flag_rejected_in_with_clause() {
    let srv = TestServer::start().await;
    let err = srv
        .exec("CREATE COLLECTION bad_graph WITH (engine='graph')")
        .await
        .unwrap_err();
    assert!(
        err.to_lowercase().contains("match")
            || err.to_lowercase().contains("graph")
            || err.to_lowercase().contains("unsupported"),
        "expected graph-rejection error, got: {err}"
    );
}