uni-db 1.1.0

Embedded graph database with OpenCypher queries, vector search, and columnar storage
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

// SPDX-License-Identifier: Apache-2.0
// Copyright 2024-2026 Dragonscale Team
//
// Tests for CASE expressions (generic CASE WHEN and simple CASE).

use anyhow::Result;
use uni_db::{DataType, Uni};

#[tokio::test]
async fn test_case_expression() -> Result<()> {
    let db = Uni::in_memory().build().await?;
    db.schema()
        .label("Person")
        .property("name", DataType::String)
        .property_nullable("age", DataType::Int64)
        .apply()
        .await?;

    let session = db.session();
    let tx = session.tx().await?;
    tx.execute("CREATE (:Person {name: 'Alice', age: 10})")
        .await?;
    tx.execute("CREATE (:Person {name: 'Bob', age: 20})")
        .await?;
    tx.execute("CREATE (:Person {name: 'Charlie', age: 30})")
        .await?;
    tx.commit().await?;

    // 1. Generic CASE WHEN
    let results = db
        .session()
        .query(
            "MATCH (n:Person) RETURN n.name AS name, \
             CASE WHEN n.age < 15 THEN 'Child' WHEN n.age < 25 THEN 'Teen' ELSE 'Adult' END AS category \
             ORDER BY name",
        )
        .await?;

    assert_eq!(results.len(), 3);
    assert_eq!(results.rows()[0].get::<String>("name")?, "Alice");
    assert_eq!(results.rows()[0].get::<String>("category")?, "Child");
    assert_eq!(results.rows()[1].get::<String>("name")?, "Bob");
    assert_eq!(results.rows()[1].get::<String>("category")?, "Teen");
    assert_eq!(results.rows()[2].get::<String>("name")?, "Charlie");
    assert_eq!(results.rows()[2].get::<String>("category")?, "Adult");

    // 2. Simple CASE
    let results = db
        .session()
        .query(
            "MATCH (n:Person) RETURN n.name AS name, \
             CASE n.name WHEN 'Alice' THEN 1 WHEN 'Bob' THEN 2 ELSE 3 END AS rank \
             ORDER BY name",
        )
        .await?;

    assert_eq!(results.rows()[0].get::<i64>("rank")?, 1); // Alice
    assert_eq!(results.rows()[1].get::<i64>("rank")?, 2); // Bob
    assert_eq!(results.rows()[2].get::<i64>("rank")?, 3); // Charlie

    Ok(())
}