creates a container supplier

Author: otaviojava

riak-driver/src/test/java/org/jnosql/diana/riak/key/BucketManagerFactorySupplier.java

@@ -0,0 +1,46 @@
+/*
+ *  Copyright (c) 2019 Otávio Santana and others
+ *   All rights reserved. This program and the accompanying materials
+ *   are made available under the terms of the Eclipse Public License v1.0
+ *   and Apache License v2.0 which accompanies this distribution.
+ *   The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
+ *   and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
+ *
+ *   You may elect to redistribute this code under either of these licenses.
+ *
+ *   Contributors:
+ *
+ *   Otavio Santana
+ */
+package org.jnosql.diana.riak.key;
+
+import org.jnosql.diana.api.Settings;
+import org.jnosql.diana.api.key.BucketManagerFactory;
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.containers.wait.strategy.Wait;
+
+import java.util.function.Supplier;
+
+public enum BucketManagerFactorySupplier implements Supplier<BucketManagerFactory> {
+
+    INSTANCE;
+
+    private final GenericContainer riak =
+            new GenericContainer("basho/riak-ts:latest")
+                    .withExposedPorts(8087)
+                    .waitingFor(Wait.forListeningPort());
+
+    {
+        riak.start();
+    }
+
+    @Override
+    public BucketManagerFactory get() {
+        Settings settings = Settings.builder()
+                .put("riak.host", riak.getContainerIpAddress() + ':' + riak.getFirstMappedPort())
+                .build();
+        RiakKeyValueConfiguration configuration = new RiakKeyValueConfiguration();
+        return configuration.get(settings);
+    }
+
+}

riak-driver/src/test/java/org/jnosql/diana/riak/key/RiakTestUtils.java

@@ -15,17 +15,12 @@
 package org.jnosql.diana.riak.key;
 
 
-import com.basho.riak.client.core.RiakNode;
 import org.jnosql.diana.api.key.BucketManagerFactory;
 
 public final class RiakTestUtils {
 
 
     public static BucketManagerFactory get() {
-        RiakKeyValueConfiguration riakKeyValueConfiguration = new RiakKeyValueConfiguration();
-        RiakNode node = new RiakNode.Builder()
-                .withRemoteAddress("localhost").build();
-        riakKeyValueConfiguration.add(node);
-        return riakKeyValueConfiguration.get();
+        return BucketManagerFactorySupplier.INSTANCE.get();
     }
 }