/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the test suite of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include <QtTest/QtTest>

#include <QtGui/qstringlistmodel.h>

#include <QtDeclarative/qdeclarativeview.h>

#include <QtDeclarative/qdeclarativeengine.h>

#include <QtDeclarative/qdeclarativecomponent.h>

#include <QtDeclarative/qdeclarativecontext.h>

#include <QtDeclarative/qdeclarativeexpression.h>

#include <QtDeclarative/private/qlistmodelinterface_p.h>

#include <QtDeclarative/private/qdeclarativegridview_p.h>

#include <QtDeclarative/private/qdeclarativetext_p.h>

#include <QtDeclarative/private/qdeclarativelistmodel_p.h>

#include "../../../shared/util.h"

#ifdef Q_OS_SYMBIAN

// In Symbian OS test data is located in applications private dir

#define SRCDIR "."

#endif

class tst_QDeclarativeGridView : public QObject

{

    Q_OBJECT

public:

    tst_QDeclarativeGridView();

private slots:

    void items();

    void changed();

    void inserted();

    void removed();

    void moved();

    void changeFlow();

    void currentIndex();

    void defaultValues();

    void properties();

    void propertyChanges();

    void componentChanges();

    void modelChanges();

    void positionViewAtIndex();

    void resetModel();

    void enforceRange();

    void QTBUG_8456();

    void manualHighlight();

    void footer();

private:

    QDeclarativeView *createView();

    template<typename T>

    T *findItem(QGraphicsObject *parent, const QString &id, int index=-1);

    template<typename T>

    QList<T*> findItems(QGraphicsObject *parent, const QString &objectName);

    void dumpTree(QDeclarativeItem *parent, int depth = 0);

};

class TestModel : public QAbstractListModel

{

public:

    enum Roles { Name = Qt::UserRole+1, Number = Qt::UserRole+2 };

    TestModel(QObject *parent=0) : QAbstractListModel(parent) {

        QHash<int, QByteArray> roles;

        roles[Name] = "name";

        roles[Number] = "number";

        setRoleNames(roles);

    }

    int rowCount(const QModelIndex &parent=QModelIndex()) const { Q_UNUSED(parent); return list.count(); }

    QVariant data(const QModelIndex &index, int role=Qt::DisplayRole) const {

        QVariant rv;

        if (role == Name)

            rv = list.at(index.row()).first;

        else if (role == Number)

            rv = list.at(index.row()).second;

        return rv;

    }

    int count() const { return rowCount(); }

    QString name(int index) const { return list.at(index).first; }

    QString number(int index) const { return list.at(index).second; }

    void addItem(const QString &name, const QString &number) {

        emit beginInsertRows(QModelIndex(), list.count(), list.count());

        list.append(QPair<QString,QString>(name, number));

        emit endInsertRows();

    }

    void insertItem(int index, const QString &name, const QString &number) {

        emit beginInsertRows(QModelIndex(), index, index);

        list.insert(index, QPair<QString,QString>(name, number));

        emit endInsertRows();

    }

    void removeItem(int index) {

        emit beginRemoveRows(QModelIndex(), index, index);

        list.removeAt(index);

        emit endRemoveRows();

    }

    void moveItem(int from, int to) {

        emit beginMoveRows(QModelIndex(), from, from, QModelIndex(), to);

        list.move(from, to);

        emit endMoveRows();

    }

    void modifyItem(int idx, const QString &name, const QString &number) {

        list[idx] = QPair<QString,QString>(name, number);

        emit dataChanged(index(idx,0), index(idx,0));

    }

    void clear() {

        int count = list.count();

        emit beginRemoveRows(QModelIndex(), 0, count-1);

        list.clear();

        emit endRemoveRows();

    }

private:

    QList<QPair<QString,QString> > list;

};

tst_QDeclarativeGridView::tst_QDeclarativeGridView()

{

}

void tst_QDeclarativeGridView::items()

