improved FrameMax test

Author: Alexandru Scvortov

src/com/rabbitmq/client/ConnectionFactory.java

@@ -335,7 +335,13 @@ protected FrameHandler createFrameHandler(Address addr)
         Socket socket = factory.createSocket();
         configureSocket(socket);
         socket.connect(new InetSocketAddress(hostName, portNumber));
-        return new SocketFrameHandler(socket);
+        return createFrameHandler(socket);
+    }
+
+    protected FrameHandler createFrameHandler(Socket sock)
+        throws IOException
+    {
+        return new SocketFrameHandler(sock);
     }
 
     /**

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

@@ -35,14 +35,9 @@
 
 import java.io.IOException;
 import java.net.Socket;
-import java.net.InetSocketAddress;
 
-import com.rabbitmq.client.Address;
 import com.rabbitmq.client.ConnectionFactory;
-import com.rabbitmq.client.MessageProperties;
-import com.rabbitmq.client.QueueingConsumer;
-import com.rabbitmq.client.QueueingConsumer.Delivery;
-import com.rabbitmq.client.impl.AMQConnection;
+import com.rabbitmq.client.GetResponse;
 import com.rabbitmq.client.impl.Frame;
 import com.rabbitmq.client.impl.FrameHandler;
 import com.rabbitmq.client.impl.SocketFrameHandler;
@@ -54,11 +49,9 @@ public class FrameMax extends BrokerTestCase {
      * than what Rabbit suggests. */
     final static int FRAME_MAX = 131008;
     final static int REAL_FRAME_MAX = FRAME_MAX - 8;
-    final static int TIMEOUT = 3000; /* Time to wait for messages. */
-    final static String EXCHANGE_NAME = "xchg1";
     final static String ROUTING_KEY = "something";
 
-    QueueingConsumer consumer;
+    private String queueName;
 
     public FrameMax() {
         connectionFactory = new MyConnectionFactory();
@@ -76,19 +69,8 @@ protected void setUp()
     protected void createResources()
         throws IOException
     {
-        channel.exchangeDeclare(EXCHANGE_NAME, "direct");
-        consumer = new QueueingConsumer(channel);
-        String queueName = channel.queueDeclare().getQueue();
-        channel.basicConsume(queueName, consumer);
-        channel.queueBind(queueName, EXCHANGE_NAME, ROUTING_KEY);
-    }
-
-    @Override
-    protected void releaseResources()
-        throws IOException
-    {
-        consumer = null;
-        channel.exchangeDelete(EXCHANGE_NAME);
+        queueName = channel.queueDeclare().getQueue();
+        channel.queueBind(queueName, "", ROUTING_KEY);
     }
 
     /* Frame content should be less or equal to frame-max - 8. */
@@ -101,8 +83,8 @@ public void testFrameSizes()
         /* Receive everything that was sent out. */
         while (howMuch > 0) {
             try {
-                Delivery delivery = consumer.nextDelivery(TIMEOUT);
-                howMuch -= delivery.getBody().length;
+                GetResponse response = channel.basicGet(queueName, false);
+                howMuch -= response.getBody().length;
             } catch (RuntimeException e) {
                 fail(e.toString());
             }
@@ -115,41 +97,18 @@ protected void produce(int howMuch)
     {
         while (howMuch > 0) {
             int size = (howMuch <= (REAL_FRAME_MAX)) ? howMuch : (REAL_FRAME_MAX);
-            publish(new byte[size]);
-            howMuch -= (REAL_FRAME_MAX);
+            basicPublishVolatile(new byte[size], ROUTING_KEY);
+            howMuch -= size;
         }
     }
 
-    /* Publish a non-persistant, non-immediate message. */
-    private void publish(byte[] msg)
-        throws IOException
-    {
-        channel.basicPublish(EXCHANGE_NAME, ROUTING_KEY,
-                             false, false,
-                             MessageProperties.MINIMAL_BASIC,
-                             msg);
-    }
-
     /* ConnectionFactory that uses MyFrameHandler rather than
      * SocketFrameHandler. */
     private static class MyConnectionFactory extends ConnectionFactory {
-        protected FrameHandler createFrameHandler(Address addr)
+        protected FrameHandler createFrameHandler(Socket sock)
             throws IOException
         {
-            String hostName = addr.getHost();
-            int portNumber = portOrDefault(addr.getPort());
-            Socket socket = getSocketFactory().createSocket();
-            configureSocket(socket);
-            socket.connect(new InetSocketAddress(hostName, portNumber));
-            return new MyFrameHandler(socket);
-        }
-
-        /* Copy-pasted from ConnectionFactory. Should be protected,
-         * rather than private. */
-        private int portOrDefault(int port){
-            if (port != USE_DEFAULT_PORT) return port;
-            else if (isSSL()) return DEFAULT_AMQP_OVER_SSL_PORT;
-            else return DEFAULT_AMQP_PORT;
+            return new MyFrameHandler(sock);
         }
     }