Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.umlg.sqlg.test.schema.TestSchemaEagerCreation Maven / Gradle / Ivy
package org.umlg.sqlg.test.schema;
import org.apache.tinkerpop.gremlin.structure.Direction;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assume;
import org.junit.Before;
import org.junit.Test;
import org.umlg.sqlg.structure.*;
import org.umlg.sqlg.test.BaseTest;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import static org.junit.Assert.*;
/**
* Date: 2016/11/23
* Time: 6:03 PM
*/
public class TestSchemaEagerCreation extends BaseTest {
@Before
public void beforeTest() {
Assume.assumeTrue(this.sqlgGraph.getSqlDialect().supportsTransactionalSchema());
}
@SuppressWarnings("OptionalGetWithoutIsPresent")
@Test
public void testVertexEdgeHasSameName() {
Vertex a1 = this.sqlgGraph.addVertex(T.label, "A", "name1", "halo1");
Vertex b2 = this.sqlgGraph.addVertex(T.label, "B", "name2", "halo2");
a1.addEdge("A", b2, "name3", "halo3");
this.sqlgGraph.tx().commit();
Map properties = this.sqlgGraph.getTopology().getPublicSchema().getVertexLabel("A").get().getProperties();
assertTrue(properties.containsKey("name1"));
properties = this.sqlgGraph.getTopology().getPublicSchema().getVertexLabel("B").get().getProperties();
assertTrue(properties.containsKey("name2"));
properties = this.sqlgGraph.getTopology().getPublicSchema().getEdgeLabel("A").get().getProperties();
assertTrue(properties.containsKey("name3"));
}
@Test
public void testModern() {
createModernSchema();
this.sqlgGraph.tx().rollback();
//test nothing is created
assertTrue(this.sqlgGraph.getTopology().getAllTables().isEmpty());
createModernSchema();
this.sqlgGraph.tx().commit();
assertEquals(4, this.sqlgGraph.getTopology().getAllTables().size());
assertEquals(2, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_person").size());
assertEquals(2, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_software").size());
assertEquals(PropertyType.STRING, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_person").get("name"));
assertEquals(PropertyType.INTEGER, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_person").get("age"));
assertEquals(PropertyType.STRING, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_software").get("name"));
assertEquals(PropertyType.STRING, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_software").get("lang"));
//test this by turning sql logging on and watch for create statements.
loadModern();
}
@Test
public void testAddVertexProperties() {
this.sqlgGraph.getTopology().ensureVertexLabelExist("A", Collections.emptyMap());
this.sqlgGraph.tx().commit();
assertEquals(1, this.sqlgGraph.getTopology().getAllTables().size());
assertTrue(this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_A").isEmpty());
Map properties = new HashMap<>();
properties.put("name", PropertyType.STRING);
properties.put("age", PropertyType.INTEGER);
assertTrue(this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_A").isEmpty());
this.sqlgGraph.getTopology().ensureVertexLabelPropertiesExist("A", properties);
this.sqlgGraph.tx().rollback();
assertTrue(this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_A").isEmpty());
this.sqlgGraph.getTopology().ensureVertexLabelExist("A", Collections.emptyMap());
this.sqlgGraph.getTopology().ensureVertexLabelPropertiesExist("A", properties);
this.sqlgGraph.tx().commit();
assertEquals(1, this.sqlgGraph.getTopology().getAllTables().size());
assertEquals(2, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_A").size());
assertEquals(PropertyType.STRING, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_A").get("name"));
assertEquals(PropertyType.INTEGER, this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".V_A").get("age"));
}
@Test
public void testAddEdgeProperties() {
VertexLabel outVertexLabel = this.sqlgGraph.getTopology().ensureVertexLabelExist("A");
VertexLabel inVertexLabel = this.sqlgGraph.getTopology().ensureVertexLabelExist("B");
this.sqlgGraph.getTopology().ensureEdgeLabelExist("ab", outVertexLabel, inVertexLabel, Collections.emptyMap());
this.sqlgGraph.tx().commit();
Map properties = new HashMap<>();
properties.put("name", PropertyType.STRING);
properties.put("age", PropertyType.INTEGER);
assertTrue(this.sqlgGraph.getTopology().getAllTables().entrySet().stream().allMatch((entry) -> entry.getValue().isEmpty()));
assertTrue(this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_ab").isEmpty());
this.sqlgGraph.getTopology().ensureEdgePropertiesExist("ab", properties);
this.sqlgGraph.tx().rollback();
assertTrue(this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_ab").isEmpty());
this.sqlgGraph.getTopology().ensureEdgePropertiesExist("ab", properties);
this.sqlgGraph.tx().commit();
assertFalse(this.sqlgGraph.getTopology().getAllTables().get(this.sqlgGraph.getSqlDialect().getPublicSchema() + ".E_ab").isEmpty());
}
@SuppressWarnings("OptionalGetWithoutIsPresent")
@Test
public void testVertexLabelPropertiesViaVertexLabel() {
Schema schema = this.sqlgGraph.getTopology().getPublicSchema();
this.sqlgGraph.addVertex(T.label, "Person");
Optional vertexLabel = schema.getVertexLabel("Person");
assertTrue(vertexLabel.isPresent());
this.sqlgGraph.tx().rollback();
vertexLabel = schema.getVertexLabel("Person");
assertFalse(vertexLabel.isPresent());
this.sqlgGraph.addVertex(T.label, "Person");
vertexLabel = schema.getVertexLabel("Person");
assertTrue(vertexLabel.isPresent());
this.sqlgGraph.tx().commit();
vertexLabel = schema.getVertexLabel("Person");
assertTrue(vertexLabel.isPresent());
vertexLabel = schema.getVertexLabel("Person");
assertTrue(vertexLabel.isPresent());
Map properties = new HashMap<>();
properties.put("name", PropertyType.STRING);
properties.put("age", PropertyType.INTEGER);
vertexLabel.get().ensurePropertiesExist(properties);
assertEquals(2, vertexLabel.get().getProperties().size());
this.sqlgGraph.tx().rollback();
assertEquals(0, vertexLabel.get().getProperties().size());
vertexLabel.get().ensurePropertiesExist(properties);
this.sqlgGraph.tx().commit();
assertEquals(2, vertexLabel.get().getProperties().size());
PropertyColumn propertyColumnName = vertexLabel.get().getProperties().get("name");
PropertyColumn propertyColumnAge = vertexLabel.get().getProperties().get("age");
assertNotNull(propertyColumnName);
assertNotNull(propertyColumnAge);
assertEquals(PropertyType.STRING, propertyColumnName.getPropertyType());
assertEquals(PropertyType.INTEGER, propertyColumnAge.getPropertyType());
}
@SuppressWarnings("OptionalGetWithoutIsPresent")
@Test
public void testEdgeLabelsProperties() {
Vertex a = this.sqlgGraph.addVertex(T.label, "A");
Vertex b = this.sqlgGraph.addVertex(T.label, "B");
a.addEdge("ab", b);
this.sqlgGraph.tx().commit();
Optional vertexLabelAOptional = this.sqlgGraph.getTopology().getPublicSchema().getVertexLabel("A");
assertTrue(vertexLabelAOptional.isPresent());
Optional vertexLabelBOptional = this.sqlgGraph.getTopology().getPublicSchema().getVertexLabel("B");
assertTrue(vertexLabelBOptional.isPresent());
Optional edgeLabelOptional = vertexLabelAOptional.get().getOutEdgeLabel("ab");
assertTrue(edgeLabelOptional.isPresent());
Map properties = new HashMap<>();
properties.put("name", PropertyType.STRING);
properties.put("age", PropertyType.INTEGER);
EdgeLabel edgeLabel = edgeLabelOptional.get();
edgeLabel.ensurePropertiesExist(properties);
this.sqlgGraph.tx().rollback();
edgeLabel = vertexLabelAOptional.get().getOutEdgeLabel("ab").get();
assertTrue(edgeLabel.getProperties().isEmpty());
edgeLabel = vertexLabelAOptional.get().getOutEdgeLabel("ab").get();
edgeLabel.ensurePropertiesExist(properties);
this.sqlgGraph.tx().commit();
edgeLabel = vertexLabelAOptional.get().getOutEdgeLabel("ab").get();
assertEquals(2, edgeLabel.getProperties().size());
}
@SuppressWarnings("OptionalGetWithoutIsPresent")
@Test
public void testEdgeLabelAddVertexLabels() {
Vertex a = this.sqlgGraph.addVertex(T.label, "A");
Vertex b = this.sqlgGraph.addVertex(T.label, "B");
a.addEdge("ab", b);
this.sqlgGraph.tx().commit();
Optional edgeLabelOptional = this.sqlgGraph.getTopology().getPublicSchema().getEdgeLabel("ab");
assertTrue(edgeLabelOptional.isPresent());
EdgeLabel edgeLabel = edgeLabelOptional.get();
VertexLabel inVertexLabel = this.sqlgGraph.getTopology().getPublicSchema().ensureVertexLabelExist("C");
edgeLabel.ensureEdgeVertexLabelExist(Direction.IN, inVertexLabel);
this.sqlgGraph.tx().rollback();
edgeLabelOptional = this.sqlgGraph.getTopology().getPublicSchema().getEdgeLabel("ab");
assertTrue(edgeLabelOptional.isPresent());
edgeLabel = edgeLabelOptional.get();
assertEquals(1, edgeLabel.getOutVertexLabels().size());
assertEquals(1, edgeLabel.getInVertexLabels().size());
inVertexLabel = this.sqlgGraph.getTopology().getPublicSchema().ensureVertexLabelExist("C");
edgeLabel.ensureEdgeVertexLabelExist(Direction.IN, inVertexLabel);
this.sqlgGraph.tx().commit();
assertEquals(1, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
VertexLabel outVertexLabel = this.sqlgGraph.getTopology().ensureSchemaExist("D").ensureVertexLabelExist("D");
try {
edgeLabel.ensureEdgeVertexLabelExist(Direction.OUT, outVertexLabel);
fail("Should fail as the out vertex is in a different schema to the edgeLabel");
} catch (IllegalStateException e) {
//swallow
}
assertEquals(1, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
this.sqlgGraph.tx().rollback();
assertEquals(1, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
outVertexLabel = this.sqlgGraph.getTopology().getPublicSchema().ensureVertexLabelExist("D");
edgeLabel.ensureEdgeVertexLabelExist(Direction.OUT, outVertexLabel);
assertEquals(2, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
this.sqlgGraph.tx().rollback();
assertEquals(1, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
outVertexLabel = this.sqlgGraph.getTopology().getPublicSchema().ensureVertexLabelExist("D");
edgeLabel.ensureEdgeVertexLabelExist(Direction.OUT, outVertexLabel);
assertEquals(2, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
this.sqlgGraph.tx().commit();
assertEquals(2, edgeLabel.getOutVertexLabels().size());
assertEquals(2, edgeLabel.getInVertexLabels().size());
}
@SuppressWarnings("OptionalGetWithoutIsPresent")
@Test
public void testAddEdgeLabelViaOutVertexLabel() {
VertexLabel a = this.sqlgGraph.getTopology()
.ensureSchemaExist("A")
.ensureVertexLabelExist("A");
Optional schemaOptional = this.sqlgGraph.getTopology().getSchema("A");
assertTrue(schemaOptional.isPresent());
VertexLabel b = schemaOptional.get().ensureVertexLabelExist("B");
a.ensureEdgeLabelExist("ab", b);
this.sqlgGraph.tx().commit();
Optional edgeLabel = this.sqlgGraph.getTopology().getSchema("A").get().getEdgeLabel("ab");
assertTrue(edgeLabel.isPresent());
assertEquals("ab", edgeLabel.get().getLabel());
}
private void createModernSchema() {
Map properties = new HashMap<>();
properties.put("name", PropertyType.STRING);
properties.put("age", PropertyType.INTEGER);
VertexLabel personVertexLabel = this.sqlgGraph.getTopology().ensureVertexLabelExist("person", properties);
properties.remove("age");
properties.put("name", PropertyType.STRING);
properties.put("lang", PropertyType.STRING);
VertexLabel softwareVertexLabel = this.sqlgGraph.getTopology().ensureVertexLabelExist("software", properties);
properties.clear();
properties.put("weight", PropertyType.DOUBLE);
this.sqlgGraph.getTopology().ensureEdgeLabelExist("knows", personVertexLabel, personVertexLabel, properties);
this.sqlgGraph.getTopology().ensureEdgeLabelExist("created", personVertexLabel, softwareVertexLabel, properties);
}
}
