File indexing completed on 2024-04-28 04:55:54

 /* ============================================================
 * Falkon - Qt web browser
 * Copyright (C) 2018 Anmol Gautam <tarptaeya@gmail.com>
 *
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 * ============================================================ */
 #include "qmlcookiesapitest.h"
 #include "autotests.h"
 #include "mainapplication.h"
 #include "cookiejar.h"
 #include "qml/api/cookies/qmlcookie.h"
 #include <QWebEngineProfile>
 
 void QmlCookiesApiTest::initTestCase()
 {
 }
 
 void QmlCookiesApiTest::cleanupTestCase()
 {
 }
 
 void QmlCookiesApiTest::testCookieAdditionRemoval()
 {
     QSignalSpy cookieAddSpy(mApp->cookieJar(), &CookieJar::cookieAdded);
     m_testHelper.evaluate(QL1S("Falkon.Cookies.set({"
                      "    name: 'Example',"
                      "    url: '.example.com',"
                      "    expirationDate: Date.now() + 60*1000"
                      "})"));
     QTRY_COMPARE(cookieAddSpy.count(), 1);
     QNetworkCookie netCookie = qvariant_cast<QNetworkCookie>(cookieAddSpy.at(0).at(0));
     QCOMPARE(netCookie.name(), QByteArrayLiteral("Example"));
     QObject *object = m_testHelper.evaluateQObject(QSL("Falkon.Cookies"));
     QVERIFY(object);
     QSignalSpy qmlCookieSpy(object, SIGNAL(changed(QVariantMap)));
     QNetworkCookie anotherNetCookie;
     anotherNetCookie.setName(QSL("Hello").toLocal8Bit());
     anotherNetCookie.setDomain(QSL(".mydomain.com"));
     anotherNetCookie.setExpirationDate(QDateTime::currentDateTime().addSecs(60));
     mApp->webProfile()->cookieStore()->setCookie(anotherNetCookie);
     QTRY_COMPARE(qmlCookieSpy.count(), 1);
     QVariantMap addedQmlCookieMap = QVariant(qmlCookieSpy.at(0).at(0)).toMap();
     auto *addedQmlCookie = qvariant_cast<QObject*>(addedQmlCookieMap.value(QSL("cookie")));
     bool removed = addedQmlCookieMap.value(QSL("removed")).toBool();
     QCOMPARE(addedQmlCookie->property("name").toString(), QSL("Hello"));
     QCOMPARE(removed, false);
 
     mApp->webProfile()->cookieStore()->deleteCookie(netCookie);
     QTRY_COMPARE(qmlCookieSpy.count(), 2);
     QVariantMap removedQmlCookieMap = QVariant(qmlCookieSpy.at(1).at(0)).toMap();
     auto *removedQmlCookie = qvariant_cast<QObject*>(removedQmlCookieMap.value(QSL("cookie")));
     removed = removedQmlCookieMap.value(QSL("removed")).toBool();
     QCOMPARE(removedQmlCookie->property("name").toString(), QSL("Example"));
     QCOMPARE(removed, true);
 
     QSignalSpy cookieRemoveSpy(mApp->cookieJar(), &CookieJar::cookieRemoved);
     m_testHelper.evaluate(QL1S("Falkon.Cookies.remove({"
                      "    name: 'Hello',"
                      "    url: '.mydomain.com',"
                      "})"));
     QTRY_COMPARE(cookieRemoveSpy.count(), 1);
     netCookie = qvariant_cast<QNetworkCookie>(cookieRemoveSpy.at(0).at(0));
     QCOMPARE(netCookie.name(), QByteArrayLiteral("Hello"));
 }
 
 void QmlCookiesApiTest::testCookieGet()
 {
     QDateTime current = QDateTime::currentDateTime();
     QSignalSpy cookieAddSpy(mApp->cookieJar(), &CookieJar::cookieAdded);
 
     QNetworkCookie netCookie_1;
     netCookie_1.setName(QSL("Apple").toLocal8Bit());
     netCookie_1.setDomain(QSL(".apple-domain.com"));
     netCookie_1.setExpirationDate(current.addSecs(60));
     mApp->webProfile()->cookieStore()->setCookie(netCookie_1);
 
     QNetworkCookie netCookie_2;
     netCookie_2.setName(QSL("Mango").toLocal8Bit());
     netCookie_2.setDomain(QSL(".mango-domain.com"));
     netCookie_2.setExpirationDate(current.addSecs(120));
     mApp->webProfile()->cookieStore()->setCookie(netCookie_2);
 
     QNetworkCookie netCookie_3;
     netCookie_3.setName(QSL("Mango").toLocal8Bit());
     netCookie_3.setDomain(QSL(".yet-another-mango-domain.com"));
     netCookie_3.setExpirationDate(current.addSecs(180));
     mApp->webProfile()->cookieStore()->setCookie(netCookie_3);
 
     QTRY_COMPARE(cookieAddSpy.count(), 3);
 
     QNetworkCookie actualMangoCookie;
     for (const QNetworkCookie &cookie : mApp->cookieJar()->getAllCookies()) {
         if (QString::fromUtf8(cookie.name()) == QSL("Mango") && cookie.domain() == QSL(".mango-domain.com")) {
             actualMangoCookie = cookie;
         }
     }
 
     QObject *mangoCookie = m_testHelper.evaluateQObject(QL1S("Falkon.Cookies.get({"
                      "    name: 'Mango',"
                      "    url: '.mango-domain.com'"
                      "})"));
     QVERIFY(mangoCookie);
     QCOMPARE(mangoCookie->property("name").toString(), QSL("Mango"));
     QCOMPARE(mangoCookie->property("expirationDate").toDateTime(), actualMangoCookie.expirationDate());
 
     QList<QVariant> mangoCookies = m_testHelper.evaluate(QSL("Falkon.Cookies.getAll({name: 'Mango'})")).toVariant().toList();
     QCOMPARE(mangoCookies.length(), 2);
 }
 
 FALKONTEST_MAIN(QmlCookiesApiTest)