/*
* 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.clustering;
import java.lang.ref.WeakReference;
import javax.jms.Connection;
import javax.jms.ConnectionFactory;
import org.jboss.jms.client.delegate.ClientClusteredConnectionFactoryDelegate;
import org.jboss.logging.Logger;
import org.jboss.test.messaging.tools.ServerManagement;
import org.jboss.test.messaging.tools.container.Command;
import org.jboss.test.messaging.tools.container.Server;
import org.jboss.test.messaging.tools.container.ServiceContainer;
/**
* @author <a href="mailto:clebert.suconic@jboss.org">Clebert Suconic</a>
* @version <tt>$Revision: 3263 $</tt>
* $Id: ClusteredClientCrashTest.java 3263 2007-10-31 01:23:04Z timfox $
*/
public class ClusteredClientCrashTest extends ClusteringTestBase
{
// Constants -----------------------------------------------------
// Static --------------------------------------------------------
private static final Logger log = Logger.getLogger(ClusteredClientCrashTest.class);
// Attributes ----------------------------------------------------
protected Server localServer;
// Constructors --------------------------------------------------
public ClusteredClientCrashTest(String name)
{
super(name);
}
// Public --------------------------------------------------------
public void setUp() throws Exception
{
nodeCount = 2;
super.setUp();
}
public void tearDown() throws Exception
{
super.tearDown();
}
/**
* Test that when a remote jms client crashes, server side resources for connections are
* cleaned-up.
*/
public void testClientCrash() throws Exception
{
ServerManagement.create(2);
Server remoteServer = ServerManagement.getServer(2);
// We need to make sure that any previously downloaded CF should be released
WeakReference ref = new WeakReference(ic[0].lookup("/ClusteredConnectionFactory"));
int count=0;
while (ref.get() != null)
{
System.gc();
Thread.sleep(1000);
if ((count++>10) && ref.get() != null)
{
fail("Thre is a leak on ClusteredConnectionFactory");
}
}
ClientClusteredConnectionFactoryDelegate cfDelegate = (ClientClusteredConnectionFactoryDelegate)cf.getDelegate();
cfDelegate.closeCallback();
ClusterClientCrash command = new ClusterClientCrash(cf);
assertEquals("OK", remoteServer.executeCommand(command));
assertEquals(new Integer(1),ServerManagement.getServer(1).executeCommand(new VerifySizeOfCFClients(cfDelegate.getUniqueName())));
ServerManagement.kill(2);
Thread.sleep((long)(60000));
assertEquals(new Integer(0), ServerManagement.getServer(0).executeCommand(new VerifySizeOfCFClients(cfDelegate.getUniqueName())));
assertEquals(new Integer(0), ServerManagement.getServer(1).executeCommand(new VerifySizeOfCFClients(cfDelegate.getUniqueName())));
}
// Package protected ---------------------------------------------
// Protected -----------------------------------------------------
// Private -------------------------------------------------------
// Inner classes -------------------------------------------------
public static class ClusterClientCrash implements Command
{
ConnectionFactory cf = null;
public ClusterClientCrash(ConnectionFactory cf)
{
this.cf = cf;
}
public Object execute(Server server) throws Exception
{
Connection conn = cf.createConnection();
conn.start();
return "OK";
}
}
public static class VerifySizeOfCFClients implements Command
{
String uniqueName;
public VerifySizeOfCFClients(String uniqueName)
{
this.uniqueName = uniqueName;
}
public Object execute(Server server) throws Exception
{
int size = server.getServerPeer().getConnectionManager().getConnectionFactoryCallback(uniqueName).length;
return new Integer(size);
}
}
}