/**
    * Copyright 2005-2006 the original author or authors.
    *
    * Licensed under the Gnu General Pubic 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.opensource.org/licenses/gpl-license.php
    *
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
   * See the Gnu General Public License for more details.
   */
  package org.figure8.join.core.messaging;
  
  import org.figure8.join.util.SpringTestCase;
  
  import javax.jms.TextMessage;
  import javax.jms.ObjectMessage;
  
  import java.util.List;
  import java.util.Properties;
  /**
   * JUnit test case for testing simple JMS message production and consumption.
   * @author <a href="mailto:laurent.broudoux@free.fr">Laurent Broudoux</a>
   * @version $Revision: 1.2 $
   */
  public class ProducerConsumerTest extends SpringTestCase{
  
     // Static -------------------------------------------------------------------
  
     /** Spring configuration files */
     private String[] configLocations = new String[]{"classpath:/org/figure8/join/core/messaging/spring.xml"};
  
  
     // Attributes ---------------------------------------------------------------
  
     /** JMS messages producer */
     protected JMSProducerBean producer = null;
     /** JMS messages consumer (echoes messages) */
     protected EchoJMSConsumerBean consumer = null;
  
  
     // Override of SpringTestCase -----------------------------------------------
  
     /** Retrieve producer and consumer after having initialized context. */
     public void setUp(){
        super.setUp();
        // Get a simple producer.
        producer = (JMSProducerBean)context.getBean("simpleProducer");
        // Get an echo consumer.
        consumer = (EchoJMSConsumerBean)context.getBean("echoConsumer");
     }
  
  
     // Public -------------------------------------------------------------------
  
     /** Test the production and the consumption of text messages. */
     public void testProductionAndConsumptionOfTextMessages(){
        // Assert beans presence.
        assertNotNull("Found a valid producer", producer);
        assertNotNull("Found a valid consumer", consumer);
        consumer.flushMessages();
        // Sending messages.
        for (int i=0; i<10; i++){
           String text = "Hello World ! (" + i + ")";
           producer.sendTextMessage(text);
        }
        // Checking received messages.
        consumer.waitForMessagesToArrive(10);
        List messages = consumer.flushMessages();
        assertEquals("Message count is correct", 10, messages.size());
        // Checkin messages content.
        for (int i=0; i<messages.size(); i++){
           try{
              TextMessage msg = (TextMessage)messages.get(i);
              assertEquals("Message content is correct", "Hello World ! (" + i + ")", msg.getText());
           }
           catch (Exception e){
              fail("Exception while checking message content");
           }
        }
     }
  
     /** Test the production and the consumption of text messages with properties. */
     public void testProductionAndConsumptionOfTextMessagesWithProperties(){
        // Assert beans presence.
        assertNotNull("Found a valid producer", producer);
        assertNotNull("Found a valid consumer", consumer);
        consumer.flushMessages();
        // Sending messages with properties.
        Properties props = new Properties();
        for (int i=0; i<10; i++){
           String text = "Hello World ! (" + i + ")";
           props.setProperty("counter", String.valueOf(i));
           producer.sendTextMessage(text, props);
        }
        // Checking received messages.
       consumer.waitForMessagesToArrive(10);
       List messages = consumer.flushMessages();
       assertEquals("Message count is correct", 10, messages.size());
       // Checkin messages content.
       for (int i=0; i<messages.size(); i++){
          try{
             TextMessage msg = (TextMessage)messages.get(i);
             assertEquals("Message content is correct", "Hello World ! (" + i + ")", msg.getText());
             assertEquals("Message property is correct", String.valueOf(i), msg.getStringProperty("counter"));
          }
          catch (Exception e){
             fail("Exception while checking message content");
          }
       }
    }
 
    /** Test the production and consumption of object messages */
    public void testProductionAndConsumptionOfObjectMessages(){
       // Assert beans presence.
       assertNotNull("Found a valid producer", producer);
       assertNotNull("Found a valid consumer", consumer);
       consumer.flushMessages();
       // Sending messages.
       for (int i=0; i<10; i++){
          String text = "Hello World ! (" + i + ")";
          producer.sendObjectMessage(text);
       }
       // Checking received messages.
       consumer.waitForMessagesToArrive(10);
       List messages = consumer.flushMessages();
       assertEquals("Message count is correct", 10, messages.size());
       // Checkin messages content.
       for (int i=0; i<messages.size(); i++){
          try{
             ObjectMessage msg = (ObjectMessage)messages.get(i);
             assertEquals("Message content is correct", "Hello World ! (" + i + ")", msg.getObject());
          }
          catch (Exception e){
             fail("Exception while checking message content");
          }
       }
    }
 
    /** Test the production and the consumption of object messages with properties. */
    public void testProductionAndConsumptionOfObjectMessagesWithProperties(){
       // Assert beans presence.
       assertNotNull("Found a valid producer", producer);
       assertNotNull("Found a valid consumer", consumer);
       consumer.flushMessages();
       // Sending messages with properties.
       Properties props = new Properties();
       for (int i=0; i<10; i++){
          String text = "Hello World ! (" + i + ")";
          props.setProperty("counter", String.valueOf(i));
          producer.sendObjectMessage(text, props);
       }
       // Checking received messages.
       consumer.waitForMessagesToArrive(10);
       List messages = consumer.flushMessages();
       assertEquals("Message count is correct", 10, messages.size());
       // Checkin messages content.
       for (int i=0; i<messages.size(); i++){
          try{
             ObjectMessage msg = (ObjectMessage)messages.get(i);
             assertEquals("Message content is correct", "Hello World ! (" + i + ")", msg.getObject());
             assertEquals("Message property is correct", String.valueOf(i), msg.getStringProperty("counter"));
          }
          catch (Exception e){
             fail("Exception while checking message content");
          }
       }
    }
 
 
    // Implementation of SpringTestCase -----------------------------------------
 
    /** @return configLocations inner field */
    public String[] getConfigLocations(){
       return configLocations;
    }
 }