RabbitMQ: Support declaring an exchange within a vhost (testcontainer…

…s#2362)
quincy · May 28, 2020 · c32098f · c32098f
1 parent dd8cfef
commit c32098f
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/modules/rabbitmq/src/main/java/org/testcontainers/containers/RabbitMQContainer.java b/modules/rabbitmq/src/main/java/org/testcontainers/containers/RabbitMQContainer.java
@@ -345,6 +345,17 @@ public RabbitMQContainer withExchange(String name, String type) {
 
     public RabbitMQContainer withExchange(String name, String type, boolean autoDelete, boolean internal, boolean durable, Map<String, Object> arguments) {
         values.add(asList("rabbitmqadmin", "declare", "exchange",
+            "name=" + name,
+            "type=" + type,
+            "auto_delete=" + autoDelete,
+            "internal=" + internal,
+            "durable=" + durable,
+            "arguments=" + toJson(arguments)));
+        return self();
+    }
+
+    public RabbitMQContainer withExchange(String vhost, String name, String type, boolean autoDelete, boolean internal, boolean durable, Map<String, Object> arguments) {
+        values.add(asList("rabbitmqadmin", "--vhost=" + vhost, "declare", "exchange",
                 "name=" + name,
                 "type=" + type,
                 "auto_delete=" + autoDelete,

diff --git a/modules/rabbitmq/src/test/java/org/testcontainers/containers/RabbitMQContainerTest.java b/modules/rabbitmq/src/test/java/org/testcontainers/containers/RabbitMQContainerTest.java
@@ -19,6 +19,7 @@
 import java.security.NoSuchAlgorithmException;
 import java.security.UnrecoverableKeyException;
 import java.security.cert.CertificateException;
+import java.util.Collections;
 import javax.net.ssl.KeyManagerFactory;
 import javax.net.ssl.SSLContext;
 import javax.net.ssl.TrustManagerFactory;
@@ -91,6 +92,20 @@ public void shouldCreateRabbitMQContainerWithExchange() throws IOException, Inte
         }
     }
 
+    @Test
+    public void shouldCreateRabbitMQContainerWithExchangeInVhost() throws IOException, InterruptedException
+    {
+        try (RabbitMQContainer container = new RabbitMQContainer()) {
+            container.withVhost("test-vhost");
+            container.withExchange("test-vhost", "test-exchange", "direct", false, false, false, Collections.emptyMap());
+
+            container.start();
+
+            assertThat(container.execInContainer("rabbitmqctl", "list_exchanges", "-p", "test-vhost").getStdout())
+                .containsPattern("test-exchange\\s+direct");
+        }
+    }
+
     @Test
     public void shouldCreateRabbitMQContainerWithQueues() throws IOException, InterruptedException
     {