/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF 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 org.apache.activemq.spring;

import javax.jms.Message;
import java.util.Iterator;
import java.util.List;

import junit.framework.TestCase;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.support.AbstractApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;

/**
 *
 */
public class SpringTestSupport extends TestCase {

   private static final Logger LOG = LoggerFactory.getLogger(SpringTest.class);
   protected AbstractApplicationContext context;
   protected SpringConsumer consumer;
   protected SpringProducer producer;

   /**
    * assert method that is used by all the test method to send and receive messages
    * based on each spring configuration.
    *
    * @param config
    * @throws Exception
    */
   protected void assertSenderConfig(String config) throws Exception {
      Thread.currentThread().setContextClassLoader(SpringTest.class.getClassLoader());
      context = new ClassPathXmlApplicationContext(config);

      consumer = (SpringConsumer) context.getBean("consumer");
      assertTrue("Found a valid consumer", consumer != null);

      consumer.start();

      // Wait a little to drain any left over messages.
      Thread.sleep(1000);
      consumer.flushMessages();

      producer = (SpringProducer) context.getBean("producer");
      assertTrue("Found a valid producer", producer != null);

      producer.start();

      // lets sleep a little to give the JMS time to dispatch stuff
      consumer.waitForMessagesToArrive(producer.getMessageCount());

      // now lets check that the consumer has received some messages
      List<Message> messages = consumer.flushMessages();
      LOG.info("Consumer has received messages....");
      for (Iterator<Message> iter = messages.iterator(); iter.hasNext(); ) {
         Object message = iter.next();
         LOG.info("Received: " + message);
      }

      assertEquals("Message count", producer.getMessageCount(), messages.size());
   }

   /**
    * Clean up method.
    *
    * @throws Exception
    */
   @Override
   protected void tearDown() throws Exception {
      if (consumer != null) {
         consumer.stop();
      }
      if (producer != null) {
         producer.stop();
      }

      if (context != null) {
         context.destroy();
      }
   }
}
