File indexing completed on 2024-12-22 04:52:56

 /*
  * This file is part of the KDE Akonadi Search Project
  * SPDX-FileCopyrightText: 2014 Christian Mollekopf <mollekopf@kolabsys.com>
  *
  * SPDX-License-Identifier: LGPL-2.1-only OR LGPL-3.0-only OR LicenseRef-KDE-Accepted-LGPL
  *
  */
 
 #include <Akonadi/Collection>
 #include <QDir>
 #include <QTest>
 
 #include "../lib/collectionquery.h"
 #include "../lib/resultiterator.h"
 #include "index.h"
 #include "query.h"
 #include <QDebug>
 
 Q_DECLARE_METATYPE(QSet<qint64>)
 Q_DECLARE_METATYPE(QList<qint64>)
 
 class CollectionQueryTest : public QObject
 {
     Q_OBJECT
 private:
     QString collectionsDir;
 
     bool removeDir(const QString &dirName)
     {
         bool result = true;
         QDir dir(dirName);
 
         if (dir.exists(dirName)) {
             const QFileInfoList infoDirs = dir.entryInfoList(QDir::NoDotAndDotDot | QDir::System | QDir::Hidden | QDir::AllDirs | QDir::Files, QDir::DirsFirst);
             for (const QFileInfo &info : infoDirs) {
                 if (info.isDir()) {
                     result = removeDir(info.absoluteFilePath());
                 } else {
                     result = QFile::remove(info.absoluteFilePath());
                 }
 
                 if (!result) {
                     return result;
                 }
             }
             result = dir.rmdir(dirName);
         }
         return result;
     }
 
     QString dbPrefix;
     Index index;
 
 private Q_SLOTS:
     void init()
     {
         dbPrefix = QDir::tempPath() + QLatin1StringView("/collectiontest");
         index.setOverrideDbPrefixPath(dbPrefix);
         index.createIndexers();
     }
 
     void cleanup()
     {
         removeDir(dbPrefix);
     }
 
     void searchByName()
     {
         Akonadi::Collection col1(3);
         col1.setName(QStringLiteral("col3"));
         index.index(col1);
 
         Akonadi::Collection col2(4);
         col2.setName(QStringLiteral("col4"));
         col2.setParentCollection(col1);
         index.index(col2);
 
         Akonadi::Search::PIM::CollectionQuery query;
         query.setDatabaseDir(dbPrefix + QStringLiteral("/collections/"));
         query.nameMatches(col1.name());
         Akonadi::Search::PIM::ResultIterator it = query.exec();
         QList<qint64> results;
         while (it.next()) {
             // qDebug() << "result " << it.id();
             results << it.id();
         }
         QCOMPARE(results.size(), 1);
         QCOMPARE(results.at(0), col1.id());
     }
 
     void searchHierarchy()
     {
         Akonadi::Collection col1(1);
         col1.setName(QStringLiteral("col1"));
         index.index(col1);
 
         Akonadi::Collection col2(2);
         col2.setName(QStringLiteral("col2"));
         col2.setParentCollection(col1);
         index.index(col2);
 
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.pathMatches(col1.name());
             Akonadi::Search::PIM::ResultIterator it = query.exec();
             QList<qint64> results;
             while (it.next()) {
                 // qDebug() << "result " << it.id();
                 results << it.id();
             }
             std::sort(results.begin(), results.end());
             QCOMPARE(results.size(), 2);
             QCOMPARE(results.at(0), col1.id());
             QCOMPARE(results.at(1), col2.id());
         }
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.pathMatches(QStringLiteral("/col1/col2"));
             Akonadi::Search::PIM::ResultIterator it = query.exec();
             QList<qint64> results;
             while (it.next()) {
                 // qDebug() << "result " << it.id();
                 results << it.id();
             }
             std::sort(results.begin(), results.end());
             QCOMPARE(results.size(), 1);
             QCOMPARE(results.at(0), col2.id());
         }
     }
 
     QList<qint64> getResults(Akonadi::Search::PIM::ResultIterator it)
     {
         QList<qint64> results;
         while (it.next()) {
             // qDebug() << "result " << it.id();
             results << it.id();
         }
         return results;
     }
 
     void collectionChanged()
     {
         Akonadi::Collection col1(1);
         col1.setName(QStringLiteral("col1"));
         index.index(col1);
 
         Akonadi::Collection col2(2);
         col2.setName(QStringLiteral("col2"));
         col2.setParentCollection(col1);
         index.index(col2);
 
         Akonadi::Collection col3(3);
         col3.setName(QStringLiteral("col3"));
         col3.setParentCollection(col2);
         index.index(col3);
 
         col1.setName(QStringLiteral("colX"));
         index.change(col1);
         col2.setParentCollection(col1);
         index.change(col2);
         col3.setParentCollection(col2);
         index.change(col3);
 
         // Old name gone
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.nameMatches(QStringLiteral("col1"));
             QList<qint64> results = getResults(query.exec());
             QCOMPARE(results.size(), 0);
         }
         // New name
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.nameMatches(QStringLiteral("colX"));
             QList<qint64> results = getResults(query.exec());
             QCOMPARE(results.size(), 1);
             QCOMPARE(results.at(0), col1.id());
         }
         // Old path gone
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.pathMatches(QStringLiteral("/col1/col2"));
             QList<qint64> results = getResults(query.exec());
             QCOMPARE(results.size(), 0);
         }
         // New paths
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.pathMatches(QStringLiteral("/colX/col2"));
             QList<qint64> results = getResults(query.exec());
             QCOMPARE(results.size(), 2);
             QCOMPARE(results.at(0), col2.id());
             QCOMPARE(results.at(1), col3.id());
         }
         {
             Akonadi::Search::PIM::CollectionQuery query;
             query.setDatabaseDir(dbPrefix + QLatin1StringView("/collections/"));
             query.pathMatches(QStringLiteral("/colX/col2/col3"));
             QList<qint64> results = getResults(query.exec());
             QCOMPARE(results.size(), 1);
             QCOMPARE(results.at(0), col3.id());
         }
     }
 };
 
 QTEST_GUILESS_MAIN(CollectionQueryTest)
 
 #include "collectionquerytest.moc"