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

**

** 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 <qeventloop.h>

#include <qlist.h>

#include <qpair.h>

#include <qheaderview.h>

#include <qtablewidget.h>

//TESTED_CLASS=

//TESTED_FILES=

class QObjectTableItem : public QObject, public QTableWidgetItem

{

    Q_OBJECT

};

class tst_QTableWidget : public QObject

{

    Q_OBJECT

public:

    tst_QTableWidget();

    ~tst_QTableWidget();

public slots:

    void initTestCase();

    void cleanupTestCase();

    void init();

    void cleanup();

private slots:

    void getSetCheck();

    void clear();

    void clearContents();

    void rowCount();

    void columnCount();

    void itemAssignment();

    void item_data();

    void item();

    void takeItem_data();

    void takeItem();

    void selectedItems_data();

    void selectedItems();

    void removeRow_data();

    void removeRow();

    void removeColumn_data();

    void removeColumn();

    void insertRow_data();

    void insertRow();

    void insertColumn_data();

    void insertColumn();

    void itemStreaming_data();

    void itemStreaming();

    void itemOwnership();

    void sortItems_data();

    void sortItems();

    void setItemWithSorting_data();

    void setItemWithSorting();

    void itemData();

    void setItemData();

    void cellWidget();

    void task231094();

    void task219380_removeLastRow();

    void task262056_sortDuplicate();

private:

    QTableWidget *testWidget;

};

typedef QPair<int, int> IntPair;

typedef QList<int> IntList;

typedef QList<IntPair> IntIntList;

Q_DECLARE_METATYPE(IntList)

Q_DECLARE_METATYPE(IntIntList)

Q_DECLARE_METATYPE(QTableWidgetSelectionRange)

Q_DECLARE_METATYPE(QModelIndex)

// Testing get/set functions

void tst_QTableWidget::getSetCheck()

{

    QTableWidget obj1;

    // int QTableWidget::rowCount()

    // void QTableWidget::setRowCount(int)

    obj1.setRowCount(0);

    QCOMPARE(0, obj1.rowCount());

    obj1.setRowCount(INT_MIN);

    QCOMPARE(0, obj1.rowCount()); // Row count can never be negative

//    obj1.setRowCount(INT_MAX);

//    QCOMPARE(INT_MAX, obj1.rowCount());

    obj1.setRowCount(100);

    QCOMPARE(100, obj1.rowCount());

    // int QTableWidget::columnCount()

    // void QTableWidget::setColumnCount(int)

    obj1.setColumnCount(0);

    QCOMPARE(0, obj1.columnCount());

    obj1.setColumnCount(INT_MIN);

    QCOMPARE(0, obj1.columnCount()); // Column count can never be negative

    obj1.setColumnCount(1000);

    QCOMPARE(1000, obj1.columnCount());

//    obj1.setColumnCount(INT_MAX);

//    QCOMPARE(INT_MAX, obj1.columnCount());

    // QTableWidgetItem * QTableWidget::currentItem()

    // void QTableWidget::setCurrentItem(QTableWidgetItem *)

    QTableWidgetItem *var3 = new QTableWidgetItem("0,0");

    obj1.setItem(0, 0, var3);

    obj1.setItem(1, 1, new QTableWidgetItem("1,1"));

    obj1.setItem(2, 2, new QTableWidgetItem("2,2"));

    obj1.setItem(3, 3, new QTableWidgetItem("3,3"));

    obj1.setCurrentItem(var3);

    QCOMPARE(var3, obj1.currentItem());

    obj1.setCurrentItem((QTableWidgetItem *)0);

    QCOMPARE((QTableWidgetItem *)0, obj1.currentItem());

    obj1.setItem(0, 0, 0);

    QCOMPARE((QTableWidgetItem *)0, obj1.item(0, 0));

    // const QTableWidgetItem * QTableWidget::itemPrototype()

    // void QTableWidget::setItemPrototype(const QTableWidgetItem *)

    const QTableWidgetItem *var4 = new QTableWidgetItem;

    obj1.setItemPrototype(var4);

    QCOMPARE(var4, obj1.itemPrototype());

    obj1.setItemPrototype((QTableWidgetItem *)0);

    QCOMPARE((const QTableWidgetItem *)0, obj1.itemPrototype());

}

tst_QTableWidget::tst_QTableWidget(): testWidget(0)

{

}

