Functional test for internal exchanges.

Author: Jerry Kuch

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

@@ -70,7 +70,7 @@ public static TestSuite suite() {
         suite.addTestSuite(RecoverAfterCancel.class);
         suite.addTestSuite(UnexpectedFrames.class);
         suite.addTestSuite(PerQueueTTL.class);
-        
+        suite.addTestSuite(InternalExchangeTest.class);
         return suite;
     }
 }

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

@@ -31,7 +31,9 @@
 
 package com.rabbitmq.client.test.functional;
 
+import com.rabbitmq.client.GetResponse;
 import com.rabbitmq.client.QueueingConsumer;
+import com.rabbitmq.client.ShutdownSignalException;
 import com.rabbitmq.client.test.BrokerTestCase;
 
 import java.io.IOException;
@@ -45,12 +47,16 @@
 //            -------            -------
 //          -/       \-        -/       \-
 //         /           \      /           \           +-------------+
-//         |    e0     +------|    e1     +-----------+    q1       |
+//         |    e0     +------|     e1    +-----------+    q1       |
 //         \           /      \           /           +-------------+
 //          -\       /-        -\       /-
 //            -------            -------
 //                              (internal)
 //
+// Where a non-internal exchange is bound to an internal exchange, which in
+// turn is bound to a queue.  A client should be able to publish to e0, but
+// not to e1, and publications to e0 should be delivered into q1.
+//
 public class InternalExchangeTest extends BrokerTestCase
 {
     private final String[] queues = new String[] { "q1" };
@@ -94,19 +100,30 @@ protected void releaseResources() throws IOException
     }
 
 
-    public void testOhForFucksSake() throws IOException, InterruptedException
+    public void testTryPublishingToInternalExchange()
+            throws IOException,
+                   InterruptedException
     {
-        System.out.println("Yes, there is a test here.");
+        byte[] testDataBody = "test-data".getBytes();
 
-        // Create a simple consumer to try to catch stuff we've published...
-        if(false)
-        {
-            QueueingConsumer consumer = new QueueingConsumer(channel);
-            channel.basicConsume("q1", false, consumer);
+        // We should be able to publish to the non-internal exchange as usual
+        // and see our message land in the queue...
+        channel.basicPublish("e0", "", null, testDataBody);
+        assertTrue(channel.isOpen());
+        GetResponse r = channel.basicGet("q1", true);
+        assertTrue(Arrays.equals(r.getBody(), testDataBody));
 
-            QueueingConsumer.Delivery delivery = consumer.nextDelivery();
-            // process delivery
-            channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
-        }
+
+        // Publishing to the internal exchange will not be allowed...
+        channel.basicPublish("e1", "", null, testDataBody);
+        Thread.sleep(250L);
+        assertFalse(channel.isOpen());
+        ShutdownSignalException sdse = channel.getCloseReason();
+        assertNotNull(sdse);
+        String message = sdse.getMessage();
+        assertTrue(message.contains("reply-code=403"));
+        assertTrue(message.contains("reply-text=ACCESS_REFUSED"));
+        assertTrue(message.contains("cannot publish to internal exchange"));
     }
+
 }