Apache
/
activemq
mirror of https://github.com/apache/activemq.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added test cases for UDP fragmentation 

git-svn-id: https://svn.apache.org/repos/asf/incubator/activemq/trunk@384640 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
James Strachan 2006-03-09 23:08:57 +00:00
parent 3d7f76719f
commit 2f1dcea144
2 changed files with 68 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
activemq-core/src/main/java/org/apache/activemq/transport/udp/CommandChannel.java
View File

@ -207,10 +207,10 @@ public class CommandChannel implements Service {
boolean lastFragment = false;
for (int fragment = 0, length = data.length; !lastFragment; fragment++) {
// write the header
writeBuffer.rewind();
writeBuffer.clear();
int chunkSize = writeBuffer.capacity() - headerMarshaller.getHeaderSize(header);
lastFragment = offset + chunkSize >= length;
if (lastFragment) {
if (chunkSize + offset > length) {
chunkSize = length - offset;
}
header.incrementCounter();

70
activemq-core/src/test/java/org/apache/activemq/transport/udp/UdpTestSupport.java
View File

@ -16,6 +16,9 @@
*/
package org.apache.activemq.transport.udp;
import org.apache.activemq.command.ActiveMQDestination;
import org.apache.activemq.command.ActiveMQQueue;
import org.apache.activemq.command.ActiveMQTextMessage;
import org.apache.activemq.command.Command;
import org.apache.activemq.command.ConsumerInfo;
import org.apache.activemq.command.WireFormatInfo;
@ -25,6 +28,8 @@ import org.apache.activemq.transport.TransportFactory;
import org.apache.activemq.transport.TransportListener;
import org.apache.activemq.transport.TransportServer;
import javax.jms.MessageNotWriteableException;
import java.io.IOException;
import java.net.URI;
@ -41,7 +46,8 @@ public abstract class UdpTestSupport extends TestCase implements TransportListen
protected Object lock = new Object();
protected Command receivedCommand;
private TransportServer server;
protected TransportServer server;
protected boolean large;
public void testSendingSmallMessage() throws Exception {
ConsumerInfo expected = new ConsumerInfo();
@ -50,7 +56,7 @@ public abstract class UdpTestSupport extends TestCase implements TransportListen
expected.setCommandId((short) 12);
expected.setExclusive(true);
expected.setPrefetchSize(3456);
try {
System.out.println("About to send: " + expected);
producer.oneway(expected);
@ -70,6 +76,54 @@ public abstract class UdpTestSupport extends TestCase implements TransportListen
}
}
public void testSendingMediumMessage() throws Exception {
String text = createMessageBodyText(4 * 105);
ActiveMQDestination destination = new ActiveMQQueue("Foo.Bar.Medium");
assertSendTextMessage(destination, text);
}
public void testSendingLargeMessage() throws Exception {
String text = createMessageBodyText(4 * 1024);
ActiveMQDestination destination = new ActiveMQQueue("Foo.Bar.Large");
assertSendTextMessage(destination, text);
}
protected void assertSendTextMessage(ActiveMQDestination destination, String text) throws MessageNotWriteableException {
large = true;
ActiveMQTextMessage expected = new ActiveMQTextMessage();
expected.setText(text);
expected.setDestination(destination);
try {
System.out.println("About to send message of type: " + expected.getClass());
producer.oneway(expected);
Command received = assertCommandReceived();
assertTrue("Should have received a ActiveMQTextMessage but was: " + received, received instanceof ActiveMQTextMessage);
ActiveMQTextMessage actual = (ActiveMQTextMessage) received;
assertEquals("getDestination", expected.getDestination(), actual.getDestination());
assertEquals("getText", expected.getText(), actual.getText());
System.out.println("Received text message with: " + actual.getText().length() + " character(s)");
}
catch (Exception e) {
System.out.println("Caught: " + e);
e.printStackTrace();
fail("Failed to send to transport: " + e);
}
}
protected String createMessageBodyText(int loopSize) {
StringBuffer buffer = new StringBuffer();
for (int i = 0; i < loopSize; i++) {
buffer.append("0123456789");
}
return buffer.toString();
}
protected void setUp() throws Exception {
server = createServer();
if (server != null) {
@ -133,7 +187,12 @@ public abstract class UdpTestSupport extends TestCase implements TransportListen
System.out.println("Got WireFormatInfo: " + command);
}
else {
System.out.println("### Received command: " + command);
if (large) {
System.out.println("### Received command: " + command.getClass() + " with id: " + command.getCommandId());
}
else {
System.out.println("### Received command: " + command);
}
synchronized (lock) {
receivedCommand = command;
@ -157,7 +216,10 @@ public abstract class UdpTestSupport extends TestCase implements TransportListen
protected Command assertCommandReceived() throws InterruptedException {
Command answer = null;
synchronized (lock) {
lock.wait(1000);
answer = receivedCommand;
if (answer == null) {
lock.wait(10000);
}
answer = receivedCommand;
}