/***************************************************************************
 *   Copyright 2011 Andrey Batyiev <batyiev@gmail.com>                     *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or         *
 *   modify it under the terms of the GNU General Public License as        *
 *   published by the Free Software Foundation; either version 2 of        *
 *   the License or (at your option) version 3 or any later version        *
 *   accepted by the membership of KDE e.V. (or its successor approved     *
 *   by the membership of KDE e.V.), which shall act as a proxy            *
 *   defined in Section 14 of version 3 of the license.                    *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 ***************************************************************************/

#include "modelstest.h"

#include <QtTest/QtTest>
#include <qtest_kde.h>

#include <vcs/models/vcsfilechangesmodel.h>
#include <tests/autotestshell.h>
#include <tests/testcore.h>

using namespace KDevelop;

void ModelsTest::initTestCase()
{
    AutoTestShell::init();
    TestCore::initialize();
}

void ModelsTest::cleanupTestCase()
{
    TestCore::shutdown();
}

void ModelsTest::testVcsFileChangesModel()
{
    VcsFileChangesModel *model = new VcsFileChangesModel(this);

    // Newly created model should be empty
    QVERIFY(model->rowCount() == 0);

    // Pull some files into
    KUrl filenames[] = {
        KUrl::fromLocalFile("foo"),
        KUrl::fromLocalFile("bar"),
        KUrl::fromLocalFile("pew"),
        KUrl::fromLocalFile("trash")
    };
    VcsStatusInfo::State states[] = {VcsStatusInfo::ItemAdded, VcsStatusInfo::ItemModified, VcsStatusInfo::ItemDeleted, VcsStatusInfo::ItemUpToDate};
    VcsStatusInfo status;

    for(int i = 0; i < 3; i++) {
        status.setUrl(filenames[i]);
        status.setState(states[i]);
        model->updateState(status);
        QVERIFY(model->rowCount() == (i+1));
    }

    // Pulling up-to-date file doesn't change anything
    {
        status.setUrl(filenames[3]);
        status.setState(states[3]);
        model->updateState(status);
        QVERIFY(model->rowCount() == 3);
    }

    // Check that all OK
    for(int i = 0; i < 3; i++) {
        QStandardItem* item = model->fileItemForUrl(filenames[i]);
        QVERIFY(item);
        VcsStatusInfo info = VcsFileChangesModel::statusInfo(item);
        QVERIFY(info.url().isValid());
        QVERIFY(info.url() == filenames[i]);
        QVERIFY(info.state() == states[i]);
    }

    // Pull it all again = nothing changed
    for(int i = 0; i < 3; i++) {
        status.setUrl(filenames[i]);
        status.setState(states[i]);
        model->updateState(status);
        QVERIFY(model->rowCount() == 3);
    }

    // Check that all OK
    for(int i = 0; i < 3; i++) {
        QStandardItem* item = model->fileItemForUrl(filenames[i]);
        QVERIFY(item);
        VcsStatusInfo info = VcsFileChangesModel::statusInfo(item);
        QVERIFY(info.url() == filenames[i]);
        QVERIFY(info.state() == states[i]);
    }

    // Remove one file
    {
        states[1] = VcsStatusInfo::ItemUpToDate;
        status.setUrl(filenames[1]);
        status.setState(states[1]);
        model->updateState(status);
        QVERIFY(model->rowCount() == 2);
    }

    // Check them all
    for(int i = 0; i < 3; i++) {
        if(states[i] != VcsStatusInfo::ItemUpToDate && states[i] != VcsStatusInfo::ItemUnknown) {
            QStandardItem* item = model->fileItemForUrl(filenames[i]);
            QVERIFY(item);
            VcsStatusInfo info = VcsFileChangesModel::statusInfo(item);
            QVERIFY(info.url() == filenames[i]);
            QVERIFY(info.state() == states[i]);
        }
    }

    // Delete them all
    model->removeRows(0, model->rowCount());
    QVERIFY(model->rowCount() == 0);

    // Pull it all again
    for(int i = 0; i < 3; i++) {
        status.setUrl(filenames[i]);
        status.setState(states[i]);
        model->updateState(status);
    }
    QVERIFY(model->rowCount() == 2);
}

QTEST_KDEMAIN(ModelsTest, GUI);