mirror of https://github.com/apache/activemq.git
[AMQ-8443] Add unit test
This commit is contained in:
parent
9d8ee0d3b7
commit
09458713aa
|
@ -16,6 +16,8 @@
|
|||
*/
|
||||
package org.apache.activemq.usecases;
|
||||
|
||||
import static org.junit.Assert.assertNotEquals;
|
||||
|
||||
import javax.jms.Connection;
|
||||
import javax.jms.Message;
|
||||
import javax.jms.MessageConsumer;
|
||||
|
@ -25,6 +27,9 @@ import javax.jms.Session;
|
|||
|
||||
import org.apache.activemq.ActiveMQConnectionFactory;
|
||||
import org.apache.activemq.JmsMultipleBrokersTestSupport;
|
||||
import org.apache.activemq.broker.TransportConnection;
|
||||
import org.apache.activemq.broker.TransportConnector;
|
||||
import org.apache.activemq.command.ConnectionControl;
|
||||
import org.apache.log4j.Logger;
|
||||
import org.springframework.core.io.ClassPathResource;
|
||||
|
||||
|
@ -87,4 +92,64 @@ public class ClientRebalanceTest extends JmsMultipleBrokersTestSupport {
|
|||
msg = consumer.receive(2000);
|
||||
assertNotNull(msg);
|
||||
}
|
||||
|
||||
public void testReconnect() throws Exception {
|
||||
createBroker(new ClassPathResource("org/apache/activemq/usecases/rebalance-broker1.xml"));
|
||||
createBroker(new ClassPathResource("org/apache/activemq/usecases/rebalance-broker2.xml"));
|
||||
|
||||
startAllBrokers();
|
||||
|
||||
brokers.get("b1").broker.waitUntilStarted();
|
||||
|
||||
LOG.info("Starting connection");
|
||||
|
||||
ActiveMQConnectionFactory factory = new ActiveMQConnectionFactory("failover:(tcp://localhost:61616)?randomize=false");
|
||||
Connection conn = factory.createConnection();
|
||||
conn.start();
|
||||
Session session = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
Queue theQueue = session.createQueue("Test.ClientReconnectTest");
|
||||
MessageProducer producer = session.createProducer(theQueue);
|
||||
MessageConsumer consumer = session.createConsumer(theQueue);
|
||||
Message message = session.createTextMessage("Test message");
|
||||
producer.send(message);
|
||||
Message msg = consumer.receive(2000);
|
||||
assertNotNull(msg);
|
||||
|
||||
TransportConnector transportConnector = brokers.get("b1").broker.getTransportConnectorByName("openwire");
|
||||
assertNotNull(transportConnector);
|
||||
|
||||
TransportConnection startFailoverConnection = findClientFailoverTransportConnection(transportConnector);
|
||||
|
||||
assertNotNull(startFailoverConnection);
|
||||
String startConnectionId = startFailoverConnection.getConnectionId();
|
||||
String startRemoteAddress = startFailoverConnection.getRemoteAddress();
|
||||
ConnectionControl simulateRebalance = new ConnectionControl();
|
||||
simulateRebalance.setReconnectTo("tcp://localhost:61616");
|
||||
startFailoverConnection.dispatchSync(simulateRebalance);
|
||||
|
||||
Thread.sleep(2000l);
|
||||
|
||||
TransportConnection afterFailoverConnection = findClientFailoverTransportConnection(transportConnector);
|
||||
assertNotNull(afterFailoverConnection);
|
||||
assertEquals(startConnectionId, afterFailoverConnection.getConnectionId());
|
||||
assertNotEquals(startRemoteAddress, afterFailoverConnection.getRemoteAddress());
|
||||
// should still be able to send without exception to broker3
|
||||
producer.send(message);
|
||||
msg = consumer.receive(2000);
|
||||
assertNotNull(msg);
|
||||
conn.close();
|
||||
}
|
||||
|
||||
protected TransportConnection findClientFailoverTransportConnection(TransportConnector transportConnector) {
|
||||
TransportConnection failoverConnection = null;
|
||||
for(TransportConnection tmpConnection : transportConnector.getConnections()) {
|
||||
if(tmpConnection.isNetworkConnection()) {
|
||||
continue;
|
||||
}
|
||||
if(tmpConnection.isFaultTolerantConnection()) {
|
||||
failoverConnection = tmpConnection;
|
||||
}
|
||||
}
|
||||
return failoverConnection;
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue