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

**

** Copyright (C) 2009 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 <qcoreapplication.h>

#include <qdebug.h>

#include <qfiledialog.h>

#include <qabstractitemdelegate.h>

#include <qdirmodel.h>

#include <qitemdelegate.h>

#include <qlistview.h>

#include <qcombobox.h>

#include <qpushbutton.h>

#include <qtoolbutton.h>

#include <qtreeview.h>

#include <qheaderview.h>

#include <qcompleter.h>

#include <qaction.h>

#include <qdialogbuttonbox.h>

#include <qsortfilterproxymodel.h>

#include <qlineedit.h>

#include <qlayout.h>

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

#include "../../../src/gui/dialogs/qsidebar_p.h"

#include "../../../src/gui/dialogs/qfilesystemmodel_p.h"

#include "../../../src/gui/dialogs/qfiledialog_p.h"

#include "../network-settings.h"

//TESTED_CLASS=

//TESTED_FILES=

#if defined(Q_OS_SYMBIAN)

# define STRINGIFY(x) #x

# define TOSTRING(x) STRINGIFY(x)

# define SRCDIR "C:/Private/" TOSTRING(SYMBIAN_SRCDIR_UID) "/"

#endif

class QNonNativeFileDialog : public QFileDialog

{

    Q_OBJECT

public:

    QNonNativeFileDialog(QWidget *parent = 0, const QString &caption = QString(), const QString &directory = QString(), const QString &filter = QString())

        : QFileDialog(parent, caption, directory, filter)

    {

        setOption(QFileDialog::DontUseNativeDialog, true);

    }

};

class tst_QFiledialog : public QObject

{

Q_OBJECT

public:

    tst_QFiledialog();

    virtual ~tst_QFiledialog();

public slots:

    void init();

    void cleanup();

private slots:

    void currentChangedSignal();

    void directoryEnteredSignal();

    void filesSelectedSignal_data();

    void filesSelectedSignal();

    void filterSelectedSignal();

    void args();

    void directory();

    void completer_data();

    void completer();

    void completer_up();

    void acceptMode();

    void confirmOverwrite();

    void defaultSuffix();

    void fileMode();

    void filters();

    void history();

    void iconProvider();

    void isReadOnly();

    void itemDelegate();

    void labelText();

    void resolveSymlinks();

    void selectFile_data();

    void selectFile();

    void selectFiles();

    void selectFilter();

    void viewMode();

    void proxymodel();

    void setNameFilter();

    void focus();

    void caption();

    void historyBack();

    void historyForward();

    void disableSaveButton_data();

    void disableSaveButton();

    void saveButtonText_data();

    void saveButtonText();

    void clearLineEdit();

    void enableChooseButton();

    void hooks();

private:

    QByteArray userSettings;

};

tst_QFiledialog::tst_QFiledialog()

{

}

tst_QFiledialog::~tst_QFiledialog()

{

}

void tst_QFiledialog::init()

{

    // Save the developers settings so they don't get mad when their sidebar folders are gone.

    QSettings settings(QSettings::UserScope, QLatin1String("Trolltech"));

    settings.beginGroup(QLatin1String("Qt"));

    userSettings = settings.value(QLatin1String("filedialog")).toByteArray();

    settings.remove(QLatin1String("filedialog"));

    // populate it with some default settings

    QNonNativeFileDialog fd;

#if defined(Q_OS_WINCE)

    QTest::qWait(1000);

#endif

}

void tst_QFiledialog::cleanup()

{

    QSettings settings(QSettings::UserScope, QLatin1String("Trolltech"));

    settings.beginGroup(QLatin1String("Qt"));

    settings.setValue(QLatin1String("filedialog"), userSettings);

}

class MyAbstractItemDelegate : public QAbstractItemDelegate

{

public:

    MyAbstractItemDelegate() : QAbstractItemDelegate() {};

    void paint(QPainter *, const QStyleOptionViewItem &, const QModelIndex &) const {}

    QSize sizeHint(const QStyleOptionViewItem &, const QModelIndex &) const { return QSize(); }

};

// emited any time the selection model emits current changed

void tst_QFiledialog::currentChangedSignal()

{

    QNonNativeFileDialog fd;

    fd.setViewMode(QFileDialog::List);

    QSignalSpy spyCurrentChanged(&fd, SIGNAL(currentChanged(const QString &)));

    QListView* listView = qFindChild<QListView*>(&fd, "listView");

    QVERIFY(listView);

    fd.setDirectory(QDir::root());

    QModelIndex root = listView->rootIndex();

    QTRY_COMPARE(listView->model()->rowCount(root) > 0, true);

    QModelIndex folder;

    for (int i = 0; i < listView->model()->rowCount(root); ++i) {

        folder = listView->model()->index(i, 0, root);

        if (listView->model()->hasChildren(folder))

            break;

    }

    QVERIFY(listView->model()->hasChildren(folder));

    listView->setCurrentIndex(folder);

    QCOMPARE(spyCurrentChanged.count(), 1);

}