{

    QDeclarativeView *canvas = createView();

    TestModel model;

    model.addItem("Fred", "12345");

    model.addItem("John", "2345");

    model.addItem("Bob", "54321");

    model.addItem("Billy", "22345");

    model.addItem("Sam", "2945");

    model.addItem("Ben", "04321");

    model.addItem("Jim", "0780");

    QDeclarativeContext *ctxt = canvas->rootContext();

    ctxt->setContextProperty("testModel", &model);

    ctxt->setContextProperty("testTopToBottom", QVariant(false));

    canvas->setSource(QUrl::fromLocalFile(SRCDIR "/data/gridview1.qml"));

    qApp->processEvents();

    QDeclarativeGridView *gridview = findItem<QDeclarativeGridView>(canvas->rootObject(), "grid");

    QTRY_VERIFY(gridview != 0);

    QDeclarativeItem *contentItem = gridview->contentItem();

    QTRY_VERIFY(contentItem != 0);

    QTRY_COMPARE(gridview->count(), model.count());

    QTRY_COMPARE(contentItem->childItems().count(), model.count()+1); // assumes all are visible, +1 for the (default) highlight item

    for (int i = 0; i < model.count(); ++i) {

        QDeclarativeText *name = findItem<QDeclarativeText>(contentItem, "textName", i);

        QTRY_VERIFY(name != 0);

        QTRY_COMPARE(name->text(), model.name(i));

        QDeclarativeText *number = findItem<QDeclarativeText>(contentItem, "textNumber", i);

        QTRY_VERIFY(number != 0);

        QTRY_COMPARE(number->text(), model.number(i));

    }

    // set an empty model and confirm that items are destroyed

    TestModel model2;

    ctxt->setContextProperty("testModel", &model2);

    int itemCount = findItems<QDeclarativeItem>(contentItem, "wrapper").count();

    QTRY_VERIFY(itemCount == 0);

    delete canvas;

}

void tst_QDeclarativeGridView::changed()

{

    QDeclarativeView *canvas = createView();

    TestModel model;

    model.addItem("Fred", "12345");

    model.addItem("John", "2345");

    model.addItem("Bob", "54321");

    model.addItem("Billy", "22345");

    model.addItem("Sam", "2945");

    model.addItem("Ben", "04321");

    model.addItem("Jim", "0780");

    QDeclarativeContext *ctxt = canvas->rootContext();

    ctxt->setContextProperty("testModel", &model);

    ctxt->setContextProperty("testTopToBottom", QVariant(false));

    canvas->setSource(QUrl::fromLocalFile(SRCDIR "/data/gridview1.qml"));

    qApp->processEvents();

    QDeclarativeFlickable *gridview = findItem<QDeclarativeFlickable>(canvas->rootObject(), "grid");

    QTRY_VERIFY(gridview != 0);

    QDeclarativeItem *contentItem = gridview->contentItem();

    QTRY_VERIFY(contentItem != 0);

    model.modifyItem(1, "Will", "9876");

    QDeclarativeText *name = findItem<QDeclarativeText>(contentItem, "textName", 1);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(1));

    QDeclarativeText *number = findItem<QDeclarativeText>(contentItem, "textNumber", 1);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(1));

    delete canvas;

}

void tst_QDeclarativeGridView::inserted()

