#include <QtTest>

#include <QString>

#include "dummyresource/resourcefactory.h"
#include "clientapi.h"
#include "synclistresult.h"
#include "commands.h"
#include "resourceconfig.h"
#include "log.h"
#include "modelresult.h"

/**
 * Test of the query system using the dummy resource.
 *
 * This test requires the dummy resource installed.
 */
class QueryTest : public QObject
{
    Q_OBJECT
private Q_SLOTS:
    void initTestCase()
    {
        Akonadi2::Log::setDebugOutputLevel(Akonadi2::Log::Trace);
        auto factory = Akonadi2::ResourceFactory::load("org.kde.dummy");
        QVERIFY(factory);
        DummyResource::removeFromDisk("org.kde.dummy.instance1");
        ResourceConfig::addResource("org.kde.dummy.instance1", "org.kde.dummy");
    }

    void cleanup()
    {
        Akonadi2::Store::shutdown(QByteArray("org.kde.dummy.instance1")).exec().waitForFinished();
        DummyResource::removeFromDisk("org.kde.dummy.instance1");
        auto factory = Akonadi2::ResourceFactory::load("org.kde.dummy");
        QVERIFY(factory);
    }

    void init()
    {
        qDebug();
        qDebug() << "-----------------------------------------";
        qDebug();
    }

    void testSingle()
    {
        //Setup
        {
            Akonadi2::ApplicationDomain::Mail mail("org.kde.dummy.instance1");
            Akonadi2::Store::create<Akonadi2::ApplicationDomain::Mail>(mail).exec().waitForFinished();
        }

        //Test
        Akonadi2::Query query;
        query.resources << "org.kde.dummy.instance1";
        query.syncOnDemand = false;
        query.processAll = false;
        query.liveQuery = true;

        //We fetch before the data is available and rely on the live query mechanism to deliver the actual data
        auto model = Akonadi2::Store::loadModel<Akonadi2::ApplicationDomain::Mail>(query);
        model->fetchMore(QModelIndex());
        QTRY_COMPARE(model->rowCount(), 1);
    }

    void testSingleWithDelay()
    {
        //Setup
        {
            Akonadi2::ApplicationDomain::Mail mail("org.kde.dummy.instance1");
            Akonadi2::Store::create<Akonadi2::ApplicationDomain::Mail>(mail).exec().waitForFinished();
        }

        //Test
        Akonadi2::Query query;
        query.resources << "org.kde.dummy.instance1";
        query.syncOnDemand = false;
        query.processAll = true;
        query.liveQuery = false;

        //We fetch after the data is available and don't rely on the live query mechanism to deliver the actual data
        auto model = Akonadi2::Store::loadModel<Akonadi2::ApplicationDomain::Mail>(query);

        //Ensure all local data is processed
        Akonadi2::Store::synchronize(query).exec().waitForFinished();

        model->fetchMore(QModelIndex());
        QVERIFY(model->rowCount() < 2);
        QTRY_COMPARE(model->rowCount(), 1);
    }

    // void testTree()
    // {
    //     //Setup
    //     {
    //         Akonadi2::ApplicationDomain::Folder folder("org.kde.dummy.instance1");
    //         Akonadi2::Store::create<Akonadi2::ApplicationDomain::Folder>(folder).exec().waitForFinished();
    //
    //         Akonadi2::Query query;
    //         query.resources << "org.kde.dummy.instance1";
    //         query.syncOnDemand = false;
    //         query.processAll = true;
    //
    //         auto model = new ModelResult<Akonadi2::ApplicationDomain::Folder>(query, QList<QByteArray>() << "summary" << "uid");
    //         QTRY_COMPARE(model->rowCount(), 1);
    //
    //         auto folderEntity = model->index(0, 0).data(ModelResult<Akonadi2::ApplicationDomain::Folder>::DomainObjectRole).value<Akonadi2::ApplicationDomain::Folder>();
    //
    //         Akonadi2::ApplicationDomain::Folder subfolder("org.kde.dummy.instance1");
    //         subfolder.setProperty("parent", folderEntity.identifier());
    //         Akonadi2::Store::create<Akonadi2::ApplicationDomain::Folder>(subfolder).exec().waitForFinished();
    //     }
    //
    //     //Test
    //     Akonadi2::Query query;
    //     query.resources << "org.kde.dummy.instance1";
    //     query.syncOnDemand = false;
    //     query.processAll = true;
    //
    //     auto model = new ModelResult<Akonadi2::ApplicationDomain::Folder>(query, QList<QByteArray>() << "summary" << "uid");
    //     QTRY_COMPARE(model->rowCount(), 1);
    //     QTRY_COMPARE(model->rowCount(model->index(0, 0)), 1);
    // }
};

QTEST_MAIN(QueryTest)
#include "querytest.moc"