uni-db 1.1.0

// SPDX-License-Identifier: Apache-2.0
// Copyright 2024-2026 Dragonscale Team

use anyhow::Result;
use std::sync::Arc;
use uni_db::{DataType, Uni, UniError};

#[tokio::test]
async fn test_complex_cypher() -> Result<()> {
    let db = Uni::in_memory().build().await?;

    db.schema()
        .label("Product")
        .property("name", DataType::String)
        .property("price", DataType::Float)
        .label("Order")
        .property("date", DataType::String)
        .edge_type("CONTAINS", &["Order"], &["Product"])
        .property("qty", DataType::Int32)
        .apply()
        .await?;

    // Data
    let tx = db.session().tx().await?;
    tx.execute("CREATE (p1:Product {name: 'Apple', price: 1.2})")
        .await?;
    tx.execute("CREATE (p2:Product {name: 'Banana', price: 0.8})")
        .await?;
    tx.execute("CREATE (o1:Order {date: '2024-01-01'})").await?;
    tx.execute("CREATE (o2:Order {date: '2024-01-02'})").await?;
    tx.execute("MATCH (o:Order {date: '2024-01-01'}), (p:Product {name: 'Apple'}) CREATE (o)-[:CONTAINS {qty: 10}]->(p)").await?;
    tx.execute("MATCH (o:Order {date: '2024-01-01'}), (p:Product {name: 'Banana'}) CREATE (o)-[:CONTAINS {qty: 5}]->(p)").await?;
    tx.execute("MATCH (o:Order {date: '2024-01-02'}), (p:Product {name: 'Apple'}) CREATE (o)-[:CONTAINS {qty: 20}]->(p)").await?;
    tx.commit().await?;

    // Aggregation
    // Total quantity per product
    let query = "
        MATCH (o:Order)-[r:CONTAINS]->(p:Product)
        RETURN p.name, sum(r.qty) as total_qty
    ";
    let result = db.session().query(query).await?;
    assert_eq!(result.len(), 2);

    let mut map = std::collections::HashMap::new();
    for row in result {
        let name: String = row.get("p.name")?;
        let qty: i64 = row.get("total_qty")?;
        map.insert(name, qty);
    }

    assert_eq!(map.get("Apple"), Some(&30));
    assert_eq!(map.get("Banana"), Some(&5));

    Ok(())
}

#[tokio::test]
async fn test_error_handling() -> Result<()> {
    let db = Uni::in_memory().build().await?;

    // 1. Invalid Syntax
    let res = db.session().query("MATCH (n").await;
    assert!(matches!(res, Err(UniError::Parse { .. })));

    // 2. Unknown Label - now supported via schemaless mode (ScanMainByLabel)
    // Instead of erroring, it returns empty results for unknown labels
    let res = db.session().query("MATCH (n:NonExistent) RETURN n").await;
    // Schemaless support: unknown labels return empty result set (not an error)
    assert!(
        res.is_ok(),
        "Unknown labels should succeed with empty results"
    );
    assert!(
        res.unwrap().rows().is_empty(),
        "Unknown label should return no rows"
    );

    // 3. Type Mismatch (Schema constraint)
    db.schema()
        .label("User")
        .property("age", DataType::Int32)
        .apply()
        .await?;
    // Uni currently doesn't enforce schema constraints on write strictly in L0
    // (L0 is schema-less/flexible), but read might fail or cast?
    let tx = db.session().tx().await?;
    tx.execute("CREATE (:User {age: 25})").await?;
    tx.commit().await?;
    let res = db.session().query("MATCH (u:User) RETURN u.age").await?;
    let row = &res.rows()[0];

    // Try to get int as bool (should fail)
    // Note: String conversion from Int IS supported in FromValue, so get::<String> succeeds.
    let err = row.get::<bool>("u.age");
    assert!(matches!(err, Err(UniError::Type { .. })));

    Ok(())
}

#[tokio::test]
async fn test_concurrency() -> Result<()> {
    let db = Arc::new(Uni::in_memory().build().await?);

    db.schema()
        .label("Counter")
        .property("val", DataType::Int32)
        .apply()
        .await?;
    let tx = db.session().tx().await?;
    tx.execute("CREATE (:Counter {val: 0})").await?;
    tx.commit().await?;

    let mut handles = Vec::new();
    for _ in 0..10 {
        let db_clone = db.clone();
        handles.push(tokio::spawn(async move {
            // Read-modify-write in transaction
            let session = db_clone.session();
            let tx = session.tx().await.unwrap();
            // Note: Uni Cypher doesn't support "SET n.val = n.val + 1" atomically in one go
            // if planner doesn't support expression on RHS of SET referencing LHS?
            // Planner supports generic SET expressions.
            // Let's verify SET n.val = n.val + 1 logic.
            // Executor::execute_set_items evaluates expr.

            // However, concurrent transactions might conflict on L0 write lock
            // or just overwrite each other (Last Write Wins) if no locking.
            // Uni L0 is protected by RwLock. Writers are serialized.
            // But Read-Modify-Write requires transaction isolation (Repeatable Read / Serializable).
            // Current Uni transaction implementation:
            // - `begin` takes a snapshot of L0 version.
            // - `commit` merges local L0 into global L0.
            // - Conflict detection?
            //   - `L0Manager::merge` just appends?
            //   - If so, we have Lost Update problem.
            //   - This test confirms behavior (likely lost updates).

            // Let's just do inserts to verify thread safety (no panics).
            tx.execute("CREATE (:Counter {val: 1})").await.unwrap();
            tx.commit().await.unwrap();
        }));
    }

    for h in handles {
        h.await?;
    }

    let count = db
        .session()
        .query("MATCH (c:Counter) RETURN count(c) as cnt")
        .await?;
    // 1 initial + 10 inserts = 11
    let cnt = count.rows()[0].get::<i64>("cnt")?;
    assert_eq!(cnt, 11);

    Ok(())
}

#[tokio::test]
async fn test_builder_config() -> Result<()> {
    let db = Uni::open("tmp/test_config")
        .config(uni_db::UniConfig {
            cache_size: 1024 * 1024, // 1MB
            parallelism: 2,
            ..Default::default()
        })
        .build()
        .await?;

    assert_eq!(db.config().cache_size, 1024 * 1024);
    assert_eq!(db.config().parallelism, 2);

    Ok(())
}