@Test
public void dropIntIndexTest() {
ThunderGraph thunderGraph = new ThunderGraph(this.dbPath);
try {
thunderGraph.createKeyIndex("name1", Vertex.class);
thunderGraph.commit();
Vertex v1 = thunderGraph.addVertex(null);
v1.setProperty("name1", 1);
Vertex v2 = thunderGraph.addVertex(null);
v2.setProperty("name1", 1);
Vertex v3 = thunderGraph.addVertex(null);
v3.setProperty("name1", 1);
thunderGraph.commit();
Assert.assertEquals(3, count(thunderGraph.getVertices("name1", 1)));
Assert.assertEquals(3, thunderGraph.getDbEntries(DbEnum.VERTEX_INT_INDEX));
thunderGraph.dropKeyIndex("name1", Vertex.class);
thunderGraph.commit();
// Still finds it just not via the index
Assert.assertEquals(0, thunderGraph.getDbEntries(DbEnum.VERTEX_INT_INDEX));
Assert.assertEquals(3, count(thunderGraph.getVertices("name1", 1)));
} finally {
thunderGraph.shutdown();
}
}
@Test
public void testRemove() {
ThunderGraph graph = new ThunderGraph(this.dbPath);
try {
graph.createKeyIndex("name", Vertex.class);
graph.commit();
Vertex v1 = graph.addVertex(null);
v1.setProperty("name", 1);
for (int i = 0; i < 10; i++) {
Vertex v2 = graph.addVertex(null);
v2.setProperty("name", 2);
Edge e = graph.addEdge(null, v1, v2, "label1");
e.setProperty("name", "cccc");
}
graph.commit();
Assert.assertEquals(11, count(graph.getVertices()));
Assert.assertEquals(1, count(graph.getVertices("name", 1)));
Assert.assertEquals(10, count(graph.getVertices("name", 2)));
Assert.assertEquals(10, count(graph.getEdges("name", "cccc")));
Iterator<Vertex> iter = graph.getVertices("name", 2).iterator();
Vertex v = iter.next();
Assert.assertEquals(1L, v.getId());
Assert.assertEquals(2, v.getProperty("name"));
iter.remove();
Assert.assertNull(graph.getVertex(1L));
Assert.assertEquals(9, count(iter));
} finally {
graph.shutdown();
}
}
@Test
public void testIndexIntOnEdge() {
ThunderGraph thunderGraph = new ThunderGraph(this.dbPath);
try {
thunderGraph.createKeyIndex("name1", Edge.class);
Vertex v1 = thunderGraph.addVertex(null);
Vertex v2 = thunderGraph.addVertex(null);
Vertex v3 = thunderGraph.addVertex(null);
Edge e1 = v1.addEdge("edge1", v2);
e1.setProperty("name1", 1);
Edge e2 = v1.addEdge("edge1", v3);
e2.setProperty("name1", 1);
thunderGraph.commit();
Assert.assertEquals(1, thunderGraph.getIndexedKeys(Edge.class).size());
Assert.assertEquals(0, count(thunderGraph.getEdges("name", 1).iterator()));
Assert.assertEquals(2, count(thunderGraph.getEdges("name1", 1).iterator()));
Assert.assertEquals(0, count(thunderGraph.getEdges("name1", 2).iterator()));
thunderGraph.printDb(DbEnum.EDGE_INT_INDEX);
} finally {
thunderGraph.shutdown();
}
}
@Test
public void testCursorRefreshOnFirst() {
ThunderGraph graph = new ThunderGraph(this.dbPath);
try {
graph.createKeyIndex("name", Vertex.class);
graph.commit();
Vertex v1 = graph.addVertex(null);
v1.setProperty("name", 1);
for (int i = 0; i < 10; i++) {
Vertex v2 = graph.addVertex(null);
v2.setProperty("name", 2);
Edge e = graph.addEdge(null, v1, v2, "label1");
e.setProperty("name", "cccc");
}
graph.commit();
Assert.assertEquals(11, count(graph.getVertices()));
Assert.assertEquals(1, count(graph.getVertices("name", 1)));
Assert.assertEquals(10, count(graph.getVertices("name", 2)));
Assert.assertEquals(10, count(graph.getEdges("name", "cccc")));
Iterator<Vertex> iter = graph.getVertices("name", 1).iterator();
// This will cause the transaction to be upgraded to a writable transaction.
// I.e. iter's cursor gets closed
graph.getEdges("name", "cccc").iterator().next().setProperty("name", "cccd");
Vertex v = iter.next();
Assert.assertEquals(1, v.getProperty("name"));
} finally {
graph.shutdown();
}
}
@Test
public void testReindexOnEdge() {
ThunderGraph thunderGraph = new ThunderGraph(this.dbPath);
try {
Vertex v1 = thunderGraph.addVertex(null);
Vertex v2 = thunderGraph.addVertex(null);
Vertex v3 = thunderGraph.addVertex(null);
Vertex v4 = thunderGraph.addVertex(null);
Edge e1 = v1.addEdge("label1", v2);
Edge e2 = v1.addEdge("label1", v3);
Edge e3 = v1.addEdge("label1", v4);
e1.setProperty("name1", 1);
e2.setProperty("name1", 1);
e3.setProperty("name1", 1);
thunderGraph.commit();
Assert.assertEquals(3, count(thunderGraph.getEdges("name1", 1)));
thunderGraph.printDb(DbEnum.EDGE_INT_INDEX);
thunderGraph.createKeyIndex("name1", Edge.class);
Assert.assertEquals(3, count(thunderGraph.getEdges("name1", 1)));
thunderGraph.printDb(DbEnum.EDGE_INT_INDEX);
} finally {
thunderGraph.shutdown();
}
}
@Test
public void testUpdateIndexedField() {
ThunderGraph thunderGraph = new ThunderGraph(this.dbPath);
try {
thunderGraph.createKeyIndex("name1", Vertex.class);
thunderGraph.commit();
Vertex v1 = thunderGraph.addVertex(null);
v1.setProperty("name1", 1);
Vertex v2 = thunderGraph.addVertex(null);
v2.setProperty("name1", 1);
Vertex v3 = thunderGraph.addVertex(null);
v3.setProperty("name1", 1);
thunderGraph.commit();
Assert.assertEquals(3, count(thunderGraph.getVertices("name1", 1)));
v1.setProperty("name1", 2);
thunderGraph.commit();
Assert.assertEquals(2, count(thunderGraph.getVertices("name1", 1)));
} finally {
thunderGraph.shutdown();
}
}
@Test
public void deleteIndexedVertex() {
ThunderGraph thunderGraph = new ThunderGraph(this.dbPath);
try {
thunderGraph.createKeyIndex("name", Vertex.class);
Vertex v1 = thunderGraph.addVertex(null);
v1.setProperty("name", 1);
thunderGraph.commit();
Assert.assertEquals(1, thunderGraph.getDbEntries(DbEnum.VERTEX_INT_INDEX));
v1.remove();
thunderGraph.commit();
Assert.assertEquals(0, thunderGraph.getDbEntries(DbEnum.VERTEX_INT_INDEX));
} finally {
thunderGraph.shutdown();
}
}
@Test
public void testIndexIntOnVertex() {
ThunderGraph thunderGraph = new ThunderGraph(this.dbPath);
try {
thunderGraph.createKeyIndex("name1", Vertex.class);
thunderGraph.createKeyIndex("name2", Vertex.class);
thunderGraph.commit();
Vertex v1 = thunderGraph.addVertex(null);
v1.setProperty("name1", 11);
v1.setProperty("name2", 12);
Vertex v2 = thunderGraph.addVertex(null);
v2.setProperty("name1", 21);
v2.setProperty("name2", 22);
thunderGraph.commit();
Set<String> indexedKeys = thunderGraph.getIndexedKeys(Vertex.class);
Assert.assertEquals(2, indexedKeys.size());
Assert.assertTrue(indexedKeys.contains("name1"));
Assert.assertTrue(indexedKeys.contains("name2"));
Iterable<Vertex> name11s = thunderGraph.getVertices("name1", 11);
Assert.assertEquals(1, countIter(name11s.iterator()));
Iterable<Vertex> name21s = thunderGraph.getVertices("name1", 21);
Assert.assertEquals(1, countIter(name21s.iterator()));
name11s = thunderGraph.getVertices("name1", 12);
Assert.assertEquals(0, countIter(name11s.iterator()));
name21s = thunderGraph.getVertices("name1", 22);
Assert.assertEquals(0, countIter(name21s.iterator()));
} finally {
thunderGraph.shutdown();
}
}