/*
 * Copyright (c) 2010-2018 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#include <vector>

#include <boost/bind.hpp>

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

#include <Swiften/Base/ByteArray.h>
#include <Swiften/Client/DummyStanzaChannel.h>
#include <Swiften/FileTransfer/ByteArrayReadBytestream.h>
#include <Swiften/FileTransfer/IBBSendSession.h>
#include <Swiften/Queries/IQRouter.h>

using namespace Swift;

class IBBSendSessionTest : public CppUnit::TestFixture {
        CPPUNIT_TEST_SUITE(IBBSendSessionTest);
        CPPUNIT_TEST(testStart);
        CPPUNIT_TEST(testStart_ResponseStartsSending);
        CPPUNIT_TEST(testResponseContinuesSending);
        CPPUNIT_TEST(testRespondToAllFinishes);
        CPPUNIT_TEST(testErrorResponseFinishesWithError);
        CPPUNIT_TEST(testStopDuringSessionCloses);
        CPPUNIT_TEST(testStopAfterFinishedDoesNotClose);
        CPPUNIT_TEST(testDataStreamPauseStopsSendingData);
        CPPUNIT_TEST(testDataStreamResumeAfterPauseSendsData);
        CPPUNIT_TEST(testDataStreamResumeBeforePauseDoesNotSendData);
        CPPUNIT_TEST(testDataStreamResumeAfterResumeDoesNotSendData);

        CPPUNIT_TEST_SUITE_END();

    public:
        void setUp() {
            stanzaChannel = new DummyStanzaChannel();
            iqRouter = new IQRouter(stanzaChannel);
            bytestream = std::make_shared<ByteArrayReadBytestream>(createByteArray("abcdefg"));
            finished = false;
        }

        void tearDown() {
            delete iqRouter;
            delete stanzaChannel;
        }

        void testStart() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(1234);

            testling->start();

            CPPUNIT_ASSERT_EQUAL(1, static_cast<int>(stanzaChannel->sentStanzas.size()));
            CPPUNIT_ASSERT(stanzaChannel->isRequestAtIndex<IBB>(0, JID("foo@bar.com/baz"), IQ::Set));
            IBB::ref ibb = stanzaChannel->sentStanzas[0]->getPayload<IBB>();
            CPPUNIT_ASSERT_EQUAL(IBB::Open, ibb->getAction());
            CPPUNIT_ASSERT_EQUAL(1234u, ibb->getBlockSize());
            CPPUNIT_ASSERT_EQUAL(std::string("myid"), ibb->getStreamID());
        }

        void testStart_ResponseStartsSending() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(3);
            testling->start();

            stanzaChannel->onIQReceived(createIBBResult());

            CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(stanzaChannel->sentStanzas.size()));
            CPPUNIT_ASSERT(stanzaChannel->isRequestAtIndex<IBB>(1, JID("foo@bar.com/baz"), IQ::Set));
            IBB::ref ibb = stanzaChannel->sentStanzas[1]->getPayload<IBB>();
            CPPUNIT_ASSERT_EQUAL(IBB::Data, ibb->getAction());
            CPPUNIT_ASSERT(createByteArray("abc") == ibb->getData());
            CPPUNIT_ASSERT_EQUAL(0, ibb->getSequenceNumber());
            CPPUNIT_ASSERT_EQUAL(std::string("myid"), ibb->getStreamID());
        }

        void testResponseContinuesSending() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());

            CPPUNIT_ASSERT_EQUAL(3, static_cast<int>(stanzaChannel->sentStanzas.size()));
            CPPUNIT_ASSERT(stanzaChannel->isRequestAtIndex<IBB>(2, JID("foo@bar.com/baz"), IQ::Set));
            IBB::ref ibb = stanzaChannel->sentStanzas[2]->getPayload<IBB>();
            CPPUNIT_ASSERT_EQUAL(IBB::Data, ibb->getAction());
            CPPUNIT_ASSERT(createByteArray("def") == ibb->getData());
            CPPUNIT_ASSERT_EQUAL(1, ibb->getSequenceNumber());
            CPPUNIT_ASSERT_EQUAL(std::string("myid"), ibb->getStreamID());
        }

        void testRespondToAllFinishes() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());

            CPPUNIT_ASSERT(finished);
            CPPUNIT_ASSERT(!error);
        }

        void testErrorResponseFinishesWithError() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(IQ::createError(JID("baz@fum.com/foo"), stanzaChannel->sentStanzas[0]->getTo(), stanzaChannel->sentStanzas[0]->getID()));

            CPPUNIT_ASSERT(finished);
            CPPUNIT_ASSERT(error);
        }

        void testStopDuringSessionCloses() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(3);
            testling->start();
            testling->stop();

            CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(stanzaChannel->sentStanzas.size()));
            CPPUNIT_ASSERT(stanzaChannel->isRequestAtIndex<IBB>(1, JID("foo@bar.com/baz"), IQ::Set));
            IBB::ref ibb = stanzaChannel->sentStanzas[1]->getPayload<IBB>();
            CPPUNIT_ASSERT_EQUAL(IBB::Close, ibb->getAction());
            CPPUNIT_ASSERT_EQUAL(std::string("myid"), ibb->getStreamID());
            CPPUNIT_ASSERT(finished);
            CPPUNIT_ASSERT(!error);
        }

        void testStopAfterFinishedDoesNotClose() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            testling->setBlockSize(16);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            CPPUNIT_ASSERT(finished);

            testling->stop();

            CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(stanzaChannel->sentStanzas.size()));
        }

        void testDataStreamPauseStopsSendingData() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            bytestream->setDataComplete(false);
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());

            CPPUNIT_ASSERT(!finished);
            CPPUNIT_ASSERT(!error);

            CPPUNIT_ASSERT_EQUAL(4, static_cast<int>(stanzaChannel->sentStanzas.size()));
        }

        void testDataStreamResumeAfterPauseSendsData() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            bytestream->setDataComplete(false);
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());

            bytestream->addData(createByteArray("xyz"));

            CPPUNIT_ASSERT_EQUAL(5, static_cast<int>(stanzaChannel->sentStanzas.size()));
        }

        void testDataStreamResumeBeforePauseDoesNotSendData() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            bytestream->setDataComplete(false);
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());

            bytestream->addData(createByteArray("xyz"));

            CPPUNIT_ASSERT_EQUAL(2, static_cast<int>(stanzaChannel->sentStanzas.size()));
        }

        void testDataStreamResumeAfterResumeDoesNotSendData() {
            std::shared_ptr<IBBSendSession> testling = createSession("foo@bar.com/baz");
            bytestream->setDataComplete(false);
            testling->setBlockSize(3);
            testling->start();
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());
            stanzaChannel->onIQReceived(createIBBResult());

            bytestream->addData(createByteArray("xyz"));
            bytestream->addData(createByteArray("xuv"));

            CPPUNIT_ASSERT_EQUAL(5, static_cast<int>(stanzaChannel->sentStanzas.size()));
        }

    private:
        IQ::ref createIBBResult() {
            return IQ::createResult(JID("baz@fum.com/dum"), stanzaChannel->sentStanzas[stanzaChannel->sentStanzas.size()-1]->getTo(), stanzaChannel->sentStanzas[stanzaChannel->sentStanzas.size()-1]->getID(), std::shared_ptr<IBB>());
        }

    private:
        std::shared_ptr<IBBSendSession> createSession(const std::string& to) {
            std::shared_ptr<IBBSendSession> session(new IBBSendSession("myid", JID(), JID(to), bytestream, iqRouter));
            session->onFinished.connect(boost::bind(&IBBSendSessionTest::handleFinished, this, _1));
            return session;
        }

        void handleFinished(boost::optional<FileTransferError> error) {
            finished = true;
            this->error = error;
        }

    private:
        DummyStanzaChannel* stanzaChannel;
        IQRouter* iqRouter;
        bool finished;
        boost::optional<FileTransferError> error;
        std::shared_ptr<ByteArrayReadBytestream> bytestream;
};

CPPUNIT_TEST_SUITE_REGISTRATION(IBBSendSessionTest);