/*
*  This file is part of aasdk library project.
*  Copyright (C) 2018 f1x.studio (Michal Szwaj)
*
*  aasdk is free software: you can redistribute it and/or modify
*  it under the terms of the GNU General Public License as published by
*  the Free Software Foundation; either version 3 of the License, or
*  (at your option) any later version.

*  aasdk 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.
*
*  You should have received a copy of the GNU General Public License
*  along with aasdk. If not, see <http://www.gnu.org/licenses/>.
*/

#include <boost/test/unit_test.hpp>
#include <aasdk/Messenger/UT/MessageInStream.mock.hpp>
#include <aasdk/Messenger/UT/MessageOutStream.mock.hpp>
#include <aasdk/Messenger/UT/ReceivePromiseHandler.mock.hpp>
#include <aasdk/Messenger/UT/SendPromiseHandler.mock.hpp>
#include <aasdk/Messenger/Messenger.hpp>


namespace aasdk
{
namespace messenger
{
namespace ut
{

using ::testing::_;
using ::testing::SaveArg;
using ::testing::Return;

class MessengerUnitTest
{
protected:
    MessengerUnitTest()
        : messageInStream_(&messageInStreamMock_, [](auto*) {})
        , messageOutStream_(&messageOutStreamMock_, [](auto*) {})
        , receivePromise_(ReceivePromise::defer(ioService_))
        , sendPromise_(SendPromise::defer(ioService_))
    {
        receivePromise_->then(std::bind(&ReceivePromiseHandlerMock::onResolve, &receivePromiseHandlerMock_, std::placeholders::_1),
                             std::bind(&ReceivePromiseHandlerMock::onReject, &receivePromiseHandlerMock_, std::placeholders::_1));

        sendPromise_->then(std::bind(&SendPromiseHandlerMock::onResolve, &sendPromiseHandlerMock_),
                          std::bind(&SendPromiseHandlerMock::onReject, &sendPromiseHandlerMock_, std::placeholders::_1));
    }