{

    QDeclarativeView *canvas = createView();

    TestModel model;

    model.addItem("Fred", "12345");

    model.addItem("John", "2345");

    model.addItem("Bob", "54321");

    QDeclarativeContext *ctxt = canvas->rootContext();

    ctxt->setContextProperty("testModel", &model);

    ctxt->setContextProperty("testTopToBottom", QVariant(false));

    canvas->setSource(QUrl::fromLocalFile(SRCDIR "/data/gridview1.qml"));

    qApp->processEvents();

    QDeclarativeGridView *gridview = findItem<QDeclarativeGridView>(canvas->rootObject(), "grid");

    QTRY_VERIFY(gridview != 0);

    QDeclarativeItem *contentItem = gridview->contentItem();

    QTRY_VERIFY(contentItem != 0);

    model.insertItem(1, "Will", "9876");

    QTRY_COMPARE(contentItem->childItems().count(), model.count()+1); // assumes all are visible, +1 for the (default) highlight item

    QDeclarativeText *name = findItem<QDeclarativeText>(contentItem, "textName", 1);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(1));

    QDeclarativeText *number = findItem<QDeclarativeText>(contentItem, "textNumber", 1);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(1));

    // Checks that onAdd is called

    int added = canvas->rootObject()->property("added").toInt();

    QTRY_COMPARE(added, 1);

    // Confirm items positioned correctly

    for (int i = 0; i < model.count(); ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        QTRY_COMPARE(item->x(), (i%3)*80.0);

        QTRY_COMPARE(item->y(), (i/3)*60.0);

    }

    model.insertItem(0, "Foo", "1111"); // zero index, and current item

    QTRY_COMPARE(contentItem->childItems().count(), model.count()+1); // assumes all are visible, +1 for the (default) highlight item

    name = findItem<QDeclarativeText>(contentItem, "textName", 0);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(0));

    number = findItem<QDeclarativeText>(contentItem, "textNumber", 0);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(0));

    QTRY_COMPARE(gridview->currentIndex(), 1);

    // Confirm items positioned correctly

    for (int i = 0; i < model.count(); ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    for (int i = model.count(); i < 30; ++i)

        model.insertItem(i, "Hello", QString::number(i));

    gridview->setContentY(120);

    // Insert item outside visible area

    model.insertItem(1, "Hello", "1324");

    QTRY_VERIFY(gridview->contentY() == 120);

    delete canvas;

}

void tst_QDeclarativeGridView::removed()

