/*  Copyright (C) 2013 VATSIM Community / contributors
 *  This Source Code Form is subject to the terms of the Mozilla Public
 *  License, v. 2.0. If a copy of the MPL was not distributed with this
 *  file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef BLACKCORETEST_EXPECT_H
#define BLACKCORETEST_EXPECT_H

#include <QObject>
#include <QPointer>
#include <QVector>
#include <QSet>
#include <functional>

namespace BlackCoreTest
{

class Expect;

//! Class representing a position in a source code file, for error reporting. Used in combination with the SOURCE_LOCATION macro.
class SourceLocation
{
public:
    SourceLocation(const QString& f, int ln) : file(f), line(ln) {} //!< Constructor
    const QString file; //!< Source code filename
    const int line; //!< Line number
};

//! Evaluates to an instance of SourceLocation representing the position where the macro is used.
#define SOURCE_LOCATION (BlackCoreTest::SourceLocation(__FILE__, __LINE__))

//! RAII class for Qt signal/slot connections. All connections are closed when object is destroyed.
class ConnectGuard
{
public:
    //! Constructor.
    ConnectGuard() {}

    //! Destructor.
    ~ConnectGuard() { cleanup(); }

    //! Add a connection to the object.
    ConnectGuard& operator+= (const QMetaObject::Connection& conn) { m_conns += conn; return *this; }

    //! Disconnect and remove all stored connections.
    void cleanup() { for (auto i = m_conns.begin(); i != m_conns.end(); ++i) QObject::disconnect(*i); m_conns.clear(); }

    //! \{
    //! Copying is only allowed when there are no connections stored.
    ConnectGuard(const ConnectGuard& other) { Q_ASSERT(other.m_conns.isEmpty()); }
    ConnectGuard& operator= (const ConnectGuard& other) { Q_ASSERT(other.m_conns.isEmpty()); }
    //! \}

private:
    QVector<QMetaObject::Connection> m_conns;
};

/*!
 * One unit of expectation, as returned by Expect::unit().
 *
 * Contains a sequence of signals to send, and a sequence of signals that are expected to be received in reply.
 */
class ExpectUnit
{
public:
    //! \{
    /*!
     * Adds a signal to the list of signals to send, with optional arguments.
     * \param slot A pointer-to-member-function of the subject class.
     * \return this object, so methods can be chained.
     */
    template <class F> ExpectUnit& send(F slot) { m_sends.push_back(std::bind(slot, subject<F>())); return *this; }
    template <class F, class T1> ExpectUnit& send(F slot, T1 arg1) { m_sends.push_back(std::bind(slot, subject<F>(), arg1)); return *this; }
    template <class F, class T1, class T2> ExpectUnit& send(F slot, T1 arg1, T2 arg2) { m_sends.push_back(std::bind(slot, subject<F>(), arg1, arg2)); return *this; }
    template <class F, class T1, class T2, class T3> ExpectUnit& send(F slot, T1 arg1, T2 arg2, T3 arg3) { m_sends.push_back(std::bind(slot, subject<F>(), arg1, arg2, arg3)); return *this; }
    template <class F, class T1, class T2, class T3, class T4> ExpectUnit& send(F slot, T1 arg1, T2 arg2, T3 arg3, T4 arg4) { m_sends.push_back(std::bind(slot, subject<F>(), arg1, arg2, arg3, arg4)); return *this; }
    template <class F, class T1, class T2, class T3, class T4, class T5> ExpectUnit& send(F slot, T1 arg1, T2 arg2, T3 arg3, T4 arg4, T5 arg5) { m_sends.push_back(std::bind(slot, subject<F>(), arg1, arg2, arg3, arg4, arg5)); return *this; }
    //! \}

    /*!
     * Adds a signal to the list of signals which are expects to be received.
     * \param signal A pointer-to-member-function of the subject class.
     * \param slot A callable object which will be connected to the signal, e.g. to validate the signal arguments.
     * \return this object, so methods can be chained.
     */
    template <class F1, class F2> ExpectUnit& expect(F1 signal, F2 slot)
    {
        auto subj = subject<F1>();
        m_expects.push_back([=]{
            m_guard += QObject::connect(subj, signal, slot);
            m_guard += QObject::connect(subj, signal, [=]{ next(); });
        });
        return *this;
    }

    /*!
     * Adds a signal to the list of signals which are expected to be received.
     * \param signal A pointer-to-member-function of the subject class.
     * \return this object, so methods can be chained.
     */
    template <class F> ExpectUnit& expect(F signal) { return expect(signal, []{}); }

    /*!
     * Sends all the queued signals, then waits for each of the expected signals in sequence.
     * If the full sequence of expected signals is not received, calls QTest::qFail().
     * \param timeout Time to wait in seconds. Qt event queue is processed when waiting.
     */
    void wait(int timeout);

private:
    friend class Expect;

    ExpectUnit(Expect* parent, QObject* subject, const SourceLocation& srcloc) : m_parent(parent), m_subject(subject), m_srcloc(srcloc) {}

    void init() const;
    void next() const;

    QPointer<Expect> m_parent;
    QPointer<QObject> m_subject;
    SourceLocation m_srcloc;
    QVector<std::function<void()>> m_sends;
    QVector<std::function<void()>> m_expects;
    mutable QVector<std::function<void()>>::const_iterator m_nextExpect;
    mutable ConnectGuard m_guard;

