/*
 * 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.
 */

#ifndef _INTEGRATION_CONNECTOR_OPENWIRE_OPENWIREEXPIRATIONTEST_H_
#define _INTEGRATION_CONNECTOR_OPENWIRE_OPENWIREEXPIRATIONTEST_H_

#include <cppunit/TestFixture.h>
#include <cppunit/extensions/HelperMacros.h>

#include <cms/MessageListener.h>
#include <cms/ExceptionListener.h>
#include <cms/ConnectionFactory.h>
#include <cms/Connection.h>
#include <cms/Session.h>
#include <cms/MessageProducer.h>

#include <activemq/concurrent/Runnable.h>

namespace integration{
namespace connector{
namespace openwire{

    class OpenwireExpirationTest : public CppUnit::TestFixture   
    {
        CPPUNIT_TEST_SUITE( OpenwireExpirationTest );
        CPPUNIT_TEST( testExpired );
        CPPUNIT_TEST( testNotExpired );
        CPPUNIT_TEST_SUITE_END();

    public:
    
        static std::string messageTag;
        
    public:
    
        OpenwireExpirationTest(){}
        virtual ~OpenwireExpirationTest(){}
        
        virtual void testExpired();
        virtual void testNotExpired();
        
    public:
    
        class Producer : public activemq::concurrent::Runnable {
        private:
        
            cms::Connection* connection;
            cms::Session* session;
            cms::Topic* destination;
            cms::MessageProducer* producer;
            int numMessages;
            long long timeToLive;
            bool disableTimeStamps;
            std::string topic;
        
        public:
        
            Producer( std::string topic, int numMessages, long long timeToLive );
        
            virtual ~Producer();
        
            virtual bool getDisableTimeStamps() const;
        
            virtual void setDisableTimeStamps( bool value );
        
            virtual void run();
        
        private:
        
            void cleanup();
        };

        class Consumer : public cms::MessageListener, public activemq::concurrent::Runnable {
        
        private:
        
            cms::Connection* connection;
            cms::Session* session;
            cms::Topic* destination;
            cms::MessageConsumer* consumer;
            long waitMillis;
            int numReceived;
            std::string topic;
        
        public:
        
            Consumer( std::string topic, long waitMillis );
            
            virtual ~Consumer();
        
            virtual int getNumReceived() const;
            
            virtual void run();
        
            virtual void onMessage( const cms::Message* message );
        
        private:
        
            void cleanup();
        };
    };

}}}

#endif /*_INTEGRATION_CONNECTOR_OPENWIRE_OPENWIREEXPIRATIONTEST_H_*/