/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.activemq.bugs;
import java.io.IOException;
import javax.jms.MessageProducer;
import javax.jms.Session;
import org.apache.activemq.ActiveMQConnection;
import org.apache.activemq.EmbeddedBrokerTestSupport;
import org.apache.activemq.broker.BrokerService;
import org.apache.activemq.command.ShutdownInfo;
import org.apache.activemq.transport.Transport;
import org.apache.activemq.transport.TransportFactory;
import org.apache.activemq.transport.TransportListener;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
public class VMTransportClosureTest extends EmbeddedBrokerTestSupport {
private static final Log LOG = LogFactory
.getLog(VMTransportClosureTest.class);
private static final long MAX_TEST_TIME_MILLIS = 300000; // 5min
private static final int NUM_ATTEMPTS = 100000;
public void setUp() throws Exception {
setAutoFail(true);
setMaxTestTime(MAX_TEST_TIME_MILLIS);
super.setUp();
}
/**
* EmbeddedBrokerTestSupport.createBroker() binds the broker to a VM
* transport address, which results in a call to
* VMTransportFactory.doBind(location):
* <p>
* <code>
* public TransportServer doBind(URI location) throws IOException {
* return bind(location, false);
*}
*</code>
* </p>
* As a result, VMTransportServer.disposeOnDisconnect is <code>false</code>.
* To expose the bug, we need to have VMTransportServer.disposeOnDisconnect
* <code>true</code>, which is the case when the VMTransportServer is not
* already bound when the first connection is made.
*/
@Override
protected BrokerService createBroker() throws Exception {
BrokerService answer = new BrokerService();
answer.setPersistent(isPersistent());
// answer.addConnector(bindAddress);
return answer;
}
/**
* This test demonstrates how the "disposeOnDisonnect" feature of
* VMTransportServer can incorrectly close all VM connections to the local
* broker.
*/
public void testPrematureClosure() throws Exception {
// Open a persistent connection to the local broker. The persistent
// connection is maintained through the test and should prevent the
// VMTransportServer from stopping itself when the local transport is
// closed.
ActiveMQConnection persistentConn = (ActiveMQConnection) createConnection();
persistentConn.start();
Session session = persistentConn.createSession(true,
Session.SESSION_TRANSACTED);
MessageProducer producer = session.createProducer(destination);
for (int i = 0; i < NUM_ATTEMPTS; i++) {
LOG.info("Attempt: " + i);
// Open and close a local transport connection. As is done by by
// most users of the transport, ensure that the transport is stopped
// when closed by the peer (via ShutdownInfo). Closing the local
// transport should not affect the persistent connection.
final Transport localTransport = TransportFactory.connect(broker
.getVmConnectorURI());
localTransport.setTransportListener(new TransportListener() {
public void onCommand(Object command) {
if (command instanceof ShutdownInfo) {
try {
localTransport.stop();
} catch (Exception ex) {
throw new RuntimeException(ex);
}
}
}
public void onException(IOException error) {
// ignore
}
public void transportInterupted() {
// ignore
}
public void transportResumed() {
// ignore
}
});
localTransport.start();
localTransport.stop();
// Ensure that the persistent connection is still usable.
producer.send(session.createMessage());
session.rollback();
}
persistentConn.close();
}
}