{

    QDeclarativeView *canvas = createView();

    TestModel model;

    for (int i = 0; i < 40; i++)

        model.addItem("Item" + QString::number(i), "");

    QDeclarativeContext *ctxt = canvas->rootContext();

    ctxt->setContextProperty("testModel", &model);

    ctxt->setContextProperty("testTopToBottom", QVariant(false));

    canvas->setSource(QUrl::fromLocalFile(SRCDIR "/data/gridview1.qml"));

    qApp->processEvents();

    QDeclarativeGridView *gridview = findItem<QDeclarativeGridView>(canvas->rootObject(), "grid");

    QTRY_VERIFY(gridview != 0);

    QDeclarativeItem *contentItem = gridview->contentItem();

    QTRY_VERIFY(contentItem != 0);

    model.removeItem(1);

    QDeclarativeText *name = findItem<QDeclarativeText>(contentItem, "textName", 1);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(1));

    QDeclarativeText *number = findItem<QDeclarativeText>(contentItem, "textNumber", 1);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(1));

    // Checks that onRemove is called

    QString removed = canvas->rootObject()->property("removed").toString();

    QTRY_COMPARE(removed, QString("Item1"));

    // Confirm items positioned correctly

    int itemCount = findItems<QDeclarativeItem>(contentItem, "wrapper").count();

    for (int i = 0; i < model.count() && i < itemCount; ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        if (!item) qWarning() << "Item" << i << "not found";

        QTRY_VERIFY(item);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    // Remove first item (which is the current item);

    model.removeItem(0);

    name = findItem<QDeclarativeText>(contentItem, "textName", 0);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(0));

    number = findItem<QDeclarativeText>(contentItem, "textNumber", 0);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(0));

    // Confirm items positioned correctly

    itemCount = findItems<QDeclarativeItem>(contentItem, "wrapper").count();

    for (int i = 0; i < model.count() && i < itemCount; ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        if (!item) qWarning() << "Item" << i << "not found";

        QTRY_VERIFY(item);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    // Remove items not visible

    model.removeItem(25);

    // Confirm items positioned correctly

    itemCount = findItems<QDeclarativeItem>(contentItem, "wrapper").count();

    for (int i = 0; i < model.count() && i < itemCount; ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        if (!item) qWarning() << "Item" << i << "not found";

        QTRY_VERIFY(item);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    // Remove items before visible

    gridview->setContentY(120);

    gridview->setCurrentIndex(10);

    // Setting currentIndex above shouldn't cause view to scroll

    QTRY_COMPARE(gridview->contentY(), 120.0);

    model.removeItem(1);

    // Confirm items positioned correctly

    for (int i = 6; i < 18; ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        if (!item) qWarning() << "Item" << i << "not found";

        QTRY_VERIFY(item);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    // Remove currentIndex

    QDeclarativeItem *oldCurrent = gridview->currentItem();

    model.removeItem(9);

    QTRY_COMPARE(gridview->currentIndex(), 9);

    QTRY_VERIFY(gridview->currentItem() != oldCurrent);

    gridview->setContentY(0);

    // let transitions settle.

    QTest::qWait(100);

    // Confirm items positioned correctly

    itemCount = findItems<QDeclarativeItem>(contentItem, "wrapper").count();

    for (int i = 0; i < model.count() && i < itemCount; ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        if (!item) qWarning() << "Item" << i << "not found";

        QTRY_VERIFY(item);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    // remove item outside current view.

    gridview->setCurrentIndex(32);

    gridview->setContentY(240);

    model.removeItem(30);

    QTRY_VERIFY(gridview->currentIndex() == 31);

    // remove current item beyond visible items.

    gridview->setCurrentIndex(20);

    gridview->setContentY(0);

    model.removeItem(20);

    QTRY_COMPARE(gridview->currentIndex(), 20);

    QTRY_VERIFY(gridview->currentItem() != 0);

    // remove item before current, but visible

    gridview->setCurrentIndex(8);

    gridview->setContentY(240);

    oldCurrent = gridview->currentItem();

    model.removeItem(6);

    QTRY_COMPARE(gridview->currentIndex(), 7);

    QTRY_VERIFY(gridview->currentItem() == oldCurrent);

    delete canvas;

}

void tst_QDeclarativeGridView::moved()

{

    QDeclarativeView *canvas = createView();

    TestModel model;

    for (int i = 0; i < 30; i++)

        model.addItem("Item" + QString::number(i), "");

    QDeclarativeContext *ctxt = canvas->rootContext();

    ctxt->setContextProperty("testModel", &model);

    ctxt->setContextProperty("testTopToBottom", QVariant(false));

    canvas->setSource(QUrl::fromLocalFile(SRCDIR "/data/gridview1.qml"));

    qApp->processEvents();

    QDeclarativeGridView *gridview = findItem<QDeclarativeGridView>(canvas->rootObject(), "grid");

    QTRY_VERIFY(gridview != 0);

    QDeclarativeItem *contentItem = gridview->contentItem();

    QTRY_VERIFY(contentItem != 0);

    model.moveItem(1, 8);

    QDeclarativeText *name = findItem<QDeclarativeText>(contentItem, "textName", 1);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(1));

    QDeclarativeText *number = findItem<QDeclarativeText>(contentItem, "textNumber", 1);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(1));

    name = findItem<QDeclarativeText>(contentItem, "textName", 8);

    QTRY_VERIFY(name != 0);

    QTRY_COMPARE(name->text(), model.name(8));

    number = findItem<QDeclarativeText>(contentItem, "textNumber", 8);

    QTRY_VERIFY(number != 0);

    QTRY_COMPARE(number->text(), model.number(8));

    // Confirm items positioned correctly

    int itemCount = findItems<QDeclarativeItem>(contentItem, "wrapper").count();

    for (int i = 0; i < model.count() && i < itemCount; ++i) {

        QDeclarativeItem *item = findItem<QDeclarativeItem>(contentItem, "wrapper", i);

        if (!item) qWarning() << "Item" << i << "not found";

        QTRY_VERIFY(item);

        QTRY_VERIFY(item->x() == (i%3)*80);

        QTRY_VERIFY(item->y() == (i/3)*60);

    }

    gridview->setContentY(120);