EmbeddedMQBroker.java

class EmbeddedMQBroker {
    public static final String ACTIVEMQ_URL = "vm://localhost";

    private static Logger logger = LoggerFactory.getLogger(EmbeddedMQBroker.class);

    Connection connection;
    BrokerService broker;

    private EmbeddedMQBroker(final String url) {
        //Keep alive mq connection to stop ActiveMq from shutting down automatically. 
        broker = new BrokerService();

        try {
            broker.addConnector(url);
            broker.setPersistent(false);
            broker.start();
            broker.waitUntilStarted();
            connection = new ActiveMQConnectionFactory(ACTIVEMQ_URL).createConnection();

        } catch (Exception e) {
            throw new RuntimeException("Unable to setup broker", e);
        }
    }

    public static EmbeddedMQBroker start() {
        return start(ACTIVEMQ_URL);
    }

    public static EmbeddedMQBroker start(final String url) {
        logger.info(">>>>>>>>>> EmbeddedMQBroker starting <<<<<<<<<<");
        return new EmbeddedMQBroker(url);
    }


    public void stop() {
        if (connection != null) {
            try {
                connection.close();
            } catch (JMSException e) {
                throw new RuntimeException("Unable to close connection", e);
            }
        }

        try {
            broker.stop();
        } catch (Exception e) {
            throw new RuntimeException("Unable to stop broker", e);
        }

        broker.waitUntilStopped();

        broker = null;
        logger.info(">>>>>>>>>> EmbeddedMQBroker stopped <<<<<<<<<<");
    }
}