/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.test.messaging.jms;
import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Queue;
import javax.jms.Session;
import javax.jms.TextMessage;
import javax.jms.Topic;
import javax.management.ObjectName;
import javax.naming.InitialContext;
import org.jboss.jms.message.JBossMessage;
import org.jboss.test.messaging.MessagingTestCase;
import org.jboss.test.messaging.tools.ServerManagement;
/**
*
* A ScheduledDeliveryTest
*
* @author <a href="mailto:tim.fox@jboss.com">Tim Fox</a>
* @version <tt>$Revision: 2081 $</tt>
*
* $Id: ScheduledDeliveryTest.java 2081 2007-01-27 14:17:00Z timfox $
*
*/
public class ScheduledDeliveryTest extends MessagingTestCase
{
// Constants -----------------------------------------------------
// Static --------------------------------------------------------
// Attributes ----------------------------------------------------
protected InitialContext ic;
protected ConnectionFactory cf;
protected Queue queue;
protected Topic topic;
// Constructors --------------------------------------------------
public ScheduledDeliveryTest(String name)
{
super(name);
}
// Public --------------------------------------------------------
public void testScheduledDelivery() throws Exception
{
Connection conn = null;
try
{
conn = cf.createConnection();
Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageProducer prod = sess.createProducer(queue);
MessageConsumer cons = sess.createConsumer(queue);
conn.start();
//Send one scheduled
long now = System.currentTimeMillis();
TextMessage tm1 = sess.createTextMessage("testScheduled1");
tm1.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 7000);
prod.send(tm1);
//First send some non scheduled messages
TextMessage tm2 = sess.createTextMessage("testScheduled2");
prod.send(tm2);
TextMessage tm3 = sess.createTextMessage("testScheduled3");
prod.send(tm3);
TextMessage tm4 = sess.createTextMessage("testScheduled4");
prod.send(tm4);
//Now send some more scheduled messages
TextMessage tm5 = sess.createTextMessage("testScheduled5");
tm5.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 5000);
prod.send(tm5);
TextMessage tm6 = sess.createTextMessage("testScheduled6");
tm6.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 4000);
prod.send(tm6);
TextMessage tm7 = sess.createTextMessage("testScheduled7");
tm7.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 3000);
prod.send(tm7);
TextMessage tm8 = sess.createTextMessage("testScheduled8");
tm8.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 6000);
prod.send(tm8);
//And one scheduled with a -ve number
TextMessage tm9 = sess.createTextMessage("testScheduled9");
tm8.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, -3);
prod.send(tm9);
//First the non scheduled messages should be received
TextMessage rm1 = (TextMessage)cons.receive(250);
assertNotNull(rm1);
assertEquals(tm2.getText(), rm1.getText());
TextMessage rm2 = (TextMessage)cons.receive(250);
assertNotNull(rm2);
assertEquals(tm3.getText(), rm2.getText());
TextMessage rm3 = (TextMessage)cons.receive(250);
assertNotNull(rm3);
assertEquals(tm4.getText(), rm3.getText());
//Now the one with a scheduled with a -ve number
TextMessage rm5 = (TextMessage)cons.receive(250);
assertNotNull(rm5);
assertEquals(tm9.getText(), rm5.getText());
//Now the scheduled
TextMessage rm6 = (TextMessage)cons.receive(3250);
assertNotNull(rm6);
assertEquals(tm7.getText(), rm6.getText());
long now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 3000);
TextMessage rm7 = (TextMessage)cons.receive(1250);
assertNotNull(rm7);
assertEquals(tm6.getText(), rm7.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 4000);
TextMessage rm8 = (TextMessage)cons.receive(1250);
assertNotNull(rm8);
assertEquals(tm5.getText(), rm8.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 5000);
TextMessage rm9 = (TextMessage)cons.receive(1250);
assertNotNull(rm9);
assertEquals(tm8.getText(), rm9.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 6000);
TextMessage rm10 = (TextMessage)cons.receive(1250);
assertNotNull(rm10);
assertEquals(tm1.getText(), rm10.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 7000);
Message m = cons.receive(1000);
assertNull(m);
}
finally
{
if (conn != null)
{
conn.close();
}
}
}
public void testScheduledDeliveryWithRestart() throws Exception
{
Connection conn = null;
try
{
conn = cf.createConnection();
Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageProducer prod = sess.createProducer(queue);
//Send one scheduled
long now = System.currentTimeMillis();
TextMessage tm1 = sess.createTextMessage("testScheduled1");
tm1.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 7000);
prod.send(tm1);
//First send some non scheduled messages
TextMessage tm2 = sess.createTextMessage("testScheduled2");
prod.send(tm2);
TextMessage tm3 = sess.createTextMessage("testScheduled3");
prod.send(tm3);
TextMessage tm4 = sess.createTextMessage("testScheduled4");
prod.send(tm4);
//Now send some more scheduled messages
TextMessage tm5 = sess.createTextMessage("testScheduled5");
tm5.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 5000);
prod.send(tm5);
TextMessage tm6 = sess.createTextMessage("testScheduled6");
tm6.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 4000);
prod.send(tm6);
TextMessage tm7 = sess.createTextMessage("testScheduled7");
tm7.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 3000);
prod.send(tm7);
TextMessage tm8 = sess.createTextMessage("testScheduled8");
tm8.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, now + 6000);
prod.send(tm8);
//And one scheduled with a -ve number
TextMessage tm9 = sess.createTextMessage("testScheduled9");
tm8.setLongProperty(JBossMessage.JMS_JBOSS_SCHEDULED_DELIVERY_PROP_NAME, -3);
prod.send(tm9);
//Now stop the server and restart it
conn.close();
ServerManagement.stopServerPeer();
ServerManagement.startServerPeer();
// Messaging server restart implies new ConnectionFactory lookup
cf = (ConnectionFactory)ic.lookup("/ConnectionFactory");
ServerManagement.deployQueue("Queue");
conn = cf.createConnection();
conn.start();
sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer cons = sess.createConsumer(queue);
//First the non scheduled messages should be received
TextMessage rm1 = (TextMessage)cons.receive(250);
assertNotNull(rm1);
assertEquals(tm2.getText(), rm1.getText());
TextMessage rm2 = (TextMessage)cons.receive(250);
assertNotNull(rm2);
assertEquals(tm3.getText(), rm2.getText());
TextMessage rm3 = (TextMessage)cons.receive(250);
assertNotNull(rm3);
assertEquals(tm4.getText(), rm3.getText());
//Now the one with a scheduled with a -ve number
TextMessage rm5 = (TextMessage)cons.receive(250);
assertNotNull(rm5);
assertEquals(tm9.getText(), rm5.getText());
//Now the scheduled
TextMessage rm6 = (TextMessage)cons.receive(3250);
assertNotNull(rm6);
assertEquals(tm7.getText(), rm6.getText());
long now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 3000);
TextMessage rm7 = (TextMessage)cons.receive(1250);
assertNotNull(rm7);
assertEquals(tm6.getText(), rm7.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 4000);
TextMessage rm8 = (TextMessage)cons.receive(1250);
assertNotNull(rm8);
assertEquals(tm5.getText(), rm8.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 5000);
TextMessage rm9 = (TextMessage)cons.receive(1250);
assertNotNull(rm9);
assertEquals(tm8.getText(), rm9.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 6000);
TextMessage rm10 = (TextMessage)cons.receive(1250);
assertNotNull(rm10);
assertEquals(tm1.getText(), rm10.getText());
now2 = System.currentTimeMillis();
assertTrue(now2 - now >= 7000);
Message m = cons.receive(1000);
assertNull(m);
}
finally
{
if (conn != null)
{
conn.close();
}
}
}
public void testDelayedRedeliveryDefault() throws Exception
{
ObjectName serverPeerObjectName = ServerManagement.getServerPeerObjectName();
String queueObjectName = "jboss.messaging.destination:service=Queue,name=Queue";
Connection conn = null;
try
{
ServerManagement.setAttribute(new ObjectName(queueObjectName), "RedeliveryDelay", String.valueOf(0));
final long delay = 3000;
ServerManagement.setAttribute(serverPeerObjectName, "DefaultRedeliveryDelay", String.valueOf(delay));
conn = cf.createConnection();
Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageProducer prod = sess.createProducer(queue);
final int NUM_MESSAGES = 5;
for (int i = 0; i < NUM_MESSAGES; i++)
{
TextMessage tm = sess.createTextMessage("message" + i);
prod.send(tm);
}
log.info("Sent messages");
Session sess2 = conn.createSession(false, Session.CLIENT_ACKNOWLEDGE);
MessageConsumer cons = sess2.createConsumer(queue);
conn.start();
log.info("Started connection");
for (int i = 0; i < NUM_MESSAGES; i++)
{
TextMessage tm = (TextMessage)cons.receive(500);
assertNotNull(tm);
log.info("Got message:" + tm.getText());
assertEquals("message" + i, tm.getText());
}
//Now close the session
//This should cancel back to the queue with a delayed redelivery
long now = System.currentTimeMillis();
sess2.close();
log.info("Closed session");
Session sess3 = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer cons2 = sess3.createConsumer(queue);
for (int i = 0; i < NUM_MESSAGES; i++)
{
TextMessage tm = (TextMessage)cons2.receive(delay + 1000);
assertNotNull(tm);
log.info("Got message 2nd time: " + tm.getText());
long time = System.currentTimeMillis();
assertTrue(time - now >= delay);
assertTrue(time - now < delay + 1000);
}
TextMessage tm = (TextMessage)cons2.receive(1000);
assertNull(tm);
}
finally
{
if (conn != null)
{
conn.close();
}
ServerManagement.setAttribute(serverPeerObjectName, "DefaultRedeliveryDelay", "0");
}
}
public void testDelayedRedeliveryOverride() throws Exception
{
ObjectName serverPeerObjectName = ServerManagement.getServerPeerObjectName();
String queueObjectName = "jboss.messaging.destination:service=Queue,name=Queue";
Connection conn = null;
try
{
final long delay = 3000;
ServerManagement.setAttribute(new ObjectName(queueObjectName), "RedeliveryDelay", String.valueOf(delay));
ServerManagement.setAttribute(serverPeerObjectName, "DefaultRedeliveryDelay", String.valueOf(delay * 3));
conn = cf.createConnection();
Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageProducer prod = sess.createProducer(queue);
final int NUM_MESSAGES = 5;
for (int i = 0; i < NUM_MESSAGES; i++)
{
TextMessage tm = sess.createTextMessage("message" + i);
prod.send(tm);
}
Session sess2 = conn.createSession(false, Session.CLIENT_ACKNOWLEDGE);
MessageConsumer cons = sess2.createConsumer(queue);
conn.start();
for (int i = 0; i < NUM_MESSAGES; i++)
{
TextMessage tm = (TextMessage)cons.receive(500);
assertNotNull(tm);
assertEquals("message" + i, tm.getText());
}
//Now close the session
//This should cancel back to the queue with a delayed redelivery
long now = System.currentTimeMillis();
sess2.close();
Session sess3 = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer cons2 = sess3.createConsumer(queue);
for (int i = 0; i < NUM_MESSAGES; i++)
{
TextMessage tm = (TextMessage)cons2.receive(delay + 1000);
long time = System.currentTimeMillis();
assertTrue(time - now >= delay);
assertTrue(time - now < delay + 1000);
assertNotNull(tm);
}
TextMessage tm = (TextMessage)cons2.receive(1000);
assertNull(tm);
}
finally
{
if (conn != null)
{
conn.close();
}
ServerManagement.setAttribute(serverPeerObjectName, "DefaultRedeliveryDelay", "0");
}
}
// Package protected ---------------------------------------------
// Protected -----------------------------------------------------
protected void setUp() throws Exception
{
super.setUp();
ServerManagement.start("all");
ic = new InitialContext(ServerManagement.getJNDIEnvironment());
cf = (ConnectionFactory)ic.lookup("/ConnectionFactory");
ServerManagement.undeployQueue("Queue");
ServerManagement.undeployTopic("Topic");
ServerManagement.deployQueue("Queue");
ServerManagement.deployTopic("Topic");
queue = (Queue)ic.lookup("/queue/Queue");
topic = (Topic)ic.lookup("/topic/Topic");
this.drainDestination(cf, queue);
}
protected void tearDown() throws Exception
{
super.tearDown();
ServerManagement.undeployQueue("Queue");
ServerManagement.undeployTopic("Topic");
// Some tests here are changing this attribute.. what would affect tests later
// Instead of restart the ServerPeer I'm just restoring the default
ServerManagement.setAttribute(ServerManagement.getServerPeerObjectName(),
"DefaultRedeliveryDelay", "0");
if (ic != null) ic.close();
}
// Private -------------------------------------------------------
// Inner classes -------------------------------------------------
}