allow overriding proxy handle in test resource

traum-ferienwohnungen · Jun 24, 2020 · e80108a · e80108a
1 parent bad55ea
commit e80108a
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 5 deletions.
diff --git a/...tainers-proxy-mountebank/src/main/java/com/traum/mountebank/ContainerMountebankProxy.java b/...tainers-proxy-mountebank/src/main/java/com/traum/mountebank/ContainerMountebankProxy.java
@@ -20,6 +20,8 @@
  * #L%
  */
 
+import java.util.Collection;
+
 public class ContainerMountebankProxy extends MountebankProxy {
 
   private final MountebankContainer container;
@@ -71,4 +73,10 @@ public String getApiUrl() {
   public String getImposterAuthority(int imposterPort) throws IllegalArgumentException {
     return container.getContainerIpAddress() + ":" + getContainer().getMappedPort(imposterPort);
   }
+
+  @Override
+  public Collection<Integer> getImposterPorts() {
+    return container.getImposterPorts();
+  }
+
 }
diff --git a/...ntainers-proxy-mountebank/src/main/java/com/traum/mountebank/ExternalMountebankProxy.java b/...ntainers-proxy-mountebank/src/main/java/com/traum/mountebank/ExternalMountebankProxy.java
@@ -20,6 +20,7 @@
  * #L%
  */
 
+import java.util.Collection;
 import java.util.Map;
 
 /** Facade for an externally managed instance of mountebank. */
@@ -45,4 +46,10 @@ public String getImposterAuthority(int imposterPort) throws IllegalArgumentExcep
     }
     throw new IllegalArgumentException("No mapped authority for imposter port " + imposterPort);
   }
+
+  @Override
+  public Collection<Integer> getImposterPorts() {
+    return imposterAuthorities.keySet();
+  }
+
 }
diff --git a/testcontainers-proxy-mountebank/src/main/java/com/traum/mountebank/MountebankProxy.java b/testcontainers-proxy-mountebank/src/main/java/com/traum/mountebank/MountebankProxy.java
@@ -31,6 +31,7 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.StandardCopyOption;
+import java.util.Collection;
 
 public abstract class MountebankProxy {
 
@@ -99,4 +100,6 @@ public void saveImposters(Path impostersOutput, boolean forReplay) {
   public abstract String getApiUrl();
 
   public abstract String getImposterAuthority(int imposterPort) throws IllegalArgumentException;
+
+  public abstract Collection<Integer> getImposterPorts();
 }
diff --git a/...c/main/java/com/traum/mountebank/quarkus/MountebankProxyTestResourceLifecycleManager.java b/...c/main/java/com/traum/mountebank/quarkus/MountebankProxyTestResourceLifecycleManager.java
@@ -29,22 +29,25 @@
 public class MountebankProxyTestResourceLifecycleManager
     implements QuarkusTestResourceLifecycleManager {
 
-  private final ContainerMountebankProxy proxy;
+  private final MountebankProxy proxy;
 
   public MountebankProxyTestResourceLifecycleManager() {
     this(new Integer[0]);
   }
 
   public MountebankProxyTestResourceLifecycleManager(Integer... imposterPorts) {
-    proxy = new ContainerMountebankProxy(imposterPorts);
+    this(new ContainerMountebankProxy(imposterPorts));
+  }
+
+  public MountebankProxyTestResourceLifecycleManager(MountebankProxy proxy) {
+    this.proxy = proxy;
   }
 
   @Override
   public Map<String, String> start() {
-    proxy.start();
+    getProxy().start();
     System.setProperty(MountebankExtension.EXTERNAL_PROXY_API_URL_PROPERTY, proxy.getApiUrl());
-    proxy
-        .getContainer()
+    getProxy()
         .getImposterPorts()
         .forEach(
             port -> {