    boost::asio::io_service ioService_;
    MessageInStreamMock messageInStreamMock_;
    IMessageInStream::Pointer messageInStream_;
    MessageOutStreamMock messageOutStreamMock_;
    IMessageOutStream::Pointer messageOutStream_;
    ReceivePromiseHandlerMock receivePromiseHandlerMock_;
    ReceivePromise::Pointer receivePromise_;
    SendPromiseHandlerMock sendPromiseHandlerMock_;
    SendPromise::Pointer sendPromise_;
};

BOOST_FIXTURE_TEST_CASE(Messenger_Receive, MessengerUnitTest)
{
    Messenger::Pointer themessenger(std::make_shared<Messenger>(ioService_, messageInStream_, messageOutStream_));
    themessenger->enqueueReceive(ChannelId::MEDIA_AUDIO, std::move(receivePromise_));

    ReceivePromise::Pointer inStreamReceivePromise;
    EXPECT_CALL(messageInStreamMock_, startReceive(_)).WillOnce(SaveArg<0>(&inStreamReceivePromise));

    ioService_.run();
    ioService_.reset();

    Message::Pointer message(std::make_shared<Message>(ChannelId::MEDIA_AUDIO, EncryptionType::ENCRYPTED, MessageType::SPECIFIC));
    inStreamReceivePromise->resolve(message);

    EXPECT_CALL(receivePromiseHandlerMock_, onReject(_)).Times(0);
    EXPECT_CALL(receivePromiseHandlerMock_, onResolve(message));
    ioService_.run();
}

BOOST_FIXTURE_TEST_CASE(Messenger_DirectReceive, MessengerUnitTest)
{
    Messenger::Pointer themessenger(std::make_shared<Messenger>(ioService_, messageInStream_, messageOutStream_));
    themessenger->enqueueReceive(ChannelId::MEDIA_AUDIO, std::move(receivePromise_));

    ReceivePromise::Pointer inStreamReceivePromise;
    EXPECT_CALL(messageInStreamMock_, startReceive(_)).WillRepeatedly(SaveArg<0>(&inStreamReceivePromise));

    ioService_.run();
    ioService_.reset();

    Message::Pointer inputChannelMessage(std::make_shared<Message>(ChannelId::INPUT, EncryptionType::ENCRYPTED, MessageType::SPECIFIC));
    inStreamReceivePromise->resolve(inputChannelMessage);

    ioService_.run();
    ioService_.reset();

    auto secondReceivePromise = ReceivePromise::defer(ioService_);
    secondReceivePromise->then(std::bind(&ReceivePromiseHandlerMock::onResolve, &receivePromiseHandlerMock_, std::placeholders::_1),
                              std::bind(&ReceivePromiseHandlerMock::onReject, &receivePromiseHandlerMock_, std::placeholders::_1));
    themessenger->enqueueReceive(ChannelId::INPUT, std::move(secondReceivePromise));

    EXPECT_CALL(receivePromiseHandlerMock_, onReject(_)).Times(0);
    EXPECT_CALL(receivePromiseHandlerMock_, onResolve(inputChannelMessage));

    ioService_.run();
    ioService_.reset();

    Message::Pointer audioChannelMessage(std::make_shared<Message>(ChannelId::MEDIA_AUDIO, EncryptionType::ENCRYPTED, MessageType::SPECIFIC));
    inStreamReceivePromise->resolve(audioChannelMessage);

    EXPECT_CALL(receivePromiseHandlerMock_, onReject(_)).Times(0);
    EXPECT_CALL(receivePromiseHandlerMock_, onResolve(audioChannelMessage));
    ioService_.run();
}

BOOST_FIXTURE_TEST_CASE(Messenger_OnlyOneReceiveAtATime, MessengerUnitTest)
{
    Messenger::Pointer themessenger(std::make_shared<Messenger>(ioService_, messageInStream_, messageOutStream_));
    themessenger->enqueueReceive(ChannelId::MEDIA_AUDIO, std::move(receivePromise_));

    ReceivePromise::Pointer inStreamReceivePromise;
    EXPECT_CALL(messageInStreamMock_, startReceive(_)).WillOnce(SaveArg<0>(&inStreamReceivePromise));

    ioService_.run();
    ioService_.reset();

    auto secondReceivePromise = ReceivePromise::defer(ioService_);
    secondReceivePromise->then(std::bind(&ReceivePromiseHandlerMock::onResolve, &receivePromiseHandlerMock_, std::placeholders::_1),
                              std::bind(&ReceivePromiseHandlerMock::onReject, &receivePromiseHandlerMock_, std::placeholders::_1));
    themessenger->enqueueReceive(ChannelId::INPUT, std::move(secondReceivePromise));

    ioService_.run();
    ioService_.reset();

    const error::Error e(error::ErrorCode::USB_TRANSFER, 41);
    inStreamReceivePromise->reject(e);

    EXPECT_CALL(receivePromiseHandlerMock_, onReject(e)).Times(2);
    EXPECT_CALL(receivePromiseHandlerMock_, onResolve(_)).Times(0);

    ioService_.run();
}

BOOST_FIXTURE_TEST_CASE(Messenger_Send, MessengerUnitTest)
{
    Messenger::Pointer themessenger(std::make_shared<Messenger>(ioService_, messageInStream_, messageOutStream_));

    Message::Pointer message(std::make_shared<Message>(ChannelId::MEDIA_AUDIO, EncryptionType::ENCRYPTED, MessageType::SPECIFIC));
    themessenger->enqueueSend(message, std::move(sendPromise_));

    SendPromise::Pointer outStreamSendPromise;
    EXPECT_CALL(messageOutStreamMock_, stream(message, _)).WillOnce(SaveArg<1>(&outStreamSendPromise));

    ioService_.run();
    ioService_.reset();

    EXPECT_CALL(sendPromiseHandlerMock_, onReject(_)).Times(0);
    EXPECT_CALL(sendPromiseHandlerMock_, onResolve());

    outStreamSendPromise->resolve();
    ioService_.run();
}

BOOST_FIXTURE_TEST_CASE(Messenger_OnlyOneSendAtATime, MessengerUnitTest)
{
    Messenger::Pointer themessenger(std::make_shared<Messenger>(ioService_, messageInStream_, messageOutStream_));

    Message::Pointer message(std::make_shared<Message>(ChannelId::MEDIA_AUDIO, EncryptionType::ENCRYPTED, MessageType::SPECIFIC));
    themessenger->enqueueSend(message, std::move(sendPromise_));

    SendPromise::Pointer outStreamSendPromise;
    EXPECT_CALL(messageOutStreamMock_, stream(message, _)).Times(2).WillRepeatedly(SaveArg<1>(&outStreamSendPromise));

    ioService_.run();
    ioService_.reset();

    auto secondSendPromise = SendPromise::defer(ioService_);
    secondSendPromise->then(std::bind(&SendPromiseHandlerMock::onResolve, &sendPromiseHandlerMock_),
                           std::bind(&SendPromiseHandlerMock::onReject, &sendPromiseHandlerMock_, std::placeholders::_1));
    themessenger->enqueueSend(message, std::move(secondSendPromise));

    ioService_.run();
    ioService_.reset();

    EXPECT_CALL(sendPromiseHandlerMock_, onReject(_)).Times(0);
    EXPECT_CALL(sendPromiseHandlerMock_, onResolve()).Times(2);
    outStreamSendPromise->resolve();
    ioService_.run();
    ioService_.reset();

    outStreamSendPromise->resolve();
    ioService_.run();
}

BOOST_FIXTURE_TEST_CASE(Messenger_SendFailed, MessengerUnitTest)
{
    Messenger::Pointer themessenger(std::make_shared<Messenger>(ioService_, messageInStream_, messageOutStream_));

    Message::Pointer message(std::make_shared<Message>(ChannelId::MEDIA_AUDIO, EncryptionType::ENCRYPTED, MessageType::SPECIFIC));
    themessenger->enqueueSend(message, std::move(sendPromise_));

    SendPromise::Pointer outStreamSendPromise;
    EXPECT_CALL(messageOutStreamMock_, stream(message, _)).WillOnce(SaveArg<1>(&outStreamSendPromise));

    ioService_.run();
    ioService_.reset();

    auto secondSendPromise = SendPromise::defer(ioService_);
    secondSendPromise->then(std::bind(&SendPromiseHandlerMock::onResolve, &sendPromiseHandlerMock_),
                           std::bind(&SendPromiseHandlerMock::onReject, &sendPromiseHandlerMock_, std::placeholders::_1));
    themessenger->enqueueSend(message, std::move(secondSendPromise));

    ioService_.run();
    ioService_.reset();

    error::Error e(error::ErrorCode::USB_TRANSFER, 67);
    outStreamSendPromise->reject(e);

    EXPECT_CALL(sendPromiseHandlerMock_, onReject(e)).Times(2);
    EXPECT_CALL(sendPromiseHandlerMock_, onResolve()).Times(0);
    ioService_.run();
}

}
}
}