#include "dynamictreemodel.h"

    void testInsertThroughProxy();

    void moveSourceItems();

}

void tst_ModelTest::testInsertThroughProxy()

{

    DynamicTreeModel *model = new DynamicTreeModel(this);

    QSortFilterProxyModel *proxy = new QSortFilterProxyModel(this);

    proxy->setSourceModel(model);

    new ModelTest(proxy, this);

    ModelInsertCommand *insertCommand = new ModelInsertCommand(model, this);

    insertCommand->setNumCols(4);

    insertCommand->setStartRow(0);

    insertCommand->setEndRow(9);

    // Parent is QModelIndex()

    insertCommand->doCommand();

    insertCommand = new ModelInsertCommand(model, this);

    insertCommand->setNumCols(4);

    insertCommand->setAncestorRowNumbers(QList<int>() << 5);

    insertCommand->setStartRow(0);

    insertCommand->setEndRow(9);

    insertCommand->doCommand();

    ModelMoveCommand *moveCommand = new ModelMoveCommand(model, this);

    moveCommand->setNumCols(4);

    moveCommand->setStartRow(0);

    moveCommand->setEndRow(0);

    moveCommand->setDestRow(9);

    moveCommand->setDestAncestors(QList<int>() << 5);

    moveCommand->doCommand();

/**

  Makes the persistent index list publicly accessible

*/

class AccessibleProxyModel : public QSortFilterProxyModel

{

    Q_OBJECT

public:

    AccessibleProxyModel(QObject *parent = 0) : QSortFilterProxyModel(parent) {}

    QModelIndexList persistent()

    {

        return persistentIndexList();

    }

};

class ObservingObject : public QObject

{

    Q_OBJECT

public:

    ObservingObject(AccessibleProxyModel  *proxy, QObject *parent = 0)

    : QObject(parent),

    m_proxy(proxy)

    {

        connect(m_proxy, SIGNAL(layoutAboutToBeChanged()), SLOT(storePersistent()));

        connect(m_proxy, SIGNAL(layoutChanged()), SLOT(checkPersistent()));

    }

public slots:

    void storePersistent(const QModelIndex &parent)

    {

        for (int row = 0; row < m_proxy->rowCount(parent); ++row) {

            QModelIndex proxyIndex = m_proxy->index(row, 0, parent);

            QModelIndex sourceIndex = m_proxy->mapToSource(proxyIndex);

            Q_ASSERT(proxyIndex.isValid());

            Q_ASSERT(sourceIndex.isValid());

            m_persistentSourceIndexes.append(sourceIndex);

            m_persistentProxyIndexes.append(proxyIndex);

            if (m_proxy->hasChildren(proxyIndex))

                storePersistent(proxyIndex);

        }

    }

    void storePersistent()

    {

        m_persistentSourceIndexes.clear();

        m_persistentProxyIndexes.clear();

        Q_ASSERT(m_proxy->persistent().isEmpty());

        storePersistent(QModelIndex());

        Q_ASSERT(!m_proxy->persistent().isEmpty());

    }

    void checkPersistent()

    {

        for (int row = 0; row < m_persistentProxyIndexes.size(); ++row) {

            QModelIndex updatedProxy = m_persistentProxyIndexes.at(row);

            QModelIndex updatedSource = m_persistentSourceIndexes.at(row);

        }

        for (int row = 0; row < m_persistentProxyIndexes.size(); ++row) {

            QModelIndex updatedProxy = m_persistentProxyIndexes.at(row);

            QModelIndex updatedSource = m_persistentSourceIndexes.at(row);

            QCOMPARE(m_proxy->mapToSource(updatedProxy), updatedSource);

        }

    }

private:

    AccessibleProxyModel  *m_proxy;

    QList<QPersistentModelIndex> m_persistentSourceIndexes;

    QList<QPersistentModelIndex> m_persistentProxyIndexes;

};

void tst_ModelTest::moveSourceItems()

{

    DynamicTreeModel *model = new DynamicTreeModel(this);

    AccessibleProxyModel *proxy = new AccessibleProxyModel(this);

    proxy->setSourceModel(model);

    ModelInsertCommand *insertCommand = new ModelInsertCommand(model, this);

    insertCommand->setStartRow(0);

    insertCommand->setEndRow(2);

    insertCommand->doCommand();

    insertCommand = new ModelInsertCommand(model, this);

    insertCommand->setAncestorRowNumbers(QList<int>() << 1);

    insertCommand->setStartRow(0);

    insertCommand->setEndRow(2);

    insertCommand->doCommand();

    ObservingObject observer(proxy);

    ModelMoveCommand *moveCommand = new ModelMoveCommand(model, this);

    moveCommand->setStartRow(0);

    moveCommand->setEndRow(0);

    moveCommand->setDestAncestors(QList<int>() << 1);

    moveCommand->setDestRow(0);

    moveCommand->doCommand();

}