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=&quot;tlee(a)redhat.com&quot;&gt;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