avs-device-sdk/ACL/test/Transport/MockTransport.h

/**
 * MockTransport.cpp
 *
 * Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License").
 * You may not use this file except in compliance with the License.
 * A copy of the License is located at
 *
 *     http://aws.amazon.com/apache2.0/
 *
 * or in the "license" file accompanying this file. This file 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 ALEXA_CLIENT_SDK_ACL_TEST_TRANSPORT_MOCK_TRANSPORT_H_
#define ALEXA_CLIENT_SDK_ACL_TEST_TRANSPORT_MOCK_TRANSPORT_H_

#include <AVSCommon/AVS/Message.h>
#include <AVSCommon/AVS/MessageRequest.h>

#include "ACL/Transport/TransportInterface.h"

#include <gmock/gmock.h>

#include <memory>

namespace alexaClientSDK {
namespace acl {
namespace transport {

using ::testing::Return;

class MockTransport: public TransportInterface {
public:
    MockTransport(): m_id{sNewId++} {};

    MOCK_METHOD0(connect, bool());
    MOCK_METHOD0(disconnect, void());
    MOCK_METHOD0(isConnected, bool());
    MOCK_METHOD0(isPendingDisconnected, bool());
    MOCK_METHOD1(send, void(std::shared_ptr<avsCommon::avs::MessageRequest>));
    MOCK_METHOD1(onAttachmentReceived, void(std::shared_ptr<avsCommon::avs::Message>));

    const int m_id;

private:
    static int sNewId;
};

int MockTransport::sNewId = 0;

/**
 * Puts the mock transport into a ready to connect state.
 */
void initializeMockTransport(MockTransport* transport) {
    ON_CALL(*transport, connect()).WillByDefault(Return(true));
    ON_CALL(*transport, isConnected()).WillByDefault(Return(false));
}

/**
 * Puts the mock transport into a connected state.
 */
void connectMockTransport(MockTransport* transport) {
    initializeMockTransport(transport);
    ON_CALL(*transport, isConnected()).WillByDefault(Return(true));
}

/**
 * Puts the mock transport into a disconnected state.
 */
void disconnectMockTransport(MockTransport* transport) {
    ON_CALL(*transport, isConnected()).WillByDefault(Return(false));
}

} // namespace transport
} // namespace acl
} // namespace alexaClientSDK

#endif // ALEXA_CLIENT_SDK_ACL_TEST_TRANSPORT_MOCK_TRANSPORT_H_