allow for multiple locations

resolves #128
dasniko · Feb 23, 2024 · 13849db · 13849db
1 parent a35be7e
commit 13849db
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/src/main/java/dasniko/testcontainers/keycloak/ExtendableKeycloakContainer.java b/src/main/java/dasniko/testcontainers/keycloak/ExtendableKeycloakContainer.java
@@ -45,6 +45,7 @@
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
+import java.util.concurrent.atomic.AtomicInteger;
 
 import static java.util.Objects.requireNonNull;
 
@@ -101,7 +102,7 @@ public abstract class ExtendableKeycloakContainer<SELF extends ExtendableKeycloa
 
     private Duration startupTimeout = DEFAULT_STARTUP_TIMEOUT;
 
-    private String providerClassLocation;
+    private List<String> providerClassLocations;
     private List<File> providerLibsLocations;
 
     /**
@@ -171,8 +172,15 @@ protected void configure() {
             commandParts.add("--https-client-auth=" + this.httpsClientAuth);
         }
 
-        if (providerClassLocation != null) {
-            createKeycloakExtensionProvider(providerClassLocation);
+        if (providerClassLocations != null && !providerClassLocations.isEmpty()) {
+            AtomicInteger index = new AtomicInteger(0);
+            providerClassLocations.forEach(providerClassLocation -> {
+                createKeycloakExtensionDeployment(
+                    DEFAULT_KEYCLOAK_PROVIDERS_LOCATION,
+                    index.getAndIncrement() + "_" + DEFAULT_KEYCLOAK_PROVIDERS_NAME,
+                    providerClassLocation
+                );
+            });
         }
 
         if (providerLibsLocations != null) {
@@ -304,12 +312,12 @@ public SELF withContextPath(String contextPath) {
     }
 
     /**
-     * Exposes the given classes location as an exploded providers.jar.
+     * Exposes the given classes locations as an exploded providers.jar.
      *
-     * @param classesLocation a classes location relative to the current classpath root.
+     * @param classesLocations classes locations relative to the current classpath root.
      */
-    public SELF withProviderClassesFrom(String classesLocation) {
-        this.providerClassLocation = classesLocation;
+    public SELF withProviderClassesFrom(String... classesLocations) {
+        this.providerClassLocations = Arrays.asList(classesLocations);
         return self();
     }