Add some methods for stoppng and restarting broker during tests.

This commit is contained in:
Timothy Bish 2013-11-27 14:02:05 -05:00
parent eead6e5110
commit e3fed4b57a
1 changed files with 26 additions and 4 deletions

View File

@ -70,12 +70,14 @@ public class AmqpTestSupport {
autoFailTestSupport.startAutoFailThread(); autoFailTestSupport.startAutoFailThread();
exceptions.clear(); exceptions.clear();
startBroker(); startBroker();
this.numberOfMessages = 2000;
} }
public void startBroker() throws Exception { private void createBroker(boolean deleteAllMessages) throws Exception {
brokerService = new BrokerService(); brokerService = new BrokerService();
brokerService.setPersistent(false); brokerService.setPersistent(false);
brokerService.setAdvisorySupport(false); brokerService.setAdvisorySupport(false);
brokerService.setDeleteAllMessagesOnStartup(deleteAllMessages);
SSLContext ctx = SSLContext.getInstance("TLS"); SSLContext ctx = SSLContext.getInstance("TLS");
ctx.init(new KeyManager[0], new TrustManager[]{new DefaultTrustManager()}, new SecureRandom()); ctx.init(new KeyManager[0], new TrustManager[]{new DefaultTrustManager()}, new SecureRandom());
@ -93,8 +95,6 @@ public class AmqpTestSupport {
brokerService.setSslContext(sslContext); brokerService.setSslContext(sslContext);
addAMQPConnector(); addAMQPConnector();
brokerService.start();
this.numberOfMessages = 2000;
} }
protected void addAMQPConnector() throws Exception { protected void addAMQPConnector() throws Exception {
@ -112,12 +112,34 @@ public class AmqpTestSupport {
LOG.debug("Using amqp+nio+ssl port " + nioPlusSslPort); LOG.debug("Using amqp+nio+ssl port " + nioPlusSslPort);
} }
@After public void startBroker() throws Exception {
if (brokerService != null) {
throw new IllegalStateException("Broker is already created.");
}
createBroker(true);
brokerService.start();
brokerService.waitUntilStarted();
}
public void restartBroker() throws Exception {
stopBroker();
createBroker(false);
brokerService.start();
brokerService.waitUntilStarted();
}
public void stopBroker() throws Exception { public void stopBroker() throws Exception {
if (brokerService != null) { if (brokerService != null) {
brokerService.stop(); brokerService.stop();
brokerService.waitUntilStopped();
brokerService = null; brokerService = null;
} }
}
@After
public void tearDown() throws Exception {
stopBroker();
autoFailTestSupport.stopAutoFailThread(); autoFailTestSupport.stopAutoFailThread();
} }