/***************************************************************************
 *   This file is part of awesome-widgets                                  *
 *                                                                         *
 *   awesome-widgets 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 3 of the    *
 *   License, or (at your option) any later version.                       *
 *                                                                         *
 *   awesome-widgets 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 awesome-widgets. If not, see http://www.gnu.org/licenses/  *
 ***************************************************************************/

#include "testawpatternfunctions.h"

#include <QRegularExpression>
#include <QtTest>

#include "awpatternfunctions.h"
#include "awtestlibrary.h"


void TestAWPatternFunctions::initTestCase()
{
    AWTestLibrary::init();
}


void TestAWPatternFunctions::cleanupTestCase() {}


void TestAWPatternFunctions::test_findFunctionCalls()
{
    auto name = QString("aw_%1").arg(AWTestLibrary::randomString(1, 10));
    auto code = AWTestLibrary::randomString(1, 20);
    auto args = AWTestLibrary::randomStringList(20);
    auto function = QString("$%1<%2>{{%3}}").arg(name).arg(args.join(',')).arg(code);

    auto pattern = AWTestLibrary::randomString() + function + AWTestLibrary::randomString();

    QList<AWPatternFunctions::AWFunction> found = AWPatternFunctions::findFunctionCalls(name, pattern);
    QCOMPARE(found.count(), 1);

    QCOMPARE(found.at(0).args, args);
    QCOMPARE(found.at(0).body, code);
    QCOMPARE(found.at(0).what, function);
}


void TestAWPatternFunctions::test_findKeys()
{
    auto count = AWTestLibrary::randomInt(200);
    QStringList allKeys;
    for (auto i = 0; i < count; i++) {
        auto key = AWTestLibrary::randomString(1, 20);
        while (allKeys.indexOf(QRegularExpression(QString("^%1.*").arg(key))) != -1)
            key = AWTestLibrary::randomString(1, 20);
        allKeys.append(key);
    }

    auto keys = AWTestLibrary::randomSelect(allKeys);
    auto bars = AWTestLibrary::randomSelect(allKeys);
    std::for_each(bars.begin(), bars.end(), [](auto &bar) { bar.prepend("bar"); });
    auto pattern = QString("$%1 $%2").arg(keys.join(" $")).arg(bars.join(" $"));

    allKeys.append(bars);
    allKeys.sort();
    std::reverse(allKeys.begin(), allKeys.end());
    keys.sort();
    bars.sort();

    auto foundKeys = AWPatternFunctions::findKeys(pattern, allKeys, false);
    foundKeys.sort();
    auto foundBars = AWPatternFunctions::findKeys(pattern, allKeys, true);
    foundBars.sort();

    QCOMPARE(foundKeys, keys);
    QCOMPARE(foundBars, bars);
}


void TestAWPatternFunctions::test_findLambdas()
{
    auto lambdas = AWTestLibrary::randomStringList(20);
    auto pattern
        = AWTestLibrary::randomString() + QString("${{%1}}").arg(lambdas.join("}}${{")) + AWTestLibrary::randomString();

    QCOMPARE(AWPatternFunctions::findLambdas(pattern), lambdas);
}


void TestAWPatternFunctions::test_expandTemplates()
{
    auto firstValue = AWTestLibrary::randomInt();
    auto secondValue = AWTestLibrary::randomInt();
    auto result = firstValue + secondValue;
    auto code = QString("$template{{%1+%2}}").arg(firstValue).arg(secondValue);
    auto prefix = AWTestLibrary::randomString();
    auto pattern = prefix + code;

    QCOMPARE(AWPatternFunctions::expandTemplates(pattern), QString("%1%2").arg(prefix).arg(result));
}


QTEST_MAIN(TestAWPatternFunctions);