Test the "impersonator" tag.

Author: Simon MacMullen

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

@@ -18,23 +18,41 @@
 
 import com.rabbitmq.client.AMQP;
 import com.rabbitmq.client.test.BrokerTestCase;
+import com.rabbitmq.tools.Host;
 
 import java.io.IOException;
 
 public class UserIDHeader extends BrokerTestCase {
+    private static final AMQP.BasicProperties GOOD = new AMQP.BasicProperties.Builder().userId("guest").build();
+    private static final AMQP.BasicProperties BAD = new AMQP.BasicProperties.Builder().userId("not the guest, honest").build();
+
     public void testValidUserId() throws IOException {
-        AMQP.BasicProperties properties = new AMQP.BasicProperties.Builder().userId("guest").build();
-        channel.basicPublish("amq.fanout", "", properties, "".getBytes());
+        publish(GOOD);
     }
 
     public void testInvalidUserId() {
-        AMQP.BasicProperties properties = new AMQP.BasicProperties.Builder().userId("not the guest, honest").build();
         try {
-            channel.basicPublish("amq.fanout", "", properties, "".getBytes());
-            channel.queueDeclare(); // To flush the channel
+            publish(BAD);
             fail("Accepted publish with incorrect user ID");
         } catch (IOException e) {
             checkShutdownSignal(AMQP.PRECONDITION_FAILED, e);
         }
     }
+
+    public void testImpersonatedUserId() throws IOException {
+        Host.rabbitmqctl("set_user_tags guest administrator impersonator");
+        connection = null;
+        channel = null;
+        setUp();
+        try {
+            publish(BAD);
+        } finally {
+            Host.rabbitmqctl("set_user_tags guest administrator");
+        }
+    }
+
+    private void publish(AMQP.BasicProperties properties) throws IOException {
+        channel.basicPublish("amq.fanout", "", properties, "".getBytes());
+        channel.queueDeclare(); // To flush the channel
+    }
 }