aboutsummaryrefslogtreecommitdiff
path: root/lib/urlfilter/test/hostlist.cpp
blob: 041cd5fd83df58d7692ff24d837f9d69d171973e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include <gtest/gtest.h>
#include "hostlist/hostlist.h"
#include <QFile>

HostList *list = nullptr;

TEST(AdBlockList, Block) {
    EXPECT_EQ(list->match(QUrl(), QUrl::fromUserInput("blockeddomain.com"),     QWebEngineUrlRequestInfo::ResourceTypeMainFrame).first, UrlFilter::Block);
    EXPECT_EQ(list->match(QUrl(), QUrl::fromUserInput("blockeddomain.first"),   QWebEngineUrlRequestInfo::ResourceTypeMainFrame).first, UrlFilter::Block);
    EXPECT_EQ(list->match(QUrl(), QUrl::fromUserInput("blockeddomain.second"),  QWebEngineUrlRequestInfo::ResourceTypeMainFrame).first, UrlFilter::Block);

    const auto r = list->match(QUrl(), QUrl::fromUserInput("localhost.localdomain"), QWebEngineUrlRequestInfo::ResourceTypeMainFrame);
    EXPECT_EQ(r.first, UrlFilter::Redirect);
    EXPECT_EQ(r.second, QString("127.0.0.1"));

    EXPECT_EQ(list->match(QUrl(), QUrl::fromUserInput("other.domain"), QWebEngineUrlRequestInfo::ResourceTypeMainFrame).first, UrlFilter::NotMatched);
}

int main(int argc, char **argv) {
    QFile f("hostlist.txt");
    if(!f.open(QIODevice::ReadOnly | QIODevice::Text)) {
        qDebug("Could not open list");
        return -1;
    }

    list = new HostList(&f);
    f.close();

    qDebug("Parsed %i rules", list->ruleCount());

    testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}