MockTransport.h

Go to the documentation of this file.

/*
 * 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 _ACTIVEMQ_TANSPORT_MOCK_MOCKTRANSPORT_H_
#define _ACTIVEMQ_TANSPORT_MOCK_MOCKTRANSPORT_H_
 
#include <activemq/util/Config.h>
#include <activemq/exceptions/ActiveMQException.h>
#include <activemq/transport/Transport.h>
#include <activemq/transport/TransportListener.h>
#include <activemq/transport/DefaultTransportListener.h>
#include <activemq/transport/mock/ResponseBuilder.h>
#include <activemq/transport/mock/InternalCommandListener.h>
#include <activemq/wireformat/WireFormat.h>
 
#include <decaf/lang/Thread.h>
#include <decaf/lang/Pointer.h>
#include <decaf/util/concurrent/Concurrent.h>
#include <decaf/util/concurrent/atomic/AtomicInteger.h>
#include <decaf/util/concurrent/CountDownLatch.h>
 
#include <cms/Message.h>
 
#include <map>
#include <set>
 
namespace activemq{
namespace transport{
namespace mock{
 
    using decaf::lang::Pointer;
    using activemq::commands::Command;
    using activemq::commands::Response;
 
    class AMQCPP_API MockTransport: public Transport {
    private:
 
        Pointer<ResponseBuilder> responseBuilder;
        Pointer<wireformat::WireFormat> wireFormat;
        TransportListener* outgoingListener;
        TransportListener* listener;
        decaf::util::concurrent::atomic::AtomicInteger nextCommandId;
        InternalCommandListener internalListener;
        static MockTransport* instance;
 
        std::string name;
 
        bool failOnSendMessage;
        int numSentMessageBeforeFail;
        int numSentMessages;
        bool failOnReceiveMessage;
        int numReceivedMessageBeforeFail;
        int numReceivedMessages;
        bool failOnKeepAliveSends;
        int numSentKeepAlivesBeforeFail;
        int numSentKeepAlives;
 
        bool failOnStart;
        bool failOnStop;
        bool failOnClose;
 
    private:
 
        MockTransport(const MockTransport&);
        MockTransport operator=(const MockTransport&);
 
    public:
 
        MockTransport(const Pointer<wireformat::WireFormat> wireFormat, const Pointer<ResponseBuilder> responseBuilder);
 
        virtual ~MockTransport() {}
 
        static MockTransport* getInstance() {
            return instance;
        }
 
        virtual void fireCommand(const Pointer<Command> command) {
            if (listener != NULL) {
                listener->onCommand(command);
            }
        }
 
        virtual void fireException(const exceptions::ActiveMQException& ex) {
            if (listener != NULL) {
                listener->onException(ex);
            }
        }
 
        void setResponseBuilder(const Pointer<ResponseBuilder> responseBuilder) {
            this->responseBuilder = responseBuilder;
        }
 
        virtual void setOutgoingListener(TransportListener* listener) {
            outgoingListener = listener;
        }
 
        Pointer<wireformat::WireFormat> getWireFormat() const {
            return this->wireFormat;
        }
 
    public: // Transport Methods
 
        virtual void oneway(const Pointer<Command> command);
 
        virtual Pointer<FutureResponse> asyncRequest(const Pointer<Command> command,
                                                     const Pointer<ResponseCallback> responseCallback);
 
        virtual Pointer<Response> request(const Pointer<Command> command);
 
        virtual Pointer<Response> request(const Pointer<Command> command, unsigned int timeout);
 
        virtual void setWireFormat(const Pointer<wireformat::WireFormat> wireFormat AMQCPP_UNUSED) {}
 
        virtual void setTransportListener(TransportListener* listener) {
            this->listener = listener;
        }
 
        virtual TransportListener* getTransportListener() const {
            return this->listener;
        }
 
        virtual void start();
 
        virtual void stop();
 
        virtual void close();
 
        virtual Transport* narrow(const std::type_info& typeId) {
            if (typeid( *this ) == typeId) {
                return this;
            }
 
            return NULL;
        }
 
        virtual bool isFaultTolerant() const {
            return false;
        }
 
        virtual bool isConnected() const {
            return true;
        }
 
        virtual bool isClosed() const {
            return false;
        }
 
        virtual std::string getRemoteAddress() const {
            return "";
        }
 
        virtual void reconnect(const decaf::net::URI& uri AMQCPP_UNUSED) {}
 
    public: // Property Getters and Setters
 
        std::string getName() const {
            return this->name;
        }
 
        void setName(const std::string& name) {
            this->name = name;
        }
 
        bool isFailOnSendMessage() const {
            return this->failOnSendMessage;
        }
 
        void setFailOnSendMessage(bool value) {
            this->failOnSendMessage = value;
        }
 
        int getNumSentMessageBeforeFail() const {
            return this->numSentMessageBeforeFail;
        }
 
        void setNumSentMessageBeforeFail(int value) {
            this->numSentMessageBeforeFail = value;
        }
 
        int getNumSentMessages() const {
            return this->numSentMessages;
        }
 
        void setNumSentMessages(int value) {
            this->numSentMessages = value;
        }
 
        bool isFailOnReceiveMessage() const {
            return this->failOnReceiveMessage;
        }
 
        void setFailOnReceiveMessage(bool value) {
            this->failOnReceiveMessage = value;
        }
 
        int getNumReceivedMessageBeforeFail() const {
            return this->numReceivedMessageBeforeFail;
        }
 
        void setNumReceivedMessageBeforeFail(int value) {
            this->numReceivedMessageBeforeFail = value;
        }
 
        int getNumReceivedMessages() const {
            return this->numReceivedMessages;
        }
 
        void setNumReceivedMessages(int value) {
            this->numReceivedMessages = value;
        }
 
        bool isFailOnKeepAliveSends() const {
            return this->failOnKeepAliveSends;
        }
 
        void setFailOnKeepAliveSends(bool value) {
            this->failOnKeepAliveSends = value;
        }
 
        int getNumSentKeepAlivesBeforeFail() const {
            return this->numSentKeepAlivesBeforeFail;
        }
 
        void setNumSentKeepAlivesBeforeFail(int value) {
            this->numSentKeepAlivesBeforeFail = value;
        }
 
        int getNumSentKeepAlives() const {
            return this->numSentKeepAlives;
        }
 
        void setNumSentKeepAlives(int value) {
            this->numSentKeepAlives = value;
        }
 
        bool isFailOnStart() const {
            return this->failOnReceiveMessage;
        }
 
        void setFailOnStart(bool value) {
            this->failOnReceiveMessage = value;
        }
 
        bool isFailOnStop() const {
            return this->failOnStop;
        }
 
        void setFailOnStop(bool value) {
            this->failOnStop = value;
        }
 
        bool isFailOnClose() const {
            return this->failOnClose;
        }
 
        void setFailOnClose(bool value) {
            this->failOnClose = value;
        }
 
        virtual bool isReconnectSupported() const {
            return false;
        }
 
        virtual bool isUpdateURIsSupported() const {
            return false;
        }
 
        virtual void updateURIs(bool rebalance AMQCPP_UNUSED, const decaf::util::List<decaf::net::URI>& uris AMQCPP_UNUSED) {
            throw decaf::io::IOException();
        }
 
    };
 
}}}
 
#endif /*_ACTIVEMQ_TANSPORT_MOCK_MOCKTRANSPORT_H_*/