fluidb/tests/creation.rs

use anyhow::Result;
use cozo::{Db, EncodedVec};
use cozorocks::DbBuilder;
use serde_json::{json, to_string_pretty};

fn create_db(name: &str) -> Db {
    let builder = DbBuilder::default()
        .path(name)
        .create_if_missing(true)
        .destroy_on_exit(true);
    Db::build(builder).unwrap()
}

fn test_send_sync<T: Send + Sync>(_: &T) {}

#[test]
fn creation() {
    let db = create_db("_test_db");
    test_send_sync(&db);
    assert!(db.current_schema().unwrap().as_array().unwrap().is_empty());
    let res = db.transact_attributes(&json!({
        "attrs": [
            {"put": {"keyword": "person/idd", "cardinality": "one", "type": "string", "index": "identity"}},
            {"put": {"keyword": "person/first_name", "cardinality": "one", "type": "string", "index": true}},
            {"put": {"keyword": "person/last_name", "cardinality": "one", "type": "string", "index": true}},
            {"put": {"keyword": "person/age", "cardinality": "one", "type": "int"}},
            {"put": {"keyword": "person/friend", "cardinality": "many", "type": "ref"}},
            {"put": {"keyword": "person/weight", "cardinality": "one", "type": "float"}},
            {"put": {"keyword": "person/covid", "cardinality": "one", "type": "bool"}},
        ]
    }))
    .unwrap();
    println!("{}", res);
    let first_id = res["results"][0][0].as_u64().unwrap();
    let last_id = res["results"][6][0].as_u64().unwrap();
    db.transact_attributes(&json!({
        "attrs": [
            {"put": {"id": first_id, "keyword": ":person/id", "cardinality": "one", "type": "string", "index": "identity"}},
            {"retract": {"id": last_id, "keyword": ":person/covid", "cardinality": "one", "type": "bool"}}
        ]
    })).unwrap();
    assert_eq!(db.current_schema().unwrap().as_array().unwrap().len(), 6);
    println!(
        "{}",
        to_string_pretty(&db.current_schema().unwrap()).unwrap()
    );
    db.transact_triples(&json!({
        "tx": [
            {"put": {
                "_temp_id": "alice",
                "person/first_name": "Alice",
                "person/age": 7,
                "person/last_name": "Amorist",
                "person/id": "alice_amorist",
                "person/weight": 25}},
            {"put": {
                "_temp_id": "bob",
                "person/first_name": "Bob",
                "person/age": 70,
                "person/last_name": "Wonderland",
                "person/id": "bob_wonderland",
                "person/weight": 100,
                "person/friend": "alice"
            }},
            {"put": {
                "_temp_id": "eve",
                "person/first_name": "Eve",
                "person/age": 18,
                "person/last_name": "Faking",
                "person/id": "eve_faking",
                "person/weight": 50,
                // "person/friend": ["alice", "bob"]
            }},
        ]
    }))
    .unwrap();

    // iteration
    let mut it = db.total_iter();
    while let Some((k_slice, v_slice)) = it.pair().unwrap() {
        let key = EncodedVec::new(k_slice);
        let val = key.debug_value(v_slice);
        dbg!(key);
        dbg!(val);
        it.next();
    }
    // let mut tx = session.transact_write().unwrap();
    // tx.new_attr(Attribute {
    //     id: AttrId(0),
    //     keyword: Keyword::try_from("hello/world").unwrap(),
    //     cardinality: AttributeCardinality::Many,
    //     val_type: AttributeTyping::Int,
    //     indexing: AttributeIndex::None,
    //     with_history: true,
    // })
    // .unwrap();
    // tx.commit_tx("", false).unwrap();
    //
    // let mut tx = session.transact_write().unwrap();
    // let attr = tx
    //     .attr_by_kw(&Keyword::try_from("hello/world").unwrap())
    //     .unwrap()
    //     .unwrap();
    // tx.new_triple(EntityId(1), &attr, &Value::Int(98765), current_validity)
    //     .unwrap();
    // tx.new_triple(EntityId(2), &attr, &Value::Int(1111111), current_validity)
    //     .unwrap();
    // tx.commit_tx("haah", false).unwrap();
    //
    // let mut tx = session.transact_write().unwrap();
    // tx.amend_attr(Attribute {
    //     id: AttrId(10000001),
    //     keyword: Keyword::try_from("hello/sucker").unwrap(),
    //     cardinality: AttributeCardinality::Many,
    //     val_type: AttributeTyping::Int,
    //     indexing: AttributeIndex::None,
    //     with_history: true,
    // })
    // .unwrap();
    // tx.commit_tx("oops", false).unwrap();
    //
    // let mut tx = session.transact().unwrap();
    // let world_found = tx
    //     .attr_by_kw(&Keyword::try_from("hello/world").unwrap())
    //     .unwrap();
    // dbg!(world_found);
    // let sucker_found = tx
    //     .attr_by_kw(&Keyword::try_from("hello/sucker").unwrap())
    //     .unwrap();
    // dbg!(sucker_found);
    // for attr in tx.all_attrs() {
    //     dbg!(attr.unwrap());
    // }
    //
    // for r in tx.triple_a_scan_all() {
    //     dbg!(r.unwrap());
    // }
    //
    // dbg!(&session);
    //
    // let mut it = session.total_iter();
    // while let Some((k, v)) = it.pair().unwrap() {
    //     let key = EncodedVec::new(k);
    //     let val = key.debug_value(v);
    //     dbg!(key);
    //     dbg!(val);
    //     it.next();
    // }
}