[hornetq-commits] JBoss hornetq SVN: r8409 - in branches/20-optimisation/tests/src/org/hornetq/tests/integration: transports and 1 other directories.

[do-not-reply at jboss.org]

Author: trustin
Date: 2009-11-26 06:55:55 -0500 (Thu, 26 Nov 2009)
New Revision: 8409

Added:
   branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/
   branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/netty/
   branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/netty/HornetQFrameDecoder2Test.java
Log:
Added a unit test case for HornetQFrameDecoder2

Added: branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/netty/HornetQFrameDecoder2Test.java
===================================================================
--- branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/netty/HornetQFrameDecoder2Test.java	                        (rev 0)
+++ branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/netty/HornetQFrameDecoder2Test.java	2009-11-26 11:55:55 UTC (rev 8409)
@@ -0,0 +1,112 @@
+/*
+ * Copyright 2009 Red Hat, Inc.
+ * Red Hat licenses this file to you under the Apache License, version
+ * 2.0 (the "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ * implied.  See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package org.hornetq.tests.integration.transports.netty;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Random;
+
+import org.hornetq.integration.transports.netty.HornetQFrameDecoder2;
+import org.hornetq.tests.util.UnitTestCase;
+import org.jboss.netty.buffer.ChannelBuffer;
+import org.jboss.netty.buffer.ChannelBuffers;
+import org.jboss.netty.buffer.DynamicChannelBuffer;
+import org.jboss.netty.handler.codec.embedder.DecoderEmbedder;
+
+/**
+ * A HornetQFrameDecoder2Test
+ *
+ * @author <a href="tlee at redhat.com">Trustin Lee</a>
+ * @version $Revision$, $Date$
+ */
+public class HornetQFrameDecoder2Test extends UnitTestCase
+{
+   private static final int MSG_CNT = 10000;
+   private static final int MSG_LEN = 1000;
+   private static final int FRAGMENT_MAX_LEN = 1500;
+   
+   private static final Random rand = new Random();
+
+   public void testOrdinaryFragmentation() throws Exception
+   {
+      final DecoderEmbedder<ChannelBuffer> decoder =
+         new DecoderEmbedder<ChannelBuffer>(new HornetQFrameDecoder2());
+
+      ChannelBuffer src = ChannelBuffers.buffer(MSG_CNT * (MSG_LEN + 4));
+      while (src.writerIndex() < src.capacity()) {
+         src.writeInt(MSG_LEN);
+         byte[] data = new byte[MSG_LEN];
+         rand.nextBytes(data);
+         src.writeBytes(data);
+      }
+
+      List<ChannelBuffer> packets = new ArrayList<ChannelBuffer>();
+      for (int i = 0; i < src.capacity();) {
+         int length = Math.min(rand.nextInt(FRAGMENT_MAX_LEN), src.capacity() - i);
+         packets.add(src.copy(i, length));
+         i += length;
+      }
+      
+      int cnt = 0;
+      for (int i = 0; i < packets.size(); i ++) {
+         ChannelBuffer p = packets.get(i);
+         decoder.offer(p.duplicate());
+         for (;;) {
+            ChannelBuffer frame = decoder.poll();
+            if (frame == null) {
+               break;
+            }
+            assertTrue("Produced frame must be a dynamic buffer",
+                       frame instanceof DynamicChannelBuffer);
+            assertEquals(4, frame.readerIndex());
+            assertEquals(MSG_LEN, frame.readableBytes());
+            assertEquals(src.slice(cnt * (MSG_LEN + 4) + 4, MSG_LEN), frame);
+            cnt ++;
+         }
+      }
+      assertEquals(MSG_CNT, cnt);
+   }
+
+   public void testExtremeFragmentation() throws Exception
+   {
+      final DecoderEmbedder<ChannelBuffer> decoder =
+         new DecoderEmbedder<ChannelBuffer>(new HornetQFrameDecoder2());
+
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 0 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 0 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 0 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 4 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 5 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 6 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 7 }));
+      assertNull(decoder.poll());
+      decoder.offer(ChannelBuffers.wrappedBuffer(new byte[] { 8 }));
+
+      ChannelBuffer frame = decoder.poll();
+      assertTrue("Produced frame must be a dynamic buffer",
+                 frame instanceof DynamicChannelBuffer);
+      assertEquals(4, frame.readerIndex());
+      assertEquals(4, frame.readableBytes());
+      assertEquals(5, frame.getByte(4));
+      assertEquals(6, frame.getByte(5));
+      assertEquals(7, frame.getByte(6));
+      assertEquals(8, frame.getByte(7));
+   }
+}


Property changes on: branches/20-optimisation/tests/src/org/hornetq/tests/integration/transports/netty/HornetQFrameDecoder2Test.java
___________________________________________________________________
Name: svn:keywords
   + Rev Date
Name: svn:eol-style
   + native