sqlitegraph 2.2.2

Embedded graph database with full ACID transactions, HNSW vector search, dual backend support, and comprehensive graph algorithms library
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

//! TDD: Edge durability across reopen
use sqlitegraph::backend::native::v3::V3Backend;
use sqlitegraph::backend::{BackendDirection, EdgeSpec, GraphBackend, NeighborQuery, NodeSpec};
use sqlitegraph::snapshot::SnapshotId;
use tempfile::TempDir;

#[test]
fn test_edge_type_survives_reopen() {
    let temp_dir = TempDir::new().unwrap();
    let db_path = temp_dir.path().join("test.graph");
    {
        let backend = V3Backend::create(&db_path).unwrap();
        let center = backend
            .insert_node(NodeSpec {
                kind: "Node".to_string(),
                name: "center".to_string(),
                file_path: None,
                data: serde_json::json!({}),
            })
            .unwrap();
        let helper = backend
            .insert_node(NodeSpec {
                kind: "Node".to_string(),
                name: "helper".to_string(),
                file_path: None,
                data: serde_json::json!({}),
            })
            .unwrap();
        let util = backend
            .insert_node(NodeSpec {
                kind: "Node".to_string(),
                name: "util".to_string(),
                file_path: None,
                data: serde_json::json!({}),
            })
            .unwrap();
        backend
            .insert_edge(EdgeSpec {
                from: center,
                to: helper,
                edge_type: "CALLS".to_string(),
                data: serde_json::json!({}),
            })
            .unwrap();
        backend
            .insert_edge(EdgeSpec {
                from: center,
                to: util,
                edge_type: "USES".to_string(),
                data: serde_json::json!({}),
            })
            .unwrap();
        backend.flush().unwrap();
    }
    {
        let backend = V3Backend::open(&db_path).unwrap();
        let current = SnapshotId::current();
        let all_ids = backend.entity_ids().unwrap();
        let center = all_ids
            .iter()
            .find(|&&id| backend.get_node(current, id).unwrap().name == "center")
            .copied()
            .unwrap();
        let all = backend
            .neighbors(
                current,
                center,
                NeighborQuery {
                    direction: BackendDirection::Outgoing,
                    edge_type: None,
                },
            )
            .unwrap();
        assert_eq!(all.len(), 2, "Should have 2 neighbors after reopen");
        let calls = backend
            .neighbors(
                current,
                center,
                NeighborQuery {
                    direction: BackendDirection::Outgoing,
                    edge_type: Some("CALLS".to_string()),
                },
            )
            .unwrap();
        assert_eq!(calls.len(), 1, "Should have 1 CALLS neighbor after reopen");
        let uses = backend
            .neighbors(
                current,
                center,
                NeighborQuery {
                    direction: BackendDirection::Outgoing,
                    edge_type: Some("USES".to_string()),
                },
            )
            .unwrap();
        assert_eq!(uses.len(), 1, "Should have 1 USES neighbor after reopen");
    }
}