01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package org.jboss.test.jbossmq.test;
23:
24: import javax.jms.Connection;
25: import javax.jms.ConnectionFactory;
26: import javax.jms.ExceptionListener;
27: import javax.jms.JMSException;
28: import javax.management.MBeanServerConnection;
29: import javax.management.ObjectName;
30:
31: import org.jboss.test.JBossTestCase;
32:
33: import EDU.oswego.cs.dl.util.concurrent.SynchronizedBoolean;
34:
35: /**
36: * A test to make sure exception listeners are fired when a service is restarted
37: *
38: * @author <a href="mailto:adrian@jboss.org>Adrian Brock</a>
39: * @version <tt>$Revision: 57211 $</tt>
40: */
41: public abstract class AbstractRestartDisconnectClientsTest extends
42: JBossTestCase implements ExceptionListener {
43: static String FACTORY = "ConnectionFactory";
44:
45: private SynchronizedBoolean exceptionListenerFired = new SynchronizedBoolean(
46: false);
47:
48: public AbstractRestartDisconnectClientsTest(String name)
49: throws Exception {
50: super (name);
51: }
52:
53: protected abstract ObjectName getRestartObjectName();
54:
55: public synchronized void onException(JMSException exception) {
56: exceptionListenerFired.set(true);
57: getLog().debug("Received notification of error: ", exception);
58: notifyAll();
59: }
60:
61: public void testDisconnect() throws Exception {
62: ConnectionFactory factory = (ConnectionFactory) getInitialContext()
63: .lookup(FACTORY);
64: Connection connection = factory.createConnection();
65: try {
66: connection.setExceptionListener(this );
67: connection.start();
68: MBeanServerConnection server = getServer();
69: ObjectName objectName = getRestartObjectName();
70: server.invoke(objectName, "stop", null, null);
71: try {
72: // Shouldn't really have to wait for the ping timeout, but just in case
73: synchronized (this ) {
74: if (exceptionListenerFired.get() == false)
75: wait(20 * 1000);
76: }
77: assertTrue(
78: "Exception Listener should have been invoked",
79: exceptionListenerFired.get());
80: } finally {
81: server.invoke(objectName, "start", null, null);
82: }
83: } finally {
84: try {
85: connection.close();
86: } catch (JMSException ignored) {
87: }
88: }
89: }
90: }
|