    mutable std::function<void(const ExpectUnit*)> m_onDone;
    void onDone(std::function<void(const ExpectUnit*)> callback) const { m_onDone = callback; }

    // Helper traits class. Given T is a pointer-to-member-of-U, ClassOf<T>::type is U.
    template <class T> struct ClassOf;
    template <class U, class R> struct ClassOf<R U::*> { typedef U type; };

    // Given T is a pointer-to-member-of-U, subject<T>() returns Expect's subject casted to U*.
    template <class T> typename ClassOf<T>::type* subject() const { return qobject_cast<typename ClassOf<T>::type*>(m_subject.data()); }
};

/*!
 * Class for writing unit tests in which signals are sent to some object, and other signals are expected to be received in reply.
 * Loosely modelled on the C library libexpect.
 *
 * Commonly used in combination with the macros EXPECT_UNIT or EXPECT_WAIT.
 */
class Expect : public QObject
{
    Q_OBJECT

public:
    /*!
     * Constructor.
     * \param testSubject The object to be tested.
     */
    explicit Expect(QObject* testSubject) : m_subject(testSubject), m_failed(false) {}

    /*!
     * Add a single expectation that will remain for the lifetime of the object.
     * \param signal A pointer-to-member-function of the subject class.
     * \param slot A callable object which will be connected to the signal, e.g. to validate the signal arguments.
     */
    template <class F1, class F2> void always(F1 signal, F2 slot)
    {
        m_guard += QObject::connect(m_subject, signal, slot);
    }

    /*!
     * Returns a new unit of expectation for the subject. Commonly called via the EXPECT_UNIT macro.
     * \param srcloc Represents the caller's location in the source code, for error reporting.
     */
    ExpectUnit unit(const SourceLocation& srcloc = SOURCE_LOCATION) { return ExpectUnit(this, m_subject, srcloc); }

    //! \{
    /*!
     * Allows two or more units of expectation to be waited on simultaneously. Commonly valled via the EXPECT_WAIT macro.
     * \param srcloc Represents the caller's location in the source code, for error reporting.
     * \param timeout Time to wait in seconds. Qt event queue is processed when waiting.
     */
    void wait(const SourceLocation& srcloc, int timeout, const ExpectUnit& u1) { m_units.insert(&u1); wait(srcloc, timeout); }
    void wait(const SourceLocation& srcloc, int timeout, const ExpectUnit& u1, const ExpectUnit& u2) { m_units.insert(&u1); m_units.insert(&u2); wait(srcloc, timeout); }
    void wait(const SourceLocation& srcloc, int timeout, const ExpectUnit& u1, const ExpectUnit& u2, const ExpectUnit& u3) { m_units.insert(&u1); m_units.insert(&u2); m_units.insert(&u3); wait(srcloc, timeout); }
    void wait(const SourceLocation& srcloc, int timeout, const ExpectUnit& u1, const ExpectUnit& u2, const ExpectUnit& u3, const ExpectUnit& u4) { m_units.insert(&u1); m_units.insert(&u2); m_units.insert(&u3); m_units.insert(&u4); wait(srcloc, timeout); }
    void wait(const SourceLocation& srcloc, int timeout, const ExpectUnit& u1, const ExpectUnit& u2, const ExpectUnit& u3, const ExpectUnit& u4, const ExpectUnit& u5) { m_units.insert(&u1); m_units.insert(&u2); m_units.insert(&u3); m_units.insert(&u4); m_units.insert(&u5); wait(srcloc, timeout); }
    //! \}

private:
    friend class ExpectUnit;

    void wait(const SourceLocation& srcloc, int timeout);

    QPointer<QObject> m_subject;
    QSet<const ExpectUnit*> m_units;
    ConnectGuard m_guard;
    bool m_failed;
};

/*!
 * Wrapper for Expect::unit() which fills in the source location parameter.
 * Returns a new unit of expectation for the subject.
 * \param EXP Instance of Expect on which to call unit().
 */
#define EXPECT_UNIT(EXP) ((EXP).unit(SOURCE_LOCATION))

//! \{
/*!
 * Wrapper for Expect::wait() which fills in the source location parameter.
 * Allows two or more units of expectation to be waited on simultaneously.
 * \param EXP Instance of Expect on which to call wait().
 * \param TIME Time to wait in seconds. Qt event queue is processed when waiting.
 */
#define EXPECT_WAIT_2(EXP, TIME, U1, U2) ((EXP).wait(SOURCE_LOCATION, (TIME), (U1), (U2)))
#define EXPECT_WAIT_3(EXP, TIME, U1, U2, U3) ((EXP).wait(SOURCE_LOCATION, (TIME), (U1), (U2), (U3)))
#define EXPECT_WAIT_4(EXP, TIME, U1, U2, U3, U4) ((EXP).wait(SOURCE_LOCATION, (TIME), (U1), (U2), (U3), (U4)))
#define EXPECT_WAIT_5(EXP, TIME, U1, U2, U3, U4, U5) ((EXP).wait(SOURCE_LOCATION, (TIME), (U1), (U2), (U3), (U4), (U5)))
//! \}

} //namespace BlackCoreTest

#endif //BLACKCORETEST_EXPECT_H