apache_age 0.6.5

#![cfg(feature = "sync")]
#![allow(unused_must_use)]

use apache_age::sync::{AgeClient, Client};
use apache_age::{AgType, NoTls, Vertex};
use rand::{distributions::Alphanumeric, Rng};
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize, Clone)]
struct Person {
    pub name: String,
    pub surname: String,
}

struct TestConnection {
    pub client: Client,
    pub graph_name: String,
}

impl TestConnection {
    pub fn new() -> Self {
        let (client, graph_name) = connect();

        Self { client, graph_name }
    }
}

impl Drop for TestConnection {
    fn drop(&mut self) {
        self.client.drop_graph(&self.graph_name);
    }
}

fn connect() -> (Client, String) {
    let mut client = Client::connect_age(
        "host=localhost user=postgres password=passwd port=8081",
        NoTls,
    )
    .unwrap();

    let graph_name = "age_test_".to_string()
        + &rand::thread_rng()
            .sample_iter(&Alphanumeric)
            .take(7)
            .map(char::from)
            .collect::<String>();

    assert!(client.create_graph(&graph_name).is_ok());

    (client, graph_name)
}

#[test]
fn simple_query() {
    let mut tc = TestConnection::new();

    tc.client.simple_query(
        &("SELECT * FROM cypher('".to_string()
            + &tc.graph_name
            + "', $$ CREATE(n:Person {name: 'T', surname: 'Doe'}) RETURN n $$) AS (v agtype)"),
    );

    tc.client.simple_query(
        &("SELECT * FROM cypher('".to_string()
            + &tc.graph_name
            + "', $$ CREATE(n:Person {name: 'Jack', surname: 'Hell'}) RETURN n $$) AS (v agtype)"),
    );

    // Query, not query_one on puropose, just checking if iterating works
    let query = tc
        .client
        .query(
            &("SELECT v FROM ag_catalog.cypher('".to_string()
                + &tc.graph_name
                + "', $$ MATCH(n: Person) WHERE n.name='T' RETURN n $$) AS (v ag_catalog.agtype)"),
            &[],
        )
        .unwrap();
    let qlen = query.len();
    for row in query {
        let person_vertex: Vertex<Person> = row.get(0);
        assert_eq!(person_vertex.properties().surname, "Doe");
        assert_eq!(person_vertex.properties().name, "T");
    }
    assert_eq!(qlen, 1);

    let query = tc
        .client
        .query(
            &("SELECT v FROM ag_catalog.cypher('".to_string()
                + &tc.graph_name
                + "', $$ MATCH(n: Person) RETURN n $$) AS (v ag_catalog.agtype)"),
            &[],
        )
        .unwrap();
    let qlen = query.len();
    assert_eq!(qlen, 2);
}

#[test]
fn person() {
    let mut tc = TestConnection::new();

    assert!(tc.client.graph_exists(&tc.graph_name).unwrap());

    assert!(tc
        .client
        .execute_cypher(
            &tc.graph_name,
            "CREATE(n: Person {name: $name, surname: $surname})",
            Some(AgType::<Person>(Person {
                name: "Alfred".into(),
                surname: "Bohr".into(),
            })),
        )
        .is_ok());

    assert!(tc
        .client
        .execute_cypher(
            &tc.graph_name,
            "CREATE(n: Person {name: $name, surname: $surname})",
            Some(AgType::<Person>(Person {
                name: "John".into(),
                surname: "Doe".into(),
            })),
        )
        .is_ok());

    let rows = tc
        .client
        .query_cypher::<()>(
            &tc.graph_name,
            "MATCH (n: Person) WHERE n.name = 'Alfred' RETURN {name: n.name, surname: n.surname}",
            None,
        )
        .unwrap();
    let x: AgType<Person> = rows[0].get(0);
    assert_eq!(x.0.surname, "Bohr");
}

#[derive(Debug, Serialize, Deserialize, Clone)]
struct IdPassing {
    id: usize,
}

#[test]
fn unique_index() {
    let mut tc = TestConnection::new();

    let result = tc.client.query_cypher::<()>(
        &tc.graph_name,
        "CREATE(n: Person {name: 'Dummy', surname: 'Name'}) RETURN id(n)",
        None,
    );

    let dummy_person: AgType<usize> = result.unwrap()[0].get(0);

    tc.client
        .unique_index(&tc.graph_name, "Person", "myconstraint", "surname");

    tc.client.execute_cypher::<IdPassing>(
        &tc.graph_name,
        "MATCH (n: Person) WHERE id(n) = $id DELETE n",
        Some(AgType::<IdPassing>(IdPassing { id: dummy_person.0 })),
    );

    tc.client
        .execute_cypher::<()>(&tc.graph_name, "CREATE(n: Person {surname: 'Name'})", None);

    println!("One must not be able to perform this operation");
    assert!(tc
        .client
        .execute_cypher::<()>(
            &tc.graph_name,
            "CREATE(n: Person {name: 'Dummy', surname: 'Name'})",
            None,
        )
        .is_err());

    println!("One must not be able to perform this operation");
    assert!(tc
        .client
        .execute_cypher::<()>(&tc.graph_name, "CREATE(n: Person {surname: 'Name'})", None,)
        .is_err());
}

#[test]
fn required_constraint() {
    let mut tc = TestConnection::new();

    let result = tc.client.query_cypher::<()>(
        &tc.graph_name,
        "CREATE(n: Person {name: 'Dummy', surname: 'Name'}) RETURN id(n)",
        None,
    );

    let dummy_person: AgType<usize> = result.unwrap()[0].get(0);

    tc.client
        .required_constraint(&tc.graph_name, "Person", "myconstraint", "surname");

    tc.client.execute_cypher::<IdPassing>(
        &tc.graph_name,
        "MATCH (n: Person) WHERE id(n) = $id DELETE n",
        Some(AgType::<IdPassing>(IdPassing { id: dummy_person.0 })),
    );

    println!("One must not be able to perform this operation");
    assert!(tc
        .client
        .execute_cypher::<()>(&tc.graph_name, "CREATE(n: Person {name: 'Name'})", None)
        .is_err());
}