test: fix to use couchbase with testcontainers

Signed-off-by: Maximillian Arruda <dearrudam@gmail.com>
eclipse · May 23, 2023 · 4077a0d · 4077a0d
1 parent d958f83
commit 4077a0d
Show file tree

Hide file tree

Showing 2 changed files with 90 additions and 10 deletions.
diff --git a/...ase/src/test/java/org/eclipse/jnosql/databases/couchbase/communication/CouchbaseUtil.java b/...ase/src/test/java/org/eclipse/jnosql/databases/couchbase/communication/CouchbaseUtil.java
@@ -17,8 +17,13 @@
 import org.eclipse.jnosql.communication.Settings;
 import org.eclipse.jnosql.communication.SettingsBuilder;
 import org.eclipse.jnosql.communication.driver.ConfigurationReader;
+import org.eclipse.jnosql.mapping.config.MappingConfigurations;
 
+import java.util.HashMap;
 import java.util.Map;
+import java.util.function.Consumer;
+
+import static org.eclipse.jnosql.databases.couchbase.communication.Database.INSTANCE;
 
 public final class CouchbaseUtil {
 
@@ -30,9 +35,32 @@ private CouchbaseUtil() {
     }
 
     public static Settings getSettings() {
+        SettingsBuilder builder = getSettingsBuilder();
+        return builder.build();
+    }
+
+    public static SettingsBuilder getSettingsBuilder() {
         Map<String, String> map = ConfigurationReader.from(CouchbaseUtil.FILE_CONFIGURATION);
         SettingsBuilder builder = Settings.builder();
         map.forEach(builder::put);
-        return builder.build();
+        return builder;
     }
+
+    public static void systemPropertySetup(Consumer<CouchbaseSettings> consumer) {
+        Map<String, Object> configuration = new HashMap<>(ConfigurationReader.from("couchbase.properties"));
+        for (Map.Entry<String, Object> entry : configuration.entrySet()) {
+            System.setProperty(entry.getKey(), entry.getValue().toString());
+        }
+        CouchbaseSettings couchbaseSettings = INSTANCE.getCouchbaseSettings();
+        System.setProperty(CouchbaseConfigurations.HOST.get(), couchbaseSettings.getHost());
+        consumer.accept(couchbaseSettings);
+    }
+
+    public static void systemPropertySetup() {
+        systemPropertySetup(settings -> {
+            System.setProperty(MappingConfigurations.DOCUMENT_DATABASE.get(), CouchbaseUtil.BUCKET_NAME);
+            System.setProperty(MappingConfigurations.DOCUMENT_PROVIDER.get(), CouchbaseDocumentConfiguration.class.getName());
+        });
+    }
+
 }
diff --git a/...ouchbase/src/test/java/org/eclipse/jnosql/databases/couchbase/communication/Database.java b/...ouchbase/src/test/java/org/eclipse/jnosql/databases/couchbase/communication/Database.java
@@ -14,28 +14,45 @@
  */
 package org.eclipse.jnosql.databases.couchbase.communication;
 
+import org.eclipse.jnosql.communication.Settings;
+import org.eclipse.jnosql.communication.SettingsBuilder;
 import org.testcontainers.couchbase.BucketDefinition;
 import org.testcontainers.couchbase.CouchbaseContainer;
 
+import java.util.Optional;
+
 public enum Database {
 
     INSTANCE;
 
-    private CouchbaseContainer container;
+    private final CouchbaseContainer container;
+    private final CouchbaseSettings settings;
 
     Database() {
-        //TODO create couchbase container to run all tests instead of run an outside container
-        BucketDefinition bucketDefinition = new BucketDefinition(CouchbaseUtil.BUCKET_NAME)
-                .withPrimaryIndex(true)
-                .withReplicas(0)
+
+        Settings fileSettings = CouchbaseUtil.getSettings();
+
+        CouchbaseDocumentConfiguration configuration = new CouchbaseDocumentConfiguration();
+
+        var bucketName = CouchbaseUtil.BUCKET_NAME;
+
+        BucketDefinition bucketDefinition = new BucketDefinition(bucketName)
                 .withFlushEnabled(true);
+
         container = new CouchbaseContainer("couchbase/server")
                 .withBucket(bucketDefinition)
-                .withExposedPorts(8091, 8092, 8093, 8094, 11207, 11210, 11211, 18091, 18092, 18093);
+                .withCredentials(configuration.getUser(fileSettings), configuration.getPassword(fileSettings))
+                .withReuse(true);
+
         container.start();
-        CouchbaseDocumentConfiguration configuration = setup(new CouchbaseDocumentConfiguration());
-        CouchbaseSettings settings = configuration.toCouchbaseSettings();
-        settings.setUp("jnosql");
+
+        configuration.update(fileSettings);
+
+        setup(configuration, configuration.toCouchbaseSettings());
+
+        settings = configuration.toCouchbaseSettings();
+
+        settings.setUp(bucketName);
     }
 
     public CouchbaseDocumentConfiguration getDocumentConfiguration() {
@@ -49,9 +66,44 @@ public CouchbaseKeyValueConfiguration getKeyValueConfiguration() {
     }
 
     private <T extends CouchbaseConfiguration> T setup(T configuration) {
+        return setup(configuration, this.settings);
+    }
+
+    private <T extends CouchbaseConfiguration> T setup(T configuration, CouchbaseSettings settings) {
         configuration.setHost(container.getConnectionString());
         configuration.setUser(container.getUsername());
         configuration.setPassword(container.getPassword());
+        settings.getCollection().ifPresent(configuration::setCollection);
+        settings.getCollections().stream().toList().forEach(configuration::addCollection);
+        settings.getScope().ifPresent(configuration::setScope);
         return configuration;
     }
+
+    public CouchbaseSettings getCouchbaseSettings() {
+        return settings;
+    }
+
+    public Settings getSettings() {
+        return getSettingsBuilder().build();
+    }
+
+    public SettingsBuilder getSettingsBuilder() {
+        CouchbaseSettings couchbaseSettings = getCouchbaseSettings();
+        SettingsBuilder builder = CouchbaseUtil.getSettingsBuilder();
+        builder.put(CouchbaseConfigurations.HOST.get(), couchbaseSettings.getHost());
+        couchbaseSettings.getScope().ifPresent(scope ->
+                builder.put(CouchbaseConfigurations.SCOPE.get(), scope));
+        Optional.ofNullable(couchbaseSettings.getIndex()).ifPresent(index ->
+                builder.put(CouchbaseConfigurations.INDEX.get(), index));
+        if (!couchbaseSettings.getCollections().isEmpty()) {
+            builder.put(CouchbaseConfigurations.COLLECTIONS.get(),
+                    String.join(",", couchbaseSettings.getCollections()));
+        }
+        Optional.ofNullable(couchbaseSettings.getUser()).ifPresent(user ->
+                builder.put(CouchbaseConfigurations.USER.get(), couchbaseSettings.getUser()));
+        Optional.ofNullable(couchbaseSettings.getPassword()).ifPresent(password ->
+                builder.put(CouchbaseConfigurations.PASSWORD.get(), password));
+        return builder;
+    }
+
 }