tst_QTableWidget::~tst_QTableWidget()

{

}

void tst_QTableWidget::initTestCase()

{

    qRegisterMetaType<QModelIndex>("QModelIndex");

    testWidget = new QTableWidget();

    testWidget->show();

}

void tst_QTableWidget::cleanupTestCase()

{

    delete testWidget;

}

void tst_QTableWidget::init()

{

    testWidget->clear();

    testWidget->setRowCount(5);

    testWidget->setColumnCount(5);

    for (int row=0; row < testWidget->rowCount(); ++row)

        testWidget->showRow(row);

    for (int column=0; column < testWidget->columnCount(); ++column)

        testWidget->showColumn(column);

}

void tst_QTableWidget::cleanup()

{

}

void tst_QTableWidget::clearContents()

{

    QTableWidgetItem *item = new QTableWidgetItem("test");

    testWidget->setHorizontalHeaderItem(0, item);

    QVERIFY(testWidget->horizontalHeaderItem(0) == item);

    testWidget->clearContents();

    QVERIFY(testWidget->horizontalHeaderItem(0) == item);

}

void tst_QTableWidget::clear()

{

    QTableWidgetItem *item = new QTableWidgetItem("foo");

    testWidget->setItem(0, 0, item);

    testWidget->setItemSelected(item, true);

    QVERIFY(testWidget->item(0, 0) == item);

    QVERIFY(testWidget->isItemSelected(item));

    QPointer<QObjectTableItem> bla = new QObjectTableItem();

    testWidget->setItem(1, 1, bla);

    testWidget->clear();

    QVERIFY(bla.isNull());

    QVERIFY(!testWidget->item(0,0));

    QVERIFY(!testWidget->selectedRanges().count());

    QVERIFY(!testWidget->selectedItems().count());

}

void tst_QTableWidget::rowCount()

{

    int rowCountBefore = 5;

    int rowCountAfter = 10;

    int rowCount = testWidget->rowCount();

    QCOMPARE(rowCount, rowCountBefore);

    testWidget->setRowCount(rowCountAfter);

    rowCount = testWidget->rowCount();

    QCOMPARE(rowCount, rowCountAfter);

    QPersistentModelIndex index(testWidget->model()->index(rowCountAfter - 1, 0,

                                                           testWidget->rootIndex()));

    QCOMPARE(index.row(), rowCountAfter - 1);

    QCOMPARE(index.column(), 0);

    QVERIFY(index.isValid());

    testWidget->setRowCount(rowCountBefore);

    QCOMPARE(index.row(), -1);

    QCOMPARE(index.column(), -1);

    QVERIFY(!index.isValid());

    rowCountBefore = testWidget->rowCount();

    testWidget->setRowCount(-1);

    QCOMPARE(testWidget->rowCount(), rowCountBefore);

}

void tst_QTableWidget::columnCount()

{

    int columnCountBefore = 5;

    int columnCountAfter = 10;

    int columnCount = testWidget->columnCount();

    QCOMPARE(columnCount, columnCountBefore);

    testWidget->setColumnCount(columnCountAfter);

    columnCount = testWidget->columnCount();

    QCOMPARE(columnCount, columnCountAfter);

    QPersistentModelIndex index(testWidget->model()->index(0, columnCountAfter - 1,

                                                           testWidget->rootIndex()));

    QCOMPARE(index.row(), 0);

    QCOMPARE(index.column(), columnCountAfter - 1);

    QVERIFY(index.isValid());

    testWidget->setColumnCount(columnCountBefore);

    QCOMPARE(index.row(), -1);

    QCOMPARE(index.column(), -1);

    QVERIFY(!index.isValid());

    columnCountBefore = testWidget->columnCount();

    testWidget->setColumnCount(-1);

    QCOMPARE(testWidget->columnCount(), columnCountBefore);

}

void tst_QTableWidget::itemAssignment()

{

    QTableWidgetItem itemInWidget("inWidget");

    testWidget->setItem(0, 0, &itemInWidget);

    itemInWidget.setFlags(itemInWidget.flags() | Qt::ItemIsTristate);

    QTableWidgetItem itemOutsideWidget("outsideWidget");

    QVERIFY(itemInWidget.tableWidget());

    QCOMPARE(itemInWidget.text(), QString("inWidget"));

    QVERIFY(itemInWidget.flags() & Qt::ItemIsTristate);

    QVERIFY(!itemOutsideWidget.tableWidget());

    QCOMPARE(itemOutsideWidget.text(), QString("outsideWidget"));

    QVERIFY(!(itemOutsideWidget.flags() & Qt::ItemIsTristate));

    itemOutsideWidget = itemInWidget;

    QVERIFY(!itemOutsideWidget.tableWidget());

    QCOMPARE(itemOutsideWidget.text(), QString("inWidget"));

    QVERIFY(itemOutsideWidget.flags() & Qt::ItemIsTristate);

}

