File indexing completed on 2024-05-19 05:17:39

 /*
    Copyright (C) 2011 Andras Mantia <amantia@kde.org>
 
    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) any later version.
 
    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, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */
 
 #include <qtest.h>
 
 #include "kimap2test/fakeserver.h"
 #include "kimap2/session.h"
 #include "kimap2/getquotarootjob.h"
 
 #include <QtTest>
 
 Q_DECLARE_METATYPE(QList<qint64>)
 
 class QuotaRootJobTest: public QObject
 {
     Q_OBJECT
 
 private Q_SLOTS:
 
     void testGetQuotaRoot_data()
     {
         QTest::addColumn<QString>("mailbox");
         QTest::addColumn<QList<QByteArray> >("roots");
         QTest::addColumn<QList<QByteArray> >("resources");
         QTest::addColumn<QList<qint64> >("usages");
         QTest::addColumn<QList<qint64> >("limits");
         QTest::addColumn<QList<QByteArray> >("scenario");
 
         QList<QByteArray> roots;
         QList<QByteArray> resources;
         QList<qint64> usages;
         QList<qint64> limits;
         QList<QByteArray> scenario;
         roots << "";
         resources << "STORAGE";
         limits << 512;
         usages << 10;
         scenario << FakeServer::preauth()
                  << "C: A000001 GETQUOTAROOT \"INBOX\""
                  << "S: * QUOTAROOT INBOX \"\" "
                  << "S: * QUOTA \"\" (STORAGE 10 512)"
                  << "S: A000001 OK GETQUOTA completed";
         QTest::newRow("one root, one resource") << "INBOX" << roots << resources << usages << limits << scenario;
 
         roots.clear();
         resources.clear();
         usages.clear();
         limits.clear();
         scenario.clear();
         roots << "";
         resources << "STORAGE" << "MESSAGE";
         usages << 10 << 8221;
         limits << 512 << 20000;
         scenario << FakeServer::preauth()
                  << "C: A000001 GETQUOTAROOT \"INBOX\""
                  << "S: * QUOTAROOT INBOX \"\" "
                  << "S: * QUOTA \"\" (STORAGE 10 512)"
                  << "S: * QUOTA \"\" ( MESSAGE 8221 20000 ) "
                  << "S: A000001 OK GETQUOTA completed";
         QTest::newRow("one root, multiple resource") << "INBOX" << roots << resources << usages << limits << scenario;
 
         roots.clear();
         resources.clear();
         usages.clear();
         limits.clear();
         scenario.clear();
         roots << "root1" << "root2";
         resources << "STORAGE" << "MESSAGE";
         usages << 10 << 8221 << 30 << 100;
         limits << 512 << 20000 << 5124 << 120000;
         scenario << FakeServer::preauth()
                  << "C: A000001 GETQUOTAROOT \"INBOX\""
                  << "S: * QUOTAROOT INBOX \"root1\" root2 "
                  << "S: * QUOTA \"root1\" (STORAGE 10 512)"
                  << "S: * QUOTA \"root1\" ( MESSAGE 8221 20000 ) "
                  << "S: * QUOTA \"root2\" ( MESSAGE 100 120000 ) "
                  << "S: * QUOTA \"root2\" (STORAGE 30 5124)"
                  << "S: A000001 OK GETQUOTA completed";
         QTest::newRow("multiple roots, multiple resource") << "INBOX" << roots << resources << usages << limits << scenario;
 
         roots.clear();
         resources.clear();
         usages.clear();
         limits.clear();
         scenario.clear();
         roots << "";
         resources << "STORAGE" << "MESSAGE";
         usages << 10 << 8221;
         limits << 512 << 20000;
         scenario << FakeServer::preauth()
                  << "C: A000001 GETQUOTAROOT \"INBOX\""
                  << "S: * QUOTAROOT INBOX"
                  << "S: * QUOTA (STORAGE 10 512)"
                  << "S: * QUOTA ( MESSAGE 8221 20000 ) "
                  << "S: A000001 OK GETQUOTA completed";
         QTest::newRow("no rootname, multiple resource") << "INBOX" << roots << resources << usages << limits << scenario;
 
         roots.clear();
         resources.clear();
         usages.clear();
         limits.clear();
         scenario.clear();
         roots << "";
         resources << "STORAGE";
         limits << 512;
         usages << 10;
         scenario << FakeServer::preauth()
                  << "C: A000001 GETQUOTAROOT \"INBOX\""
                  << "S: * QUOTAROOT INBOX \"\" "
                  << "S: * QUOTA (STORAGE 10 512)"
                  << "S: A000001 OK GETQUOTA completed";
         QTest::newRow("no rootname in QUOTA, one resource") << "INBOX" << roots << resources << usages << limits << scenario;
 
         roots.clear();
         resources.clear();
         usages.clear();
         limits.clear();
         scenario.clear();
         roots << "";
         resources << "STORAGE";
         limits << 512;
         usages << 10;
         scenario << FakeServer::preauth()
                  << "C: A000001 GETQUOTAROOT \"INBOX\""
                  << "S: * QUOTAROOT INBOX \"root1\" "
                  << "S: * QUOTA \"root2\" (STORAGE 10 512)"
                  << "S: A000001 OK GETQUOTA completed";
         QTest::newRow("rootname missmatch") << "INBOX" << roots << resources << usages << limits << scenario;
     }
 
     void testGetQuotaRoot()
     {
         QFETCH(QString, mailbox);
         QFETCH(QList<QByteArray>, roots);
         QFETCH(QList<QByteArray>, resources);
         QFETCH(QList<qint64>, usages);
         QFETCH(QList<qint64>, limits);
         QFETCH(QList<QByteArray>, scenario);
 
         FakeServer fakeServer;
         fakeServer.setScenario(scenario);
         fakeServer.startAndWait();
 
         KIMAP2::Session session(QStringLiteral("127.0.0.1"), 5989);
 
         KIMAP2::GetQuotaRootJob *job = new KIMAP2::GetQuotaRootJob(&session);
         job->setMailBox(mailbox);
         bool result = job->exec();
         QEXPECT_FAIL("bad" , "Expected failure on BAD response", Continue);
         QEXPECT_FAIL("no" , "Expected failure on NO response", Continue);
         QVERIFY(result);
         QEXPECT_FAIL("rootname missmatch" , "Expected failure on rootname missmatch in QUOTAROOT and QUOTA response", Abort);
         QCOMPARE(job->roots(), roots);
         for (int rootIdx = 0; rootIdx < roots.size(); rootIdx++) {
             const QByteArray &root = roots[rootIdx];
             for (int i = 0; i < resources.size(); i++) {
                 int idx = i + rootIdx * roots.size();
                 QByteArray resource = resources[i];
                 QCOMPARE(job->limit(root, resource), limits[idx]);
                 QCOMPARE(job->usage(root, resource), usages[idx]);
             }
         }
 
         fakeServer.quit();
     }
 
 };
 
 QTEST_GUILESS_MAIN(QuotaRootJobTest)
 
 #include "quotarootjobtest.moc"