ConfigurationBuilder sets gremlim.graph property

ArangoDB-Community · Dec 25, 2018 · 271ce8b · 271ce8b
1 parent 95e5859
commit 271ce8b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/main/java/com/arangodb/tinkerpop/gremlin/utils/ArangoDBConfigurationBuilder.java b/src/main/java/com/arangodb/tinkerpop/gremlin/utils/ArangoDBConfigurationBuilder.java
@@ -18,6 +18,7 @@
 import org.apache.commons.configuration.BaseConfiguration;
 import org.apache.commons.lang3.tuple.ImmutableTriple;
 import org.apache.commons.lang3.tuple.Triple;
+import org.apache.tinkerpop.gremlin.structure.Graph;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -189,6 +190,7 @@ public BaseConfiguration build() {
 		if (!hosts.isEmpty()) {
 			config.addProperty(fullPropertyKey(PROPERTY_KEY_HOSTS), hosts.stream().collect(Collectors.joining(",")));
 		}
+		config.addProperty(Graph.GRAPH, ArangoDBGraph.class.getName());
 		return config;
 	}
 

diff --git a/src/test/java/com/arangodb/tinkerpop/gremlin/ArangoDBGraphProvider.java b/src/test/java/com/arangodb/tinkerpop/gremlin/ArangoDBGraphProvider.java
@@ -205,9 +205,7 @@ else if(testMethodName.startsWith("shouldReadWriteDetachedVertexNoEdges")) {
 				}
 			}
 		}
-		Configuration conf = builder.build();
-		conf.addProperty(Graph.GRAPH, ArangoDBGraph.class.getName());
-		return conf;
+		return builder.build();
 	}
 
 	@Override