velesdb-cli 1.14.1

Interactive CLI and REPL for VelesDB with VelesQL support
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

//! Tests for collection_helpers module (Phase 2.1)

use tempfile::TempDir;
use velesdb_core::{Database, DistanceMetric, GraphSchema};

use crate::collection_helpers::{collection_type_label, resolve_collection, TypedCollection};

// =============================================================================
// resolve_collection tests
// =============================================================================

#[test]
fn test_resolve_collection_vector() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_vector_collection("vectors", 128, DistanceMetric::Cosine)
        .unwrap();

    let result = resolve_collection(&db, "vectors");
    assert!(result.is_some());
    assert!(matches!(result.unwrap(), TypedCollection::Vector(_)));
}

#[test]
fn test_resolve_collection_graph() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_graph_collection("knowledge", GraphSchema::schemaless())
        .unwrap();

    let result = resolve_collection(&db, "knowledge");
    assert!(result.is_some());
    assert!(matches!(result.unwrap(), TypedCollection::Graph(_)));
}

#[test]
fn test_resolve_collection_metadata() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_metadata_collection("catalog").unwrap();

    let result = resolve_collection(&db, "catalog");
    assert!(result.is_some());
    assert!(matches!(result.unwrap(), TypedCollection::Metadata(_)));
}

#[test]
fn test_resolve_collection_not_found() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();

    let result = resolve_collection(&db, "nonexistent");
    assert!(result.is_none());
}

#[test]
fn test_resolve_collection_multiple_types() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_vector_collection("vec_col", 64, DistanceMetric::Euclidean)
        .unwrap();
    db.create_graph_collection("graph_col", GraphSchema::schemaless())
        .unwrap();
    db.create_metadata_collection("meta_col").unwrap();

    assert!(matches!(
        resolve_collection(&db, "vec_col").unwrap(),
        TypedCollection::Vector(_)
    ));
    assert!(matches!(
        resolve_collection(&db, "graph_col").unwrap(),
        TypedCollection::Graph(_)
    ));
    assert!(matches!(
        resolve_collection(&db, "meta_col").unwrap(),
        TypedCollection::Metadata(_)
    ));
}

// =============================================================================
// collection_type_label tests
// =============================================================================

#[test]
fn test_collection_type_label_vector() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_vector_collection("docs", 768, DistanceMetric::Cosine)
        .unwrap();

    assert_eq!(collection_type_label(&db, "docs"), "Vector");
}

#[test]
fn test_collection_type_label_graph() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_graph_collection("kg", GraphSchema::schemaless())
        .unwrap();

    assert_eq!(collection_type_label(&db, "kg"), "Graph");
}

#[test]
fn test_collection_type_label_metadata() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();
    db.create_metadata_collection("products").unwrap();

    assert_eq!(collection_type_label(&db, "products"), "Metadata");
}

#[test]
fn test_collection_type_label_unknown() {
    let dir = TempDir::new().unwrap();
    let db = Database::open(dir.path()).unwrap();

    assert_eq!(collection_type_label(&db, "nonexistent"), "unknown");
}