Tests for DLX policy

Author: Simon MacMullen

test/src/com/rabbitmq/client/test/functional/Policies.java

@@ -1,5 +1,6 @@
 package com.rabbitmq.client.test.functional;
 
+import com.rabbitmq.client.GetResponse;
 import com.rabbitmq.client.MessageProperties;
 import com.rabbitmq.client.test.BrokerTestCase;
 import com.rabbitmq.tools.Host;
@@ -14,6 +15,7 @@
 public class Policies extends BrokerTestCase {
     @Override protected void createResources() throws IOException {
         Host.rabbitmqctl("set_policy AE ^has-ae {\\\"alternate-exchange\\\":\\\"ae\\\"}");
+        Host.rabbitmqctl("set_policy DLX ^has-dlx {\\\"dead-letter-exchange\\\":\\\"dlx\\\",\\\"dead-letter-routing-key\\\":\\\"rk\\\"}");
         channel.exchangeDeclare("has-ae", "fanout");
         Map<String, Object> args = new HashMap<String, Object>();
         args.put("alternate-exchange", "ae2");
@@ -37,8 +39,39 @@ public void testAlternateExchangeArgs() throws IOException {
         assertDelivered(q, 1);
     }
 
+    public void testDeadLetterExchange() throws IOException, InterruptedException {
+        Map<String, Object> args = new HashMap<String, Object>();
+        args.put("x-message-ttl", 0);
+        String src = channel.queueDeclare("has-dlx", false, true, false, args).getQueue();
+        String dest = channel.queueDeclare().getQueue();
+        channel.exchangeDeclare("dlx", "fanout", false, true, null);
+        channel.queueBind(dest, "dlx", "");
+        channel.basicPublish("", src, MessageProperties.BASIC, "".getBytes());
+        Thread.sleep(10);
+        GetResponse resp = channel.basicGet(dest, true);
+        assertEquals("rk", resp.getEnvelope().getRoutingKey());
+    }
+
+    // again the argument takes priority over the policy
+    public void testDeadLetterExchangeArgs() throws IOException, InterruptedException {
+        Map<String, Object> args = new HashMap<String, Object>();
+        args.put("x-message-ttl", 0);
+        args.put("x-dead-letter-exchange", "dlx2");
+        args.put("x-dead-letter-routing-key", "rk2");
+        String src = channel.queueDeclare("has-dlx-args", false, true, false, args).getQueue();
+        String dest = channel.queueDeclare().getQueue();
+        channel.exchangeDeclare("dlx2", "fanout", false, true, null);
+        channel.queueBind(dest, "dlx2", "");
+        channel.basicPublish("", src, MessageProperties.BASIC, "".getBytes());
+        Thread.sleep(10);
+        GetResponse resp = channel.basicGet(dest, true);
+        assertEquals("rk2", resp.getEnvelope().getRoutingKey());
+    }
+
     @Override protected void releaseResources() throws IOException {
         Host.rabbitmqctl("clear_policy AE");
+        Host.rabbitmqctl("clear_policy DLX");
         channel.exchangeDelete("has-ae");
+        channel.exchangeDelete("has-ae-args");
     }
 }