org.umlg.sqlg.test.schema.TestCaptureSchemaTableEdges Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlg-test Show documentation
There is a newer version: 3.1.1
package org.umlg.sqlg.test.schema;

import org.apache.commons.lang3.tuple.Pair;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assert;
import org.junit.Assume;
import org.junit.Test;
import org.umlg.sqlg.structure.SchemaManager;
import org.umlg.sqlg.structure.SchemaTable;
import org.umlg.sqlg.structure.SqlgGraph;
import org.umlg.sqlg.test.BaseTest;

import java.util.Map;
import java.util.Set;

/**
 * Date: 2015/01/02
 * Time: 8:13 PM
 */
public class TestCaptureSchemaTableEdges extends BaseTest {

    @Test
    public void testCaptureSchemaTableLabels() {
        Vertex person1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "pieter");
        Vertex car1 = this.sqlgGraph.addVertex(T.label, "Car", "name", "bmw");
        person1.addEdge("drives", car1);
        Vertex car2 = this.sqlgGraph.addVertex(T.label, "Car", "name", "toyota");
        person1.addEdge("drives", car2);
        Vertex bmw = this.sqlgGraph.addVertex(T.label, "Model", "name", "bmw");
        car1.addEdge("model", bmw);
        Vertex toyota = this.sqlgGraph.addVertex(T.label, "Model", "name", "toyota");
        car2.addEdge("model", toyota);
        this.sqlgGraph.tx().commit();

        Map, Set>> localTabels = this.sqlgGraph.getSchemaManager().getLocalTableLabels();
        Assert.assertTrue(localTabels.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Person")));
        Assert.assertTrue(localTabels.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Car")));
        Assert.assertTrue(localTabels.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Model")));
        Pair, Set> person = localTabels.get(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Person"));
        Assert.assertEquals(0, person.getLeft().size());
        Assert.assertEquals(1, person.getRight().size());
        Assert.assertEquals(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_drives", person.getRight().iterator().next().toString());
        Pair, Set> car = localTabels.get(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Car"));
        Assert.assertEquals(1, car.getLeft().size());
        Assert.assertEquals(1, car.getRight().size());
        Assert.assertEquals(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_drives", car.getLeft().iterator().next().toString());

        Pair, Set> model = localTabels.get(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Model"));
        Assert.assertEquals(1, model.getLeft().size());
        Assert.assertEquals(0, model.getRight().size());
        Assert.assertEquals(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_model", model.getLeft().iterator().next().toString());
    }

    @Test
    public void testCaptureSchemaTableLabelsRollback() {
        Assume.assumeTrue(this.sqlgGraph.getSqlDialect().supportsTransactionalSchema());
        Vertex person1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "pieter");
        Vertex car1 = this.sqlgGraph.addVertex(T.label, "Car", "name", "bmw");
        person1.addEdge("drives", car1);
        Vertex car2 = this.sqlgGraph.addVertex(T.label, "Car", "name", "toyota");
        person1.addEdge("drives", car2);
        Vertex bmw = this.sqlgGraph.addVertex(T.label, "Model", "name", "bmw");
        car1.addEdge("model", bmw);
        Vertex toyota = this.sqlgGraph.addVertex(T.label, "Model", "name", "toyota");
        car2.addEdge("model", toyota);
        this.sqlgGraph.tx().rollback();
        Map, Set>> localTables = this.sqlgGraph.getSchemaManager().getLocalTableLabels();
        Assert.assertTrue(localTables.containsKey(SchemaTable.of(SchemaManager.SQLG_SCHEMA, "V_vertex")));
        Assert.assertFalse(localTables.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Person")));
        Assert.assertFalse(localTables.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Car")));
        Assert.assertFalse(localTables.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Model")));
        Assert.assertFalse(localTables.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "E_drives")));
        Assert.assertFalse(localTables.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "E_model")));
    }

    @Test
    public void testLoadTableLabels() throws Exception {
        Vertex person1 = this.sqlgGraph.addVertex(T.label, "Person", "name", "pieter");
        Vertex car1 = this.sqlgGraph.addVertex(T.label, "Car", "name", "bmw");
        person1.addEdge("drives", car1);
        Vertex car2 = this.sqlgGraph.addVertex(T.label, "Car", "name", "toyota");
        person1.addEdge("drives", car2);
        Vertex bmw = this.sqlgGraph.addVertex(T.label, "Model", "name", "bmw");
        car1.addEdge("model", bmw);
        Vertex toyota = this.sqlgGraph.addVertex(T.label, "Model", "name", "toyota");
        car2.addEdge("model", toyota);
        this.sqlgGraph.tx().commit();

        this.sqlgGraph.close();
        try (SqlgGraph sqlgGraph = SqlgGraph.open(configuration)) {
            Map, Set>> localTabels = this.sqlgGraph.getSchemaManager().getLocalTableLabels();
            Assert.assertTrue(localTabels.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Person")));
            Assert.assertTrue(localTabels.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Car")));
            Assert.assertTrue(localTabels.containsKey(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Model")));
            Pair, Set> person = localTabels.get(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Person"));
            Assert.assertEquals(0, person.getLeft().size());
            Assert.assertEquals(1, person.getRight().size());
            Assert.assertEquals(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_drives", person.getRight().iterator().next().toString());
            Pair, Set> car = localTabels.get(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Car"));
            Assert.assertEquals(1, car.getLeft().size());
            Assert.assertEquals(1, car.getRight().size());
            Assert.assertEquals(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_drives", car.getLeft().iterator().next().toString());

            Pair, Set> model = localTabels.get(SchemaTable.of(this.sqlgGraph.getSqlDialect().getPublicSchema(), "V_Model"));
            Assert.assertEquals(1, model.getLeft().size());
            Assert.assertEquals(0, model.getRight().size());
            Assert.assertEquals(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_model", model.getLeft().iterator().next().toString());
        }
    }
}