Test durable message expiry with restart

Author: Emile Joubert

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

@@ -26,6 +26,7 @@
 public class PerMessageTTL extends TTLHandling {
 
     protected Object sessionTTL;
+    private volatile String retrievedMsg = null;
 
     @Override
     protected void publish(String msg) throws IOException {
@@ -58,4 +59,33 @@ public void testExpiryWhenConsumerIsLateToTheParty() throws Exception {
         assertNull("Message should have been expired!!", c.nextDelivery(100));
     }
 
+    public void testRestartingExpiry() throws Exception {
+        final String restartDelay = "3000";
+        declareDurableQueue(TTL_QUEUE_NAME);
+        bindQueue();
+        channel.basicPublish(TTL_EXCHANGE, TTL_QUEUE_NAME,
+                MessageProperties.MINIMAL_PERSISTENT_BASIC
+                        .builder()
+                        .expiration(restartDelay)
+                        .build(), new byte[]{});
+
+        Thread delayedConsume =
+            new Thread(new Runnable() {
+                @Override
+                public void run() {
+                    try {
+                        Thread.sleep(Integer.parseInt(restartDelay));
+                        retrievedMsg = get();
+                    } catch (IOException e) {
+                    } catch (InterruptedException e) {
+                    }
+
+                }
+            });
+        delayedConsume.start();
+        restart();
+        delayedConsume.join();
+        assertNull("Message should have expired after broker restart", retrievedMsg);
+    }
+
 }