io.fabric8.msg.gateway.TestGateway.java Source code

Introduction

Here is the source code for io.fabric8.msg.gateway.TestGateway.java

Source

/*
 *
 *  * Copyright 2005-2016 Red Hat, Inc.
 *  * Red Hat 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 io.fabric8.msg.gateway;

import io.fabric8.msg.gateway.brokers.impl.TestBrokerControlImpl;
import org.apache.activemq.ActiveMQConnectionFactory;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Ignore;
import org.springframework.context.ApplicationContextInitializer;
import org.springframework.context.ConfigurableApplicationContext;

import javax.jms.Connection;
import javax.jms.Destination;
import javax.jms.Message;
import javax.jms.MessageConsumer;
import javax.jms.MessageProducer;
import javax.jms.Session;

public class TestGateway implements ApplicationContextInitializer {
    private MessageGateway messageGateway;

    @Before
    public void doStart() throws Exception {
        messageGateway = new MessageGateway();
        messageGateway.setBrokerControl(new TestBrokerControlImpl());
        messageGateway.start();
    }

    @After
    public void doStop() throws Exception {
        if (messageGateway != null) {
            messageGateway.stop();
        }
    }

    /**
     * TODO - lets figure out a way of mocking kubernetes, running a ZK ensemble, an Artemis broker and testing it!
     * @throws Exception
     */
    @Ignore
    public void simpleTest() throws Exception {
        int numberOfMessages = 10;
        String destinationName = "jms.queue.test.foo";
        String brokerURL = "tcp://0.0.0.0:61616";
        ActiveMQConnectionFactory factory = new ActiveMQConnectionFactory(brokerURL);

        Connection consumerConnection = factory.createConnection();
        consumerConnection.start();
        Session consumerSession = consumerConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
        Destination consumerDestination = consumerSession.createQueue(destinationName);
        MessageConsumer messageConsumer = consumerSession.createConsumer(consumerDestination);

        Connection producerConnection = factory.createConnection();
        producerConnection.start();
        Session producerSession = producerConnection.createSession(false, Session.AUTO_ACKNOWLEDGE);
        Destination destination = producerSession.createQueue(destinationName);
        MessageProducer producer = producerSession.createProducer(destination);

        for (int i = 0; i < numberOfMessages; i++) {
            Message message = producerSession.createTextMessage("test message: " + i);
            producer.send(message);
            //System.err.println("Sent message " + message);

        }

        Message message;

        for (int i = 0; i < numberOfMessages; i++) {
            message = messageConsumer.receive(5000);
            Assert.assertNotNull(message);
            //System.err.println("Got Message " + message);

        }
        messageConsumer.close();

    }

    @Override
    public void initialize(ConfigurableApplicationContext configurableApplicationContext) {

    }
}