feat: #220 Allow adding env var to an ensemble containers

Signed-off-by: Laurent Broudoux <[email protected]>

Author: lbroudoux

src/main/java/io/github/microcks/testcontainers/MicrocksContainersEnsemble.java

@@ -27,7 +27,9 @@
 import org.testcontainers.utility.DockerImageName;
 
 import java.util.Arrays;
+import java.util.HashMap;
 import java.util.HashSet;
+import java.util.Map;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -51,6 +53,10 @@ public class MicrocksContainersEnsemble implements Startable {
    private MicrocksAsyncMinionContainer asyncMinion;
    private final MicrocksContainer microcks;
 
+   private final Map<String, String> postmanEnvVars = new HashMap();
+   private final Map<String, String> asyncMinionEnvVars = new HashMap();
+   private final Map<String, String> microcksEnvVars = new HashMap();
+
    /**
     * Build a new MicrocksContainersEnsemble with its base container image name as string. This image must
     * be compatible with quay.io/microcks/microcks-uber image.
@@ -94,6 +100,17 @@ public MicrocksContainersEnsemble(Network network, DockerImageName image) {
                   + ":" + MicrocksAsyncMinionContainer.MICROCKS_ASYNC_MINION_HTTP_PORT);
    }
 
+   /**
+    * Add an environment variable to Postman runtime container.
+    * @param key The env var name
+    * @param value The env var value
+    * @return self
+    */
+   public MicrocksContainersEnsemble withMicrocksEnv(String key, String value) {
+      microcksEnvVars.put(key, value);
+      return this;
+   }
+
    /**
     * Enable the Postman runtime container with default container image.
     * @return self
@@ -128,6 +145,17 @@ public MicrocksContainersEnsemble withPostman(DockerImageName image) {
       return this;
    }
 
+   /**
+    * Add an environment variable to Postman runtime container.
+    * @param key The env var name
+    * @param value The env var value
+    * @return self
+    */
+   public MicrocksContainersEnsemble withPostmanEnv(String key, String value) {
+      postmanEnvVars.put(key, value);
+      return this;
+   }
+
    /**
     * Enable the Async Feature container with default container image (deduced from Microcks main one).
     * @return self
@@ -160,6 +188,18 @@ public MicrocksContainersEnsemble withAsyncFeature(DockerImageName image) {
       return this;
    }
 
+   /**
+    * Add an environment variable to Async Minion container.
+    * @param key The env var name
+    * @param value The env var value
+    * @return self
+    */
+   public MicrocksContainersEnsemble withAsyncMinionEnv(String key, String value) {
+      ensureAsyncFeatureIsEnabled();
+      asyncMinionEnvVars.put(key, value);
+      return this;
+   }
+
    /**
     * Once the Async Feature is enabled, connects to a Kafka broker.
     * @param connection Connection details to a Kafka broker.
@@ -355,11 +395,14 @@ public MicrocksAsyncMinionContainer getAsyncMinionContainer() {
    @Override
    public void start() {
       // Sequential start to avoid resource contention on CI systems with weaker hardware.
+      microcksEnvVars.forEach(microcks::withEnv);
       microcks.start();
       if (postman != null) {
+         postmanEnvVars.forEach(postman::withEnv);
          postman.start();
       }
       if (asyncMinion != null) {
+         asyncMinionEnvVars.forEach(asyncMinion::withEnv);
          asyncMinion.start();
       }
    }