MessageConsumerFactoryTest.java

package org.darkphoenixs.mq.common;

import org.darkphoenixs.mq.consumer.AbstractConsumer;
import org.darkphoenixs.mq.consumer.Consumer;
import org.darkphoenixs.mq.exception.MQException;
import org.darkphoenixs.mq.message.MessageBeanImpl;
import org.junit.Assert;
import org.junit.Test;

@SuppressWarnings("deprecation")
public class MessageConsumerFactoryTest {

    @Test
    public void test_0() throws Exception {

        MessageConsumerFactory factory = (MessageConsumerFactory) MessageConsumerFactory
                .getInstance();

        factory.init();

        factory.destroy();

        factory = (MessageConsumerFactory) MessageConsumerFactory
                .getInstance();

        factory.init();

        factory.destroy();

        ConsumerTest consumer1 = new ConsumerTest();
        consumer1.setConsumerKey("ProtocolId1");

        ConsumerTest consumer2 = new ConsumerTest();
        consumer2.setConsumerKey("ProtocolId2");

        factory.setConsumers(new Consumer[]{consumer1});
        factory.addConsumer(consumer2);
        factory.init();

        Assert.assertEquals(consumer1, factory.getConsumer("ProtocolId1"));
        Assert.assertEquals(consumer2, factory.getConsumer("ProtocolId2"));
        Assert.assertNull(factory.getConsumer("ProtocolId3"));

        factory.destroy();
    }

    private class ConsumerTest extends AbstractConsumer<MessageBeanImpl> {

        @Override
        protected void doReceive(MessageBeanImpl message) throws MQException {

            System.out.println(message);
        }
    }
}