// only emited from the views, sidebar, or lookin combo

void tst_QFiledialog::directoryEnteredSignal()

{

#if defined QT_BUILD_INTERNAL

    QNonNativeFileDialog fd(0, "", QDir::root().path());

    fd.setOptions(QFileDialog::DontUseNativeDialog);

    fd.show();

    QTRY_COMPARE(fd.isVisible(), true);

    QSignalSpy spyDirectoryEntered(&fd, SIGNAL(directoryEntered(const QString &)));

    // sidebar

    QSidebar *sidebar = qFindChild<QSidebar*>(&fd, "sidebar");

    sidebar->setCurrentIndex(sidebar->model()->index(1, 0));

    QTest::keyPress(sidebar->viewport(), Qt::Key_Return);

    QCOMPARE(spyDirectoryEntered.count(), 1);

    spyDirectoryEntered.clear();

    // lookInCombo

    QComboBox *comboBox = qFindChild<QComboBox*>(&fd, "lookInCombo");

    comboBox->showPopup();

    QVERIFY(comboBox->view()->model()->index(1, 0).isValid());

    comboBox->view()->setCurrentIndex(comboBox->view()->model()->index(1, 0));

    QTest::keyPress(comboBox->view()->viewport(), Qt::Key_Return);

    QCOMPARE(spyDirectoryEntered.count(), 1);

    spyDirectoryEntered.clear();

    // view

    /*

    // platform specific

    fd.setViewMode(QFileDialog::ViewMode(QFileDialog::List));

    QListView* listView = qFindChild<QListView*>(&fd, "listView");

    QVERIFY(listView);

    QModelIndex root = listView->rootIndex();

    QTRY_COMPARE(listView->model()->rowCount(root) > 0, true);

    QModelIndex folder;

    for (int i = 0; i < listView->model()->rowCount(root); ++i) {

        folder = listView->model()->index(i, 0, root);

        if (listView->model()->hasChildren(folder))

            break;

    }

    QVERIFY(listView->model()->hasChildren(folder));

    listView->setCurrentIndex(folder);

    QTRY_COMPARE((listView->indexAt(listView->visualRect(folder).center())), folder);

    QTest::mouseDClick(listView->viewport(), Qt::LeftButton, 0, listView->visualRect(folder).center());

    QTRY_COMPARE(spyDirectoryEntered.count(), 1);

    */

#endif

}

Q_DECLARE_METATYPE(QFileDialog::FileMode)

void tst_QFiledialog::filesSelectedSignal_data()

{

    QTest::addColumn<QFileDialog::FileMode>("fileMode");

    QTest::newRow("any") << QFileDialog::AnyFile;

    QTest::newRow("existing") << QFileDialog::ExistingFile;

    QTest::newRow("directory") << QFileDialog::Directory;

    QTest::newRow("directoryOnly") << QFileDialog::DirectoryOnly;

    QTest::newRow("existingFiles") << QFileDialog::ExistingFiles;

}

// emited when the dialog closes with the selected files

void tst_QFiledialog::filesSelectedSignal()

{

    QNonNativeFileDialog fd;

    fd.setViewMode(QFileDialog::List);

    fd.setOptions(QFileDialog::DontUseNativeDialog);

    QDir testDir(SRCDIR"/../../..");

    fd.setDirectory(testDir);

    QFETCH(QFileDialog::FileMode, fileMode);

    fd.setFileMode(fileMode);

    QSignalSpy spyFilesSelected(&fd, SIGNAL(filesSelected(const QStringList &)));

    fd.show();

    QTest::qWait(500);

    QListView *listView = qFindChild<QListView*>(&fd, "listView");

    QVERIFY(listView);

    QModelIndex root = listView->rootIndex();

    QTRY_COMPARE(listView->model()->rowCount(root) > 0, true);

    QModelIndex file;

    for (int i = 0; i < listView->model()->rowCount(root); ++i) {

        file = listView->model()->index(i, 0, root);

        if (fileMode == QFileDialog::Directory || fileMode == QFileDialog::DirectoryOnly) {

            if (listView->model()->hasChildren(file))

                break;

        } else {

            if (!listView->model()->hasChildren(file))

                break;

        }

        file = QModelIndex();

    }

    QVERIFY(file.isValid());

    listView->selectionModel()->select(file, QItemSelectionModel::Select | QItemSelectionModel::Rows);

    listView->setCurrentIndex(file);

    QDialogButtonBox *buttonBox = qFindChild<QDialogButtonBox*>(&fd, "buttonBox");

    QPushButton *button = buttonBox->button(QDialogButtonBox::Open);

    QVERIFY(button);

    QVERIFY(button->isEnabled());

    button->animateClick();

    QTRY_COMPARE(fd.isVisible(), false);

    QCOMPARE(spyFilesSelected.count(), 1);

}