void tst_QTableWidget::item_data()

{

    QTest::addColumn<int>("rowCount");

    QTest::addColumn<int>("columnCount");

    QTest::addColumn<int>("row");

    QTest::addColumn<int>("column");

    QTest::addColumn<bool>("expectItem");

    QTest::newRow("0x0 take [0,0]") << 0 << 0 << 0 << 0 << false;

    QTest::newRow("0x0 take [4,4]") << 0 << 0 << 4 << 4 << false;

    QTest::newRow("4x4 take [0,0]") << 4 << 4 << 0 << 0 << true;

    QTest::newRow("4x4 take [4,4]") << 4 << 4 << 4 << 4 << false;

    QTest::newRow("4x4 take [2,2]") << 4 << 4 << 2 << 2 << true;

}

void tst_QTableWidget::item()

{

    QFETCH(int, rowCount);

    QFETCH(int, columnCount);

    QFETCH(int, row);

    QFETCH(int, column);

    QFETCH(bool, expectItem);

    testWidget->setRowCount(rowCount);

    testWidget->setColumnCount(columnCount);

    QCOMPARE(testWidget->rowCount(), rowCount);

    QCOMPARE(testWidget->columnCount(), columnCount);

    for (int r = 0; r < testWidget->rowCount(); ++r)

        for (int c = 0; c < testWidget->columnCount(); ++c)

            testWidget->setItem(r, c, new QTableWidgetItem(QString::number(r * c + c)));

    for (int r = 0; r < testWidget->rowCount(); ++r)

        for (int c = 0; c < testWidget->columnCount(); ++c)

            QCOMPARE(testWidget->item(r, c)->text(), QString::number(r * c + c));

    QTableWidgetItem *item = testWidget->item(row, column);

    QCOMPARE(!!item, expectItem);

    if (expectItem)

        QCOMPARE(item->text(), QString::number(row * column + column));

}

void tst_QTableWidget::takeItem_data()

{

    QTest::addColumn<int>("rowCount");

    QTest::addColumn<int>("columnCount");

    QTest::addColumn<int>("row");

    QTest::addColumn<int>("column");

    QTest::addColumn<bool>("expectItem");

    QTest::newRow("0x0 take [0,0]") << 0 << 0 << 0 << 0 << false;

    QTest::newRow("0x0 take [4,4]") << 0 << 0 << 4 << 4 << false;

    QTest::newRow("4x4 take [0,0]") << 4 << 4 << 0 << 0 << true;

    QTest::newRow("4x4 take [4,4]") << 4 << 4 << 4 << 4 << false;

    QTest::newRow("4x4 take [2,2]") << 4 << 4 << 2 << 2 << true;

}

void tst_QTableWidget::takeItem()

{

    QFETCH(int, rowCount);

    QFETCH(int, columnCount);

    QFETCH(int, row);

    QFETCH(int, column);

    QFETCH(bool, expectItem);

    testWidget->setRowCount(rowCount);

    testWidget->setColumnCount(columnCount);

    QCOMPARE(testWidget->rowCount(), rowCount);

    QCOMPARE(testWidget->columnCount(), columnCount);

    for (int r = 0; r < testWidget->rowCount(); ++r)

        for (int c = 0; c < testWidget->columnCount(); ++c)

            testWidget->setItem(r, c, new QTableWidgetItem(QString::number(r * c + c)));

    for (int r = 0; r < testWidget->rowCount(); ++r)

        for (int c = 0; c < testWidget->columnCount(); ++c)

            QCOMPARE(testWidget->item(r, c)->text(), QString::number(r * c + c));

    QTableWidgetItem *item = testWidget->takeItem(row, column);

    QCOMPARE(!!item, expectItem);

    if (expectItem) {

        QCOMPARE(item->text(), QString::number(row * column + column));

        delete item;

    }

    QVERIFY(!testWidget->takeItem(row, column));

}

