Coverage Summary for Class: TestValidateTopology (org.umlg.sqlg.test.topology)
| Class |
Method, %
|
Branch, %
|
Line, %
|
| TestValidateTopology |
100%
(10/10)
|
27.6%
(21/76)
|
79.7%
(118/148)
|
| TestValidateTopology$1 |
100%
(1/1)
|
100%
(2/2)
|
| TestValidateTopology$2 |
100%
(1/1)
|
100%
(2/2)
|
| TestValidateTopology$3 |
100%
(1/1)
|
100%
(2/2)
|
| TestValidateTopology$4 |
100%
(1/1)
|
100%
(2/2)
|
| Total |
100%
(14/14)
|
27.6%
(21/76)
|
80.8%
(126/156)
|
package org.umlg.sqlg.test.topology;
import org.apache.commons.configuration2.builder.fluent.Configurations;
import org.apache.commons.configuration2.ex.ConfigurationException;
import org.apache.tinkerpop.gremlin.structure.T;
import org.apache.tinkerpop.gremlin.structure.Vertex;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.umlg.sqlg.structure.*;
import org.umlg.sqlg.structure.topology.*;
import org.umlg.sqlg.test.BaseTest;
import java.net.URL;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.*;
/**
* Date: 2017/01/15
* Time: 4:30 PM
*/
public class TestValidateTopology extends BaseTest {
@BeforeClass
public static void beforeClass() {
URL sqlProperties = Thread.currentThread().getContextClassLoader().getResource("sqlg.properties");
try {
Configurations configs = new Configurations();
configuration = configs.properties(sqlProperties);
configuration.addProperty("validate.topology", true);
} catch (ConfigurationException e) {
throw new RuntimeException(e);
}
}
@Test
public void testChangingMultiplicityOnVertex() {
Schema publicSchema = this.sqlgGraph.getTopology().getPublicSchema();
publicSchema.ensureVertexLabelExist("A", new HashMap<>() {{
put("a", PropertyDefinition.of(PropertyType.STRING, Multiplicity.of(1, 1)));
}});
this.sqlgGraph.tx().commit();
Map<String, PropertyDefinition> properties = new HashMap<>();
properties.put("a", PropertyDefinition.of(PropertyType.STRING, Multiplicity.of(0, 1)));
try {
publicSchema.ensureVertexLabelExist("A", properties);
} catch (IllegalStateException e) {
if (isPostgres()) {
String message = String.format("Column '%s' with multiplicity '%s' and incoming property '%s' with multiplicity '%s' are incompatible.",
"public.A.a", Multiplicity.of(1, 1), "public.A.a", Multiplicity.of(0, 1));
Assert.assertEquals(e.getMessage(), message);
} else if (isHsqldb() || isH2()) {
String message = String.format("Column '%s' with multiplicity '%s' and incoming property '%s' with multiplicity '%s' are incompatible.",
"PUBLIC.A.a", Multiplicity.of(1, 1), "PUBLIC.A.a", Multiplicity.of(0, 1));
Assert.assertEquals(e.getMessage(), message);
}
}
this.sqlgGraph.tx().rollback();
}
@Test
public void testChangingMultiplicityOnEdge() {
Schema publicSchema = this.sqlgGraph.getTopology().getPublicSchema();
VertexLabel aVertexLabel = publicSchema.ensureVertexLabelExist("A", new HashMap<>() {{
put("a", PropertyDefinition.of(PropertyType.STRING, Multiplicity.of(1, 1)));
}});
VertexLabel bVertexLabel = publicSchema.ensureVertexLabelExist("B", new HashMap<>() {{
put("b", PropertyDefinition.of(PropertyType.STRING, Multiplicity.of(1, 1)));
}});
aVertexLabel.ensureEdgeLabelExist("ab", bVertexLabel, new HashMap<>() {{
put("ab", PropertyDefinition.of(PropertyType.STRING, Multiplicity.of(1, 1)));
}});
this.sqlgGraph.tx().commit();
Map<String, PropertyDefinition> properties = new HashMap<>();
properties.put("ab", PropertyDefinition.of(PropertyType.STRING, Multiplicity.of(0, 1)));
try {
aVertexLabel.ensureEdgeLabelExist("ab", bVertexLabel, properties);
} catch (IllegalStateException e) {
if (isPostgres()) {
String message = String.format("Column '%s' with multiplicity '%s' and incoming property '%s' with multiplicity '%s' are incompatible.",
"public.ab.ab", Multiplicity.of(1, 1), "public.ab.ab", Multiplicity.of(0, 1));
Assert.assertEquals(e.getMessage(), message);
} else if (isHsqldb() || isH2()) {
String message = String.format("Column '%s' with multiplicity '%s' and incoming property '%s' with multiplicity '%s' are incompatible.",
"PUBLIC.ab.ab", Multiplicity.of(1, 1), "PUBLIC.ab.ab", Multiplicity.of(0, 1));
Assert.assertEquals(e.getMessage(), message);
}
}
this.sqlgGraph.tx().rollback();
}
@Test
public void testSchemaDoesNotExist() {
this.sqlgGraph.addVertex(T.label, "A.A");
this.sqlgGraph.tx().commit();
Connection conn = this.sqlgGraph.tx().getConnection();
try (Statement statement = conn.createStatement()) {
if (this.sqlgGraph.getSqlDialect().needsSchemaDropCascade()) {
statement.execute("DROP SCHEMA " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + " CASCADE");
} else {
if (this.sqlgGraph.getSqlDialect().isMssqlServer()) {
statement.execute("DROP TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") +
"." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A"));
statement.execute("DROP SCHEMA " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A"));
} else {
statement.execute("DROP SCHEMA " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A"));
}
}
this.sqlgGraph.tx().commit();
} catch (SQLException e) {
Assert.fail(e.getMessage());
}
this.sqlgGraph.close();
try (SqlgGraph sqlgGraph1 = SqlgGraph.open(configuration)) {
Assert.assertEquals(1, sqlgGraph1.getTopology().getValidationErrors().size());
}
}
@Test
public void testVertexLabelDoesNotExist() {
this.sqlgGraph.addVertex(T.label, "A.A");
this.sqlgGraph.tx().commit();
Connection conn = this.sqlgGraph.tx().getConnection();
try (Statement statement = conn.createStatement()) {
statement.execute("DROP TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A") +
(this.sqlgGraph.getSqlDialect().supportsCascade() ? " CASCADE " : ""));
this.sqlgGraph.tx().commit();
} catch (SQLException e) {
Assert.fail(e.getMessage());
}
this.sqlgGraph.close();
try (SqlgGraph sqlgGraph1 = SqlgGraph.open(configuration)) {
Assert.assertEquals(1, sqlgGraph1.getTopology().getValidationErrors().size());
}
}
@Test
public void testEdgeLabelDoesNotExist() {
Vertex a = this.sqlgGraph.addVertex(T.label, "A.A");
Vertex b = this.sqlgGraph.addVertex(T.label, "B.B");
a.addEdge("ab", b);
this.sqlgGraph.tx().commit();
Connection conn = this.sqlgGraph.tx().getConnection();
try (Statement statement = conn.createStatement()) {
statement.execute("DROP TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("E_ab") +
(this.sqlgGraph.getSqlDialect().supportsCascade() ? " CASCADE " : ""));
this.sqlgGraph.tx().commit();
} catch (SQLException e) {
Assert.fail(e.getMessage());
}
this.sqlgGraph.close();
try (SqlgGraph sqlgGraph1 = SqlgGraph.open(configuration)) {
Assert.assertEquals(1, sqlgGraph1.getTopology().getValidationErrors().size());
}
}
@Test
public void testVertexLabelPropertyDoesNotExist() {
this.sqlgGraph.addVertex(T.label, "A.A", "name", "aaa");
this.sqlgGraph.tx().commit();
Connection conn = this.sqlgGraph.tx().getConnection();
try (Statement statement = conn.createStatement()) {
if (!this.sqlgGraph.getSqlDialect().isMssqlServer()) {
statement.execute("ALTER TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A") + " DROP " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("name"));
} else {
statement.execute("ALTER TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A") + " DROP COLUMN " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("name"));
}
this.sqlgGraph.tx().commit();
} catch (SQLException e) {
Assert.fail(e.getMessage());
}
this.sqlgGraph.close();
try (SqlgGraph sqlgGraph1 = SqlgGraph.open(configuration)) {
Assert.assertEquals(1, sqlgGraph1.getTopology().getValidationErrors().size());
}
}
@Test
public void testEdgeLabelPropertyDoesNotExist() {
Vertex a = this.sqlgGraph.addVertex(T.label, "A.A", "name", "aaa");
Vertex b = this.sqlgGraph.addVertex(T.label, "B.B", "name", "bbb");
a.addEdge("ab", b, "name", "asdadasdasd");
this.sqlgGraph.tx().commit();
Connection conn = this.sqlgGraph.tx().getConnection();
try (Statement statement = conn.createStatement()) {
if (!this.sqlgGraph.getSqlDialect().isMssqlServer()) {
statement.execute("ALTER TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("E_ab") + " DROP " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("name"));
} else {
statement.execute("ALTER TABLE " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("E_ab") + " DROP COLUMN " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("name"));
}
this.sqlgGraph.tx().commit();
} catch (SQLException e) {
Assert.fail(e.getMessage());
}
this.sqlgGraph.close();
try (SqlgGraph sqlgGraph1 = SqlgGraph.open(configuration)) {
Assert.assertEquals(1, sqlgGraph1.getTopology().getValidationErrors().size());
}
}
@Test
public void testIndexDoesNotExist() {
this.sqlgGraph.addVertex(T.label, "A.A", "name", "aaa");
List<PropertyColumn> properties = new ArrayList<>(this.sqlgGraph.getTopology().getSchema("A").get().getVertexLabel("A").get().getProperties().values());
this.sqlgGraph.getTopology().getSchema("A").get().getVertexLabel("A").get().ensureIndexExists(IndexType.UNIQUE, properties);
this.sqlgGraph.tx().commit();
Connection conn = this.sqlgGraph.tx().getConnection();
try (Statement statement = conn.createStatement()) {
String indexName = this.sqlgGraph.getSqlDialect().indexName(SchemaTable.of("A", "A"), Topology.VERTEX_PREFIX, Collections.singletonList("name"));
if (this.sqlgGraph.getSqlDialect().isMssqlServer()) {
statement.execute("DROP INDEX " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes(indexName));
} else if (this.sqlgGraph.getSqlDialect().isMariaDb()) {
statement.execute("DROP INDEX " +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes(indexName) +
" ON " +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("V_A"));
} else {
statement.execute("DROP INDEX " + this.sqlgGraph.getSqlDialect().maybeWrapInQoutes("A") + "." +
this.sqlgGraph.getSqlDialect().maybeWrapInQoutes(indexName));
}
this.sqlgGraph.tx().commit();
} catch (SQLException e) {
Assert.fail(e.getMessage());
}
this.sqlgGraph.close();
try (SqlgGraph sqlgGraph1 = SqlgGraph.open(configuration)) {
Assert.assertEquals(1, sqlgGraph1.getTopology().getValidationErrors().size());
}
}
}