test: make database public

Signed-off-by: Otavio Santana <otaviopolianasantana@gmail.com>
eclipse · May 19, 2023 · ed7e688 · ed7e688
1 parent 8e9196d
commit ed7e688
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/...b/src/test/java/org/eclipse/jnosql/databases/arangodb/communication/DocumentDatabase.java b/...b/src/test/java/org/eclipse/jnosql/databases/arangodb/communication/DocumentDatabase.java
@@ -24,12 +24,12 @@
 
 import java.util.function.Supplier;
 
-enum DocumentDatabase implements Supplier<ArangoDBDocumentManagerFactory> {
+public enum DocumentDatabase implements Supplier<ArangoDBDocumentManagerFactory> {
 
     INSTANCE;
 
-    private final GenericContainer arangodb =
-            new GenericContainer("arangodb/arangodb:latest")
+    private final GenericContainer<?> arangodb =
+            new GenericContainer<>("arangodb/arangodb:latest")
                     .withExposedPorts(8529)
                     .withEnv("ARANGO_NO_AUTH", "1")
                     .waitingFor(Wait.forHttp("/")
@@ -45,4 +45,13 @@ public ArangoDBDocumentManagerFactory get() {
         configuration.addHost(arangodb.getHost(), arangodb.getFirstMappedPort());
         return configuration.apply(Settings.builder().build());
     }
+
+    public ArangoDBDocumentManager get(String database) {
+        ArangoDBDocumentManagerFactory managerFactory = get();
+        return managerFactory.apply(database);
+    }
+
+    public String host() {
+        return arangodb.getHost() + ":" + arangodb.getFirstMappedPort();
+    }
 }