void tst_QTableWidget::selectedItems_data()

{

    QTest::addColumn<int>("rowCount");

    QTest::addColumn<int>("columnCount");

    QTest::addColumn<IntIntList>("createItems");

    QTest::addColumn<IntList>("hiddenRows");

    QTest::addColumn<IntList>("hiddenColumns");

    QTest::addColumn<QTableWidgetSelectionRange>("selectionRange");

    QTest::addColumn<IntIntList>("expectedItems");

    QTest::newRow("3x3 empty cells, no hidden rows/columns, none selected")

        << 3 << 3

        << IntIntList()

        << IntList()

        << IntList()

        << QTableWidgetSelectionRange()

        << IntIntList();

    QTest::newRow("3x3 empty cells,no hidden rows/columnms, all selected")

        << 3 << 3

        << IntIntList()

        << IntList()

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << IntIntList();

    QTest::newRow("3x3 (1,1) exists, no hidden rows/columnms, all selected")

        << 3 << 3

        << (IntIntList() << IntPair(1,1))

        << IntList()

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << (IntIntList() << IntPair(1,1));

    QTest::newRow("3x3 (1,1) exists, row 1 hidden, all selected")

        << 3 << 3

        << (IntIntList() << IntPair(1,1))

        << (IntList() << 1)

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << IntIntList();

    QTest::newRow("3x3 (1,1) exists, column 1 hidden, all selected")

        << 3 << 3

        << (IntIntList() << IntPair(1,1))

        << IntList()

        << (IntList() << 1)

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << IntIntList();

    QTest::newRow("3x3 all exists, no hidden rows/columns, all selected")

        << 3 << 3

        << (IntIntList()

            << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

            << IntPair(1,0) << IntPair(1,1) << IntPair(1,2)

            << IntPair(2,0) << IntPair(2,1) << IntPair(2,2))

        << IntList()

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << (IntIntList()

            << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

            << IntPair(1,0) << IntPair(1,1) << IntPair(1,2)

            << IntPair(2,0) << IntPair(2,1) << IntPair(2,2));

    QTest::newRow("3x3 all exists, row 1 hidden, all selected")

        << 3 << 3

        << (IntIntList()

            << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

            << IntPair(1,0) << IntPair(1,1) << IntPair(1,2)

            << IntPair(2,0) << IntPair(2,1) << IntPair(2,2))

        << (IntList() << 1)

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << (IntIntList()

            << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

            << IntPair(2,0) << IntPair(2,1) << IntPair(2,2));

    QTest::newRow("3x3 all exists, column 1 hidden, all selected")

        << 3 << 3

        << (IntIntList()

            << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

            << IntPair(1,0) << IntPair(1,1) << IntPair(1,2)

            << IntPair(2,0) << IntPair(2,1) << IntPair(2,2))

        << IntList()

        << (IntList() << 1)

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << (IntIntList()

            << IntPair(0,0) << IntPair(0,2)

            << IntPair(1,0) << IntPair(1,2)

            << IntPair(2,0) << IntPair(2,2));

    QTest::newRow("3x3 none exists, no hidden rows/columns, all selected")

        << 3 << 3

        << IntIntList()

        << IntList()

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << IntIntList();

//         << (IntIntList()

//             << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

//             << IntPair(1,0) << IntPair(1,1) << IntPair(1,2)

//             << IntPair(2,0) << IntPair(2,1) << IntPair(2,2));

    QTest::newRow("3x3 none exists,  row 1 hidden, all selected, filling empty cells")

        << 3 << 3

        << IntIntList()

        << (IntList() << 1)

        << IntList()

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << IntIntList();

//         << (IntIntList()

//             << IntPair(0,0) << IntPair(0,1) << IntPair(0,2)

//             << IntPair(2,0) << IntPair(2,1) << IntPair(2,2));

    QTest::newRow("3x3 none exists,  column 1 hidden, all selected")

        << 3 << 3

        << IntIntList()

        << IntList()

        << (IntList() << 1)

        << QTableWidgetSelectionRange(0, 0, 2, 2)

        << IntIntList();

//         << (IntIntList()

//             << IntPair(0,0) << IntPair(0,2)

//             << IntPair(1,0) << IntPair(1,2)

//             << IntPair(2,0) << IntPair(2,2));

}

void tst_QTableWidget::selectedItems()