/*
* 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;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import javax.jms.Destination;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.Session;
import javax.jms.Topic;
import javax.naming.InitialContext;
import javax.sql.DataSource;
import javax.transaction.TransactionManager;
import org.jboss.jms.message.MessageIdGeneratorFactory;
import org.jboss.logging.Logger;
import org.jboss.test.messaging.tools.ServerManagement;
import org.jboss.test.messaging.util.ProxyAssertSupport;
import org.jboss.tm.TransactionManagerService;
/**
* The base case for messaging tests.
*
* @author <a href="mailto:adrian@jboss.org">Adrian Brock</a>
* @author <a href="mailto:ovidiu@jboss.org">Ovidiu Feodorov</a>
* @author <a href="mailto:tim.fox@jboss.org">Tim Fox</a>
* @author <a href="mailto:juha@jboss.org">Juha Lindfors</a>
*
* @version <tt>$Revision: 2495 $</tt>
* $Id: MessagingTestCase.java 2495 2007-02-28 14:31:19Z timfox $
*/
public class MessagingTestCase extends ProxyAssertSupport
{
// Constants -----------------------------------------------------
public final static int MAX_TIMEOUT = 1000 * 10 /* seconds */;
public final static int MIN_TIMEOUT = 1000 * 1 /* seconds */;
// Static --------------------------------------------------------
// Attributes ----------------------------------------------------
protected Logger log = Logger.getLogger(getClass());
// Constructors --------------------------------------------------
public MessagingTestCase(String name)
{
super(name);
}
// Public --------------------------------------------------------
// Package protected ---------------------------------------------
// Protected -----------------------------------------------------
protected void setUp() throws Exception
{
String banner =
"####################################################### Start " +
(isRemote() ? "REMOTE" : "IN-VM") + " test: " + getName();
log.info(banner);
if (isRemote())
{
// log the test start in the remote log, this will make hunting through logs so much easier
ServerManagement.log(ServerManagement.INFO, banner);
}
MessageIdGeneratorFactory.instance.clear();
}
protected void tearDown() throws Exception
{
String banner =
"####################################################### Stop " +
(isRemote() ? "REMOTE" : "IN-VM") + " test: " + getName();
log.info(banner);
if (isRemote())
{
// log the test stop in the remote log, this will make hunting through logs so much easier
ServerManagement.log(ServerManagement.INFO, banner);
}
}
protected void drainDestination(ConnectionFactory cf, Destination dest) throws Exception
{
Connection conn = null;
try
{
conn = cf.createConnection();
Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer cons = sess.createConsumer(dest);
Message m = null;
conn.start();
log.trace("Draining messages from " + dest);
while (true)
{
m = cons.receive(500);
if (m == null) break;
log.trace("Drained message");
}
}
finally
{
if (conn!= null) conn.close();
}
}
protected void drainSub(ConnectionFactory cf, Topic topic, String subName, String clientID) throws Exception
{
Connection conn = null;
try
{
conn = cf.createConnection();
conn.setClientID(clientID);
Session sess = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
MessageConsumer cons = sess.createDurableSubscriber(topic, subName);
Message m = null;
conn.start();
log.trace("Draining messages from " + topic + ":" + subName);
while (true)
{
m = cons.receive(500);
if (m == null) break;
log.trace("Drained message");
}
}
finally
{
if (conn!= null) conn.close();
}
}
protected boolean checkNoBindingData() throws Exception
{
InitialContext ctx = new InitialContext();
TransactionManager mgr = (TransactionManager)ctx.lookup(TransactionManagerService.JNDI_NAME);
DataSource ds = (DataSource)ctx.lookup("java:/DefaultDS");
javax.transaction.Transaction txOld = mgr.suspend();
mgr.begin();
java.sql.Connection conn = null;
PreparedStatement ps = null;
ResultSet rs = null;
try
{
conn = ds.getConnection();
String sql = "SELECT * FROM JBM_POSTOFFICE";
ps = conn.prepareStatement(sql);
rs = ps.executeQuery();
return rs.next();
}
finally
{
if (rs != null) rs.close();
if (ps != null) ps.close();
if (conn != null) conn.close();
mgr.commit();
if (txOld != null)
{
mgr.resume(txOld);
}
}
}
protected boolean checkNoMessageData() throws Exception
{
//Can't do this remotely
if (ServerManagement.isRemote())
{
return false;
}
InitialContext ctx = new InitialContext();
TransactionManager mgr = (TransactionManager)ctx.lookup(TransactionManagerService.JNDI_NAME);
DataSource ds = (DataSource)ctx.lookup("java:/DefaultDS");
javax.transaction.Transaction txOld = mgr.suspend();
mgr.begin();
java.sql.Connection conn = null;
PreparedStatement ps = null;
ResultSet rs = null;
try
{
conn = ds.getConnection();
String sql = "SELECT * FROM JBM_MSG_REF";
ps = conn.prepareStatement(sql);
rs = ps.executeQuery();
boolean exists = rs.next();
if (!exists)
{
rs.close();
ps.close();
ps = conn.prepareStatement("SELECT * FROM JBM_MSG");
rs = ps.executeQuery();
exists = rs.next();
}
return exists;
}
finally
{
if (rs != null) rs.close();
if (ps != null) ps.close();
if (conn != null) conn.close();
mgr.commit();
if (txOld != null)
{
mgr.resume(txOld);
}
}
}
/**
* @return true if this test is ran in "remote" mode, i.e. the server side of the test runs in a
* different VM than this one (that is running the client side)
*/
protected boolean isRemote()
{
return ServerManagement.isRemote();
}
// Private -------------------------------------------------------
// Inner classes -------------------------------------------------
}