package com.arangodb.tinkerpop.gremlin;
import java.util.HashSet;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import com.arangodb.tinkerpop.gremlin.structure.*;
import org.apache.commons.configuration.Configuration;
import org.apache.commons.configuration.ConfigurationConverter;
import org.apache.tinkerpop.gremlin.AbstractGraphProvider;
import org.apache.tinkerpop.gremlin.LoadGraphWith;
import org.apache.tinkerpop.gremlin.LoadGraphWith.GraphData;
import org.apache.tinkerpop.gremlin.structure.Element;
import org.apache.tinkerpop.gremlin.structure.Graph;
import com.arangodb.tinkerpop.gremlin.client.ArangoDBGraphClient;
import com.arangodb.tinkerpop.gremlin.utils.ArangoDBConfigurationBuilder;
/**
* The Class ArangoDBGraphProvider. This provider assumes that there is a local ArangoDB running (i.e.
* http://127.0.0.1:8529) with a tinkerpop database and a gremlin user that has Administrate permissions
* on the db.
*
*/
public class ArangoDBGraphProvider extends AbstractGraphProvider {
/** The Constant IMPLEMENTATIONS. */
private static final Set<Class> IMPLEMENTATIONS = new HashSet<Class>() {{
add(ArangoDBEdge.class);
add(ArangoDBGraph.class);
add(ArangoDBGraphVariables.class);
add(ArangoDBEdgeProperty.class);
add(ArangoDBElementProperty.class);
add(ArangoDBPropertyProperty.class);
add(ArangoDBVertex.class);
add(ArangoDBVertexProperty.class);
}};
@Override
public Configuration newGraphConfiguration(final String graphName, final Class<?> test,
final String testMethodName,
final Map<String, Object> configurationOverrides,
final LoadGraphWith.GraphData loadGraphWith) {
Configuration conf = getConfiguration(graphName, test, testMethodName, loadGraphWith);
// assign overrides but don't allow gremlin.graph setting to be overridden. the test suite should
// not be able to override that.
configurationOverrides.entrySet().stream()
.filter(c -> !c.getKey().equals(Graph.GRAPH))
.forEach(e -> conf.setProperty(e.getKey(), e.getValue()));
return conf;
}
private Configuration getConfiguration(
String graphName,
Class<?> test,
String testMethodName,
GraphData loadGraphWith) {
ArangoDBConfigurationBuilder builder = new ArangoDBConfigurationBuilder();
builder.graph(graphName);
if (loadGraphWith != null) {
switch(loadGraphWith) {
case CLASSIC:
System.out.println("CLASSIC");
builder.withEdgeCollection("knows");
builder.withEdgeCollection("created");
builder.configureEdge("knows", "vertex", "vertex");
builder.configureEdge("created", "vertex", "vertex");
break;
case CREW:
System.out.println("CREW");
builder.withVertexCollection("software");
builder.withVertexCollection("person");
builder.withEdgeCollection("uses");
builder.withEdgeCollection("develops");
builder.withEdgeCollection("traverses");
builder.configureEdge("uses", "person", "software");
builder.configureEdge("develops", "person", "software");
builder.configureEdge("traverses", "software", "software");
break;
case GRATEFUL:
System.out.println("GRATEFUL");
break;
case MODERN:
System.out.println("MODERN");
builder.withVertexCollection("software");
builder.withVertexCollection("person");
builder.withEdgeCollection("knows");
builder.withEdgeCollection("created");
builder.configureEdge("knows", "person", "person");
builder.configureEdge("created", "person", "software");
break;
default:
System.out.println("default");
break;
}
}
else {
if (testMethodName.startsWith("shouldProcessVerticesEdges")
|| testMethodName.startsWith("shouldGenerate")
|| testMethodName.startsWith("shouldSetValueOnEdge")
|| testMethodName.startsWith("shouldAutotype")) {
builder.withEdgeCollection("knows");
}
else if(testMethodName.startsWith("shouldIterateEdgesWithStringIdSupport")) {
builder.withEdgeCollection("self");
}
else if(testMethodName.startsWith("shouldSupportUserSuppliedIds")) {
builder.withEdgeCollection("test");
}
else if(testMethodName.startsWith("shouldSupportUUID")) {
builder.withEdgeCollection("friend");
}
else if(testMethodName.startsWith("shouldReadWriteVertexWithBOTHEdges")) {
builder.withEdgeCollection("friends");
}
else if(testMethodName.startsWith("shouldReadWriteVertexWithINEdges")) {
builder.withEdgeCollection("friends");
}
else if(testMethodName.startsWith("shouldReadWriteVertexMultiPropsNoEdges")) {
builder.withEdgeCollection("friends");
}
else if(testMethodName.startsWith("shouldReadWriteDetachedVertexAsReferenceNoEdges")) {
builder.withEdgeCollection("friends");
}
else if(testMethodName.startsWith("shouldReadWriteVertexNoEdges")) {
builder.withEdgeCollection("friends");
}
else if(testMethodName.startsWith("shouldReadWriteVertexWithOUTEdges")) {
builder.withEdgeCollection("friends");
}
else if(testMethodName.startsWith("shouldReadWriteDetachedVertexNoEdges")) {
builder.withEdgeCollection("friends");
}
else {
// Perhaps change for startsWith, but then it would be more verbose. Perhaps a set?
switch (testMethodName) {
case "shouldGetPropertyKeysOnEdge":
case "shouldNotGetConcurrentModificationException":
builder.withEdgeCollection("friend");
break;
case "shouldTraverseInOutFromVertexWithMultipleEdgeLabelFilter":
case "shouldTraverseInOutFromVertexWithSingleEdgeLabelFilter":
builder.withEdgeCollection("hate");
builder.withEdgeCollection("friend");
break;
case "shouldPersistDataOnClose":
builder.withEdgeCollection("collaborator");
break;
case "shouldTestTreeConnectivity":
builder.withEdgeCollection("test1");
builder.withEdgeCollection("test2");
builder.withEdgeCollection("test3");
break;
case "shouldEvaluateConnectivityPatterns":
builder.withEdgeCollection("knows");
builder.withEdgeCollection("knows");
break;
case "shouldRemoveEdgesWithoutConcurrentModificationException":
builder.withEdgeCollection("link");
break;
case "shouldGetValueThatIsNotPresentOnEdge":
case "shouldHaveStandardStringRepresentationForEdgeProperty":
case "shouldHaveTruncatedStringRepresentationForEdgeProperty":
case "shouldValidateIdEquality":
case "shouldValidateEquality":
builder.withEdgeCollection("self");
break;
case "shouldAllowRemovalFromEdgeWhenAlreadyRemoved":
case "shouldRespectWhatAreEdgesAndWhatArePropertiesInMultiProperties":
case "shouldProcessEdges":
case "shouldReturnOutThenInOnVertexIterator":
case "shouldReturnEmptyIteratorIfNoProperties":
builder.withEdgeCollection("knows");
break;
case "shouldNotHaveAConcurrentModificationExceptionWhenIteratingAndRemovingAddingEdges":
builder.withEdgeCollection("knows");
builder.withEdgeCollection("pets");
builder.withEdgeCollection("walks");
builder.withEdgeCollection("livesWith");
break;
case "shouldHaveStandardStringRepresentation":
builder.withEdgeCollection("friends");
break;
case "shouldReadWriteSelfLoopingEdges":
builder.withEdgeCollection("CONTROL");
builder.withEdgeCollection("SELFLOOP");
break;
case "shouldReadGraphML":
case "shouldReadGraphMLUnorderedElements":
case "shouldTransformGraphMLV2ToV3ViaXSLT":
case "shouldReadLegacyGraphSON":
builder.withEdgeCollection("knows");
builder.withEdgeCollection("created");
break;
default:
System.out.println("case \"" + testMethodName + "\":");
}
}
}
return builder.build();
}
@Override
public void clear(Graph graph, Configuration configuration) throws Exception {
ArangoDBGraphClient client;
if (graph ==null) {
Configuration arangoConfig = configuration.subset(ArangoDBGraph.PROPERTY_KEY_PREFIX);
Properties arangoProperties = ConfigurationConverter.getProperties(arangoConfig);
client = new ArangoDBGraphClient(null, arangoProperties, "tinkerpop", 0, true);
client.deleteGraph(arangoConfig.getString(ArangoDBGraph.PROPERTY_KEY_GRAPH_NAME));
}
else {
ArangoDBGraph agraph = (ArangoDBGraph) graph;
client = agraph.getClient();
client.clear(agraph);
agraph.close();
}
}
@Override
public Set<Class> getImplementations() {
return IMPLEMENTATIONS;
}
@Override
public Map<String, Object> getBaseConfiguration(String graphName, Class<?> test, String testMethodName,
GraphData loadGraphWith) {
// TODO Auto-generated method stub
return null;
}
@Override
public Object convertId(Object id, Class<? extends Element> c) {
return id.toString();
}
}
