#include "certlistparsertest.h"
#include "certificatelist.h"
#include "certificate.h"

#include <QDebug>

void CertListTest::testValidList()
{
    const char *fname = "list-valid-signed.txt";
    QFile validList(fname);
    validList.open(QIODevice::ReadOnly);
    QString validData = QString::fromLatin1(validList.readAll());
    QStringList instLines;
    QStringList remoLines;

    CertificateList *certList = testWithFile(fname);
    QCOMPARE(certList->getStatus(), Valid);
    QVERIFY(certList->isValid());

    QVERIFY(certList->rawData() == validData.toLatin1());

    const QList<Certificate> instList = certList->getInstallCertificates();
    const QList<Certificate> remoList = certList->getInstallCertificates();

    foreach (QString act, validData.split("\r\n")) {
        if (act.startsWith("I:")) {
            instLines << act;
            continue;
        }
        if (act.startsWith("R:")) {
            remoLines << act;
            continue;
        }
    }

    foreach (const Certificate& cert, instList) {
        QVERIFY(cert.isValid());
        QVERIFY(instLines.contains(cert.base64Line()));
    }
    foreach (const Certificate& cert, remoList) {
        QVERIFY(cert.isValid());
        QVERIFY(remoLines.contains(cert.base64Line()));
    }

    /* Be variable if test data changes later */
    QVERIFY(instList.size() >= 1);
    QVERIFY(remoList.size() >= 1);
    QVERIFY(instLines.size() == instList.size());
    QVERIFY(remoLines.size() == remoList.size());


    delete certList;
}

void CertListTest::testInvalidSig()
{
    const char *fnames[] = {"list-invalid-signed.txt",
        "list-valid-other-signature.txt",
        "list-valid-sha1-signature.txt",
        NULL};
    for (int i=0; fnames[i] != NULL; i++) {
        CertificateList *certList = testWithFile(fnames[i]);
        QCOMPARE (certList->getStatus(), InvalidSignature);
        delete certList;
    }
}

void verifyInvalidFile(const char *fName) {
    CertificateList *certList = new CertificateList(fName);
    QVERIFY (certList->getStatus() != Valid);
    delete certList;
}

void CertListTest::testInvalidFileNames()
{
    verifyInvalidFile("/dev/random");
    verifyInvalidFile("/tmp/");
    verifyInvalidFile(NULL);
    verifyInvalidFile("ä");
    verifyInvalidFile("💩 ");
}

void CertListTest::testEmptyFile()
{
    const char *fname = "empty_file";
    CertificateList *certList = testWithFile(fname);
    printf("Status :%i\n ",certList->getStatus());
    QCOMPARE (certList->getStatus(), UnknownError);
    delete certList;
}

void CertListTest::testGarbage()
{
    const char *fnames[] = {"garbage_500KB",
        "list-with-null.txt",
        NULL};
    for (int i=0; fnames[i] != NULL; i++) {
        CertificateList *certList = testWithFile(fnames[i]);
        QCOMPARE (certList->getStatus(), InvalidFormat);
        delete certList;
    }
}

void CertListTest::testTooLarge()
{
    const char *fname = "garbage_2MB";
    CertificateList *certList = testWithFile(fname);
    QCOMPARE (certList->getStatus(), TooLarge);
    QVERIFY (!certList->isValid());
    delete certList;
}

void CertListTest::benchmarkValid()
{
    const char *fname = "list-valid-signed.txt";

    QBENCHMARK_ONCE {
        CertificateList *certList = testWithFile(fname);
        delete certList;
    }
}

CertificateList* CertListTest::testWithFile(const char *filename)
{
    QDir dataDir = QDir(SOURCE_DIR"/data/");
    QString fileName = dataDir.absoluteFilePath(filename);
    return new CertificateList(fileName.toLocal8Bit().data());
}

int main( int argc, char **argv )
{
  CertListTest tc;
  return QTest::qExec( &tc, argc, argv );
}