added additional test cases for the other stomp verbs/features

git-svn-id: https://svn.apache.org/repos/asf/incubator/activemq/trunk@370081 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Jonas B. Lim 2006-01-18 09:10:49 +00:00
parent 85402ce387
commit 377780985c
1 changed files with 250 additions and 31 deletions

View File

@ -22,17 +22,19 @@ import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.net.URI;
import java.net.SocketTimeoutException;
import javax.jms.Connection;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.jms.*;
import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.activemq.CombinationTestSupport;
import org.apache.activemq.broker.BrokerService;
import org.apache.activemq.broker.TransportConnector;
import org.apache.activemq.broker.region.policy.SimpleDispatchPolicy;
import org.apache.activemq.broker.region.policy.PolicyEntry;
import org.apache.activemq.broker.region.policy.RoundRobinDispatchPolicy;
import org.apache.activemq.broker.region.policy.PolicyMap;
import org.apache.activemq.command.ActiveMQQueue;
import org.apache.activemq.transport.stomp.Stomp;
@ -49,6 +51,7 @@ public class StompTest extends CombinationTestSupport {
protected void setUp() throws Exception {
broker = new BrokerService();
broker.setPersistent(false);
connector = broker.addConnector("stomp://localhost:0");
broker.start();
@ -61,6 +64,8 @@ public class StompTest extends CombinationTestSupport {
session = connection.createSession(false,Session.AUTO_ACKNOWLEDGE);
queue = new ActiveMQQueue("TEST");
connection.start();
}
protected void tearDown() throws Exception {
@ -95,9 +100,17 @@ public class StompTest extends CombinationTestSupport {
}
}
}
public void sendMessage(String msg) throws Exception {
MessageProducer producer = session.createProducer(queue);
TextMessage message = session.createTextMessage(msg);
producer.send(message);
}
public void testConnect() throws Exception {
String connect_frame = "CONNECT\n" + "login: brianm\n" + "passcode: wombats\n" + "\n" + Stomp.NULL;
sendFrame(connect_frame);
@ -119,46 +132,252 @@ public class StompTest extends CombinationTestSupport {
frame = receiveFrame(10000);
assertTrue(frame.startsWith("CONNECTED"));
frame =
"SEND\n" +
"destination:/queue/TEST\n\n" +
"Hello World" +
frame =
"SEND\n" +
"destination:/queue/TEST\n\n" +
"Hello World" +
Stomp.NULL;
sendFrame(frame);
TextMessage message = (TextMessage) consumer.receive(1000);
assertNotNull(message);
assertEquals("Hello World", message.getText());
}
public void testSubscribeWithAutoAck() throws Exception {
String frame =
"CONNECT\n" +
"login: brianm\n" +
String frame =
"CONNECT\n" +
"login: brianm\n" +
"passcode: wombats\n\n"+
Stomp.NULL;
sendFrame(frame);
frame = receiveFrame(10000000);
frame = receiveFrame(100000);
assertTrue(frame.startsWith("CONNECTED"));
frame =
"SUBSCRIBE\n" +
frame =
"SUBSCRIBE\n" +
"destination:/queue/TEST\n" +
"ack:auto\n\n" +
"ack:auto\n\n" +
Stomp.NULL;
sendFrame(frame);
MessageProducer producer = session.createProducer(queue);
TextMessage message = session.createTextMessage(getName());
producer.send(message);
sendMessage(getName());
frame = receiveFrame(10000);
assertTrue(frame.startsWith("MESSAGE"));
frame =
"DISCONNECT\n" +
"\n\n"+
Stomp.NULL;
sendFrame(frame);
}
public void testSubscribeWithClientAck() throws Exception {
String frame =
"CONNECT\n" +
"login: brianm\n" +
"passcode: wombats\n\n"+
Stomp.NULL;
sendFrame(frame);
frame = receiveFrame(10000);
assertTrue(frame.startsWith("CONNECTED"));
frame =
"SUBSCRIBE\n" +
"destination:/queue/TEST\n" +
"ack:client\n\n"+
Stomp.NULL;
sendFrame(frame);
sendMessage(getName());
frame = receiveFrame(10000);
assertTrue(frame.startsWith("MESSAGE"));
frame =
"DISCONNECT\n" +
"\n\n"+
Stomp.NULL;
sendFrame(frame);
// message should be received since message was not acknowledged
MessageConsumer consumer = session.createConsumer(queue);
TextMessage message = (TextMessage) consumer.receive(1000);
assertNotNull(message);
assertTrue(message.getJMSRedelivered());
}
public void testUnsubscribe() throws Exception {
String frame =
"CONNECT\n" +
"login: brianm\n" +
"passcode: wombats\n\n"+
Stomp.NULL;
sendFrame(frame);
frame = receiveFrame(100000);
assertTrue(frame.startsWith("CONNECTED"));
frame =
"SUBSCRIBE\n" +
"destination:/queue/TEST\n" +
"ack:auto\n\n" +
Stomp.NULL;
sendFrame(frame);
//send a message to our queue
sendMessage(getName());
//receive message from socket
frame = receiveFrame(10000);
assertTrue(frame.startsWith("MESSAGE"));
//remove suscription
frame =
"UNSUBSCRIBE\n" +
"destination:/queue/TEST\n" +
"\n\n" +
Stomp.NULL;
sendFrame(frame);
//send a message to our queue
sendMessage(getName());
try {
frame = receiveFrame(1000);
fail("No message should have been received since subscription was removed");
}catch (SocketTimeoutException e){
}
}
public void testTransactionCommit() throws Exception {
String frame =
"CONNECT\n" +
"login: brianm\n" +
"passcode: wombats\n\n"+
Stomp.NULL;
sendFrame(frame);
String f = receiveFrame(1000);
assertTrue(f.startsWith("CONNECTED"));
frame =
"BEGIN\n" +
"transaction: tx1\n" +
"\n\n" +
Stomp.NULL;
sendFrame(frame);
frame =
"SEND\n" +
"destination:/queue/TEST\n" +
"transaction: tx1\n" +
"\n\n" +
"Hello World" +
Stomp.NULL;
sendFrame(frame);
frame =
"COMMIT\n" +
"transaction: tx1\n" +
"\n\n" +
Stomp.NULL;
sendFrame(frame);
// This test case is currently failing
MessageConsumer consumer = session.createConsumer(queue);
TextMessage message = (TextMessage) consumer.receive(1000);
assertNotNull(message);
}
public void testTransactionRollback() throws Exception {
String frame =
"CONNECT\n" +
"login: brianm\n" +
"passcode: wombats\n\n"+
Stomp.NULL;
sendFrame(frame);
String f = receiveFrame(1000);
assertTrue(f.startsWith("CONNECTED"));
frame =
"BEGIN\n" +
"transaction: tx1\n" +
"\n\n" +
Stomp.NULL;
sendFrame(frame);
frame =
"SEND\n" +
"destination:/queue/TEST\n" +
"transaction: tx1\n" +
"\n\n" +
"first message" +
Stomp.NULL;
sendFrame(frame);
//rollback first message
frame =
"ABORT\n" +
"transaction: tx1\n" +
"\n\n" +
Stomp.NULL;
sendFrame(frame);
frame =
"SEND\n" +
"destination:/queue/TEST\n" +
"transaction: tx1\n" +
"\n\n" +
"second message" +
Stomp.NULL;
sendFrame(frame);
frame =
"COMMIT\n" +
"transaction: tx1\n" +
"\n\n" +
Stomp.NULL;
sendFrame(frame);
// This test case is currently failing
//only second msg should be received since first msg was rolled back
MessageConsumer consumer = session.createConsumer(queue);
TextMessage message = (TextMessage) consumer.receive(1000);
assertNotNull(message);
assertEquals("second message", message.getText());
}
}