// only emited when the combo box is activated

void tst_QFiledialog::filterSelectedSignal()

{

    QNonNativeFileDialog fd;

    fd.setAcceptMode(QFileDialog::AcceptSave);

    fd.show();

    QSignalSpy spyFilterSelected(&fd, SIGNAL(filterSelected(const QString &)));

    QStringList filterChoices;

    filterChoices << "Image files (*.png *.xpm *.jpg)"

                  << "Text files (*.txt)"

                  << "Any files (*.*)";

    fd.setFilters(filterChoices);

    QCOMPARE(fd.filters(), filterChoices);

    QComboBox *filters = qFindChild<QComboBox*>(&fd, "fileTypeCombo");

    QVERIFY(filters);

    QVERIFY(filters->view());

    QCOMPARE(filters->isVisible(), true);

    QTest::keyPress(filters, Qt::Key_Down);

    QCOMPARE(spyFilterSelected.count(), 1);

}

void tst_QFiledialog::args()

{

    QWidget *parent = 0;

    QString caption = "caption";

    QString directory = QDir::tempPath();

    QString filter = "*.mp3";

    QNonNativeFileDialog fd(parent, caption, directory, filter);

    QCOMPARE(fd.parent(), (QObject *)parent);

    QCOMPARE(fd.windowTitle(), caption);

#ifndef Q_OS_WIN

    QCOMPARE(fd.directory(), QDir(directory));

#endif

    QCOMPARE(fd.filters(), QStringList(filter));

}

void tst_QFiledialog::directory()

{

    QNonNativeFileDialog fd;

    fd.setViewMode(QFileDialog::List);

    QFileSystemModel *model = qFindChild<QFileSystemModel*>(&fd, "qt_filesystem_model");

    QVERIFY(model);

    fd.setDirectory(QDir::currentPath());

    QSignalSpy spyCurrentChanged(&fd, SIGNAL(currentChanged(const QString &)));

    QSignalSpy spyDirectoryEntered(&fd, SIGNAL(directoryEntered(const QString &)));

    QSignalSpy spyFilesSelected(&fd, SIGNAL(filesSelected(const QStringList &)));

    QSignalSpy spyFilterSelected(&fd, SIGNAL(filterSelected(const QString &)));

    QCOMPARE(QDir::current().absolutePath(), fd.directory().absolutePath());

    QDir temp = QDir::temp();

    QString tempPath = temp.absolutePath();

#ifdef Q_OS_WIN

    // since the user can have lowercase temp dir, check that we are actually case-insensitive.

    tempPath = tempPath.toLower();

#endif

    fd.setDirectory(tempPath);

#ifndef Q_OS_WIN

    QCOMPARE(tempPath, fd.directory().absolutePath());

#endif

    QCOMPARE(spyCurrentChanged.count(), 0);

    QCOMPARE(spyDirectoryEntered.count(), 0);

    QCOMPARE(spyFilesSelected.count(), 0);

    QCOMPARE(spyFilterSelected.count(), 0);

    // Check my way

    QList<QListView*> list = qFindChildren<QListView*>(&fd, "listView");

    QVERIFY(list.count() > 0);

#ifdef Q_OS_WIN

    QCOMPARE(list.at(0)->rootIndex().data().toString().toLower(), temp.dirName().toLower());

#else

    QCOMPARE(list.at(0)->rootIndex().data().toString(), temp.dirName());

#endif

    QNonNativeFileDialog *dlg = new QNonNativeFileDialog(0, "", tempPath);

    QCOMPARE(model->index(tempPath), model->index(dlg->directory().absolutePath()));

    QCOMPARE(model->index(tempPath).data(QFileSystemModel::FileNameRole).toString(),

             model->index(dlg->directory().absolutePath()).data(QFileSystemModel::FileNameRole).toString());

    delete dlg;

    dlg = new QNonNativeFileDialog();

    QCOMPARE(model->index(tempPath), model->index(dlg->directory().absolutePath()));

    delete dlg;

}

void tst_QFiledialog::completer_data()

