JmsQueueIntegrationTest.java

/*
 * Copyright LGPL3
 * YES Technology Association
 * http://yestech.org
 *
 * http://www.opensource.org/licenses/lgpl-3.0.html
 */

/*
 *
 * Original Author:  Artie Copeland
 * Last Modified Date: $DateTime: $
 */
package org.yestech.notify.service;

import org.jmock.Mockery;
import org.jmock.Expectations;
import org.junit.Test;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertEquals;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.yestech.notify.objectmodel.INotificationJob;
import org.yestech.notify.objectmodel.NotificationJob;
import org.yestech.notify.objectmodel.DefaultNotification;
import org.apache.activemq.broker.BrokerService;

import java.util.UUID;

/**
 * @author $Author: $
 * @version $Revision: $
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath:/org/yestech/notify/service/JmsQueueIntegrationTest-context.xml"})
public class JmsQueueIntegrationTest {
    private Mockery context = new Mockery();

    @Autowired
    private JmsQueueNotificationConsumer consumer;

    @Autowired
    private JmsQueueNotificationProducer producer;

    @Test
    public void setSending100Messages() throws InterruptedException {
        TestProcessor processor = new TestProcessor();
        consumer.setProcessor(processor);

        final int total = 100;
        for (int i = 0; i < total; i++) {
            INotificationJob job = new NotificationJob(UUID.randomUUID());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            job.addNotification(new DefaultNotification());
            producer.send(job);
        }
         Thread.sleep(5000); //sleep for 5 secs... to make sure broker get fully initialized and processes all the messages
        assertTrue(processor.isCalled());
        assertEquals(total, processor.getCount());

    }

    private class TestProcessor implements INotificationProcessor {
        boolean called;
        int count;

        public int getCount() {
            return count;
        }

        public void setCount(int count) {
            this.count = count;
        }

        private TestProcessor() {
        }

        public boolean isCalled() {
            return called;
        }

        public void setCalled(boolean called) {
            this.called = called;
        }

        public void process(INotificationJob notificationJob) {
            called = true;
            ++count;
        }
    }
}