indradb-lib 0.12.1

A graph database library
Documentation
use super::super::{Datastore, EdgeKey, EdgeQuery, Transaction, Type, VertexQuery};
use util::generate_random_secret;
use uuid::Uuid;
use serde_json::Value as JsonValue;

pub fn should_handle_global_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let name = format!("global-metadata-{}", generate_random_secret(8));
    let trans = datastore.transaction().unwrap();

    // Check to make sure there's no initial value
    let result = trans.get_global_metadata(&name);
    assert_eq!(result.unwrap(), None);

    // Set and get the value as true
    trans
        .set_global_metadata(&name, &JsonValue::Bool(true))
        .unwrap();

    let result = trans.get_global_metadata(&name);
    assert_eq!(result.unwrap(), Some(JsonValue::Bool(true)));

    // Set and get the value as false
    trans
        .set_global_metadata(&name, &JsonValue::Bool(false))
        .unwrap();

    let result = trans.get_global_metadata(&name);
    assert_eq!(result.unwrap(), Some(JsonValue::Bool(false)));

    // Delete & check that it's deleted
    trans.delete_global_metadata(&name).unwrap();

    let result = trans.get_global_metadata(&name);
    assert_eq!(result.unwrap(), None);
}

pub fn should_handle_vertex_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let trans = datastore.transaction().unwrap();
    let t = Type::new("test_edge_type".to_string()).unwrap();
    let owner_id = trans.create_vertex(&t).unwrap();
    let name = format!("vertex-metadata-{}", generate_random_secret(8));
    let q = VertexQuery::Vertices {
        ids: vec![owner_id],
    };

    // Check to make sure there's no initial value
    let result = trans.get_vertex_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 0);

    // Set and get the value as true
    trans
        .set_vertex_metadata(&q, &name, &JsonValue::Bool(true))
        .unwrap();
    let result = trans.get_vertex_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 1);
    assert_eq!(result[0].id, owner_id);
    assert_eq!(result[0].value, JsonValue::Bool(true));

    // Set and get the value as false
    trans
        .set_vertex_metadata(&q, &name, &JsonValue::Bool(false))
        .unwrap();
    let result = trans.get_vertex_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 1);
    assert_eq!(result[0].id, owner_id);
    assert_eq!(result[0].value, JsonValue::Bool(false));

    // Delete & check that it's deleted
    trans.delete_vertex_metadata(&q, &name).unwrap();
    let result = trans.get_vertex_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 0);
}

pub fn should_not_set_invalid_vertex_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let trans = datastore.transaction().unwrap();
    let q = VertexQuery::Vertices {
        ids: vec![Uuid::default()],
    };
    trans
        .set_vertex_metadata(&q, "foo", &JsonValue::Null)
        .unwrap();
    let result = trans.get_vertex_metadata(&q, "foo").unwrap();
    assert_eq!(result.len(), 0);
}

pub fn should_not_delete_invalid_vertex_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let trans = datastore.transaction().unwrap();
    let q = VertexQuery::Vertices {
        ids: vec![Uuid::default()],
    };
    trans.delete_vertex_metadata(&q, "foo").unwrap();

    let vertex_id = trans
        .create_vertex(&Type::new("foo".to_string()).unwrap())
        .unwrap();
    let q = VertexQuery::Vertices {
        ids: vec![vertex_id],
    };
    trans.delete_vertex_metadata(&q, "foo").unwrap();
}

pub fn should_handle_edge_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let trans = datastore.transaction().unwrap();
    let vertex_t = Type::new("test_edge_type".to_string()).unwrap();
    let outbound_id = trans.create_vertex(&vertex_t).unwrap();
    let inbound_id = trans.create_vertex(&vertex_t).unwrap();
    let edge_t = Type::new("test_edge_type".to_string()).unwrap();
    let key = EdgeKey::new(outbound_id, edge_t.clone(), inbound_id);
    let q = EdgeQuery::Edges {
        keys: vec![key.clone()],
    };
    let name = format!("edge-metadata-{}", generate_random_secret(8));

    trans.create_edge(&key).unwrap();

    // Check to make sure there's no initial value
    let result = trans.get_edge_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 0);

    // Set and get the value as true
    trans
        .set_edge_metadata(&q, &name, &JsonValue::Bool(true))
        .unwrap();
    let result = trans.get_edge_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 1);
    assert_eq!(result[0].key, key);
    assert_eq!(result[0].value, JsonValue::Bool(true));

    // Set and get the value as false
    trans
        .set_edge_metadata(&q, &name, &JsonValue::Bool(false))
        .unwrap();
    let result = trans.get_edge_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 1);
    assert_eq!(result[0].key, key);
    assert_eq!(result[0].value, JsonValue::Bool(false));

    // Delete & check that it's deleted
    trans.delete_edge_metadata(&q, &name).unwrap();
    let result = trans.get_edge_metadata(&q, &name).unwrap();
    assert_eq!(result.len(), 0);
}

pub fn should_not_set_invalid_edge_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let trans = datastore.transaction().unwrap();
    let q = EdgeQuery::Edges {
        keys: vec![
            EdgeKey::new(
                Uuid::default(),
                Type::new("foo".to_string()).unwrap(),
                Uuid::default(),
            ),
        ],
    };
    trans
        .set_edge_metadata(&q, "bar", &JsonValue::Null)
        .unwrap();
    let result = trans.get_edge_metadata(&q, "bar").unwrap();
    assert_eq!(result.len(), 0);
}

pub fn should_not_delete_invalid_edge_metadata<D, T>(datastore: &mut D)
where
    D: Datastore<T>,
    T: Transaction,
{
    let trans = datastore.transaction().unwrap();
    let q = EdgeQuery::Edges {
        keys: vec![
            EdgeKey::new(
                Uuid::default(),
                Type::new("foo".to_string()).unwrap(),
                Uuid::default(),
            ),
        ],
    };
    trans.delete_edge_metadata(&q, "bar").unwrap();

    let outbound_id = trans
        .create_vertex(&Type::new("foo".to_string()).unwrap())
        .unwrap();
    let inbound_id = trans
        .create_vertex(&Type::new("bar".to_string()).unwrap())
        .unwrap();
    let key = EdgeKey::new(
        outbound_id,
        Type::new("baz".to_string()).unwrap(),
        inbound_id,
    );
    trans.create_edge(&key).unwrap();
    trans
        .delete_edge_metadata(&EdgeQuery::Edges { keys: vec![key] }, "bleh")
        .unwrap();
}