{

    QTest::addColumn<QString>("startPath");

    QTest::addColumn<QString>("input");

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

    QTest::newRow("r, 10")   << "" << "r"   << 10;

    QTest::newRow("x, 0")    << "" << "x"   << 0;

    QTest::newRow("../, -1") << "" << "../" << -1;

    QTest::newRow("goto root")     << QString()        << QDir::rootPath() << -1;

    QTest::newRow("start at root") << QDir::rootPath() << QString()        << -1;

    QDir root = QDir::root();

    QStringList list = root.entryList();

    QString folder;

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

        if (list.at(0) == QChar('.'))

            continue;

        QFileInfo info(QDir::rootPath() + list[i]);

        if (info.isDir()) {

            folder = QDir::rootPath() + list[i];

            break;

        }

    }

    QTest::newRow("start at one below root r") << folder << "r" << -1;

    QTest::newRow("start at one below root ../") << folder << "../" << -1;

}

void tst_QFiledialog::completer()

{

    QFETCH(QString, input);

    QFETCH(QString, startPath);

    QFETCH(int, expected);

    QString tempPath = QDir::tempPath() + '/' + "QFileDialogTestDir";

    if (startPath.isEmpty())

        startPath = tempPath;

    startPath = QDir::cleanPath(startPath);

    // make temp dir and files

    {

        QDir cleanup(tempPath);

        QStringList x = cleanup.entryList();

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

            QFile::remove(tempPath + '/' + x[i]);

        cleanup.rmdir(tempPath);

    }

    QDir tmp(QDir::tempPath());

    if (!tmp.exists(tempPath))

        QVERIFY(tmp.mkdir("QFileDialogTestDir"));

    QList<QTemporaryFile*> files;

    QT_TRY {

    for (int i = 0; i < 10; ++i) {

        QScopedPointer<QTemporaryFile> file(new QTemporaryFile(tempPath + "/rXXXXXX"));

        file->open();

        files.append(file.take());

    }

    // ### flesh this out more

    QNonNativeFileDialog fd(0,QString("Test it"),startPath);

    fd.setOptions(QFileDialog::DontUseNativeDialog);

    fd.show();

    QVERIFY(fd.isVisible());

    QFileSystemModel *model = qFindChild<QFileSystemModel*>(&fd, "qt_filesystem_model");

    QVERIFY(model);

    QLineEdit *lineEdit = qFindChild<QLineEdit*>(&fd, "fileNameEdit");

    QVERIFY(lineEdit);

    QCompleter *completer = lineEdit->completer();

    QVERIFY(completer);

    QAbstractItemModel *cModel = completer->completionModel();

    QVERIFY(cModel);

    //wait a bit

    QTest::qWait(500);

    // path C:\depot\qt\examples\dialogs\standarddialogs

    // files

    //       [debug] [release] [tmp] dialog dialog main makefile makefile.debug makefile.release standarddialgos

    //

    // d      -> D:\ debug dialog.cpp dialog.h

    // ..\    -> ..\classwizard ..\configdialog ..\dialogs.pro

    // c      -> C:\ control panel

    // c:     -> C:\ (nothing more)

    // C:\    -> C:\, C:\_viminfo, ...

    // \      -> \_viminfo

    // c:\depot  -> 'nothing'

    // c:\depot\ -> C:\depot\devtools, C:\depot\dteske

    QCOMPARE(model->index(fd.directory().path()), model->index(startPath));

    if (input.isEmpty()) {

        QModelIndex r = model->index(model->rootPath());

        QVERIFY(model->rowCount(r) > 0);

        QModelIndex idx = model->index(0, 0, r);

        input = idx.data().toString().at(0);

    }

    // press 'keys' for the input

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

        QTest::keyPress(lineEdit, input[i].toAscii());

    QStringList expectedFiles;

    if (expected == -1) {

        QString fullPath = startPath.isEmpty() ? tempPath : startPath;

        if (!fullPath.endsWith(QLatin1Char('/')))

            fullPath.append(QLatin1Char('/'));

        fullPath.append(input);

        bool inputStartsWithRootPath = false;

        if (input.startsWith(QDir::rootPath())) {

            fullPath = input;

            input.clear();

            inputStartsWithRootPath = true;

        }

        QFileInfo fi(fullPath);

        QDir x(fi.absolutePath());

        expectedFiles = x.entryList(model->filter());

        expected = 0;

        if (input.startsWith(".."))

            input.clear();

        for (int ii = 0; ii < expectedFiles.count(); ++ii) {

#if defined(Q_OS_WIN) || defined(Q_OS_SYMBIAN)

            if (expectedFiles.at(ii).startsWith(input,Qt::CaseInsensitive))

#else

            if (expectedFiles.at(ii).startsWith(input))