From c74367d82c1c7bec393548d2e5014c794333822f Mon Sep 17 00:00:00 2001
From: Aqua-sama <aqua@iserlohn-fortress.net>
Date: Tue, 16 Oct 2018 17:25:40 +0200
Subject: urlfilter: Add FilterTree class

FilterTree is a class that holds filter rules, sorted by the domain they
are to be applied on. The rules are to follow FilterLeaf as interface.
- Add a hostlist rule format to FilterTree.
- Add a test for hostlist format.
---
 test/hostlist/hostlisttest.cpp | 42 ++++++++++++++++++++++++++++++++++++++++++
 test/hostlist/hostlisttest.h   | 19 +++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 test/hostlist/hostlisttest.cpp
 create mode 100644 test/hostlist/hostlisttest.h

(limited to 'test/hostlist')

diff --git a/test/hostlist/hostlisttest.cpp b/test/hostlist/hostlisttest.cpp
new file mode 100644
index 0000000..46f6a85
--- /dev/null
+++ b/test/hostlist/hostlisttest.cpp
@@ -0,0 +1,42 @@
+#include "hostlisttest.h"
+#include <QtTest/QtTest>
+
+void HostlistTest::parseList()
+{
+    //FilterTree tree;
+
+    // load filters
+    QFile hostlist("hostlist.txt");
+    QCOMPARE(hostlist.open(QIODevice::ReadOnly | QIODevice::Text), true);
+    QCOMPARE(loadHostlist(hostlist, &tree), true);
+
+    QCOMPARE(tree.branches().length(), 1);
+}
+
+void HostlistTest::checkRules()
+{
+    // test block
+    QVector<const FilterLeaf *> block = tree.match(QString(), "blockeddomain.com");
+    QCOMPARE(block.length(), 1);
+    QCOMPARE(block.constFirst()->action(), FilterLeaf::Block);
+
+    // test redirect
+    QVector<const FilterLeaf *> redirectResult = tree.match(QString(), "localhost.localdomain");
+    QCOMPARE(redirectResult.length(), 1);
+    QCOMPARE(redirectResult.at(0)->action(), FilterLeaf::Redirect);
+    QCOMPARE(redirectResult.at(0)->redirect(), "127.0.0.1");
+
+    // two domains on one line
+    QVector<const FilterLeaf *> blockFirst = tree.match(QString(), "blockeddomain.first");
+    QCOMPARE(blockFirst.length(), 1);
+    QCOMPARE(blockFirst.constFirst()->action(), FilterLeaf::Block);
+    QVector<const FilterLeaf *> blockSecond = tree.match(QString(), "blockeddomain.second");
+    QCOMPARE(blockSecond.length(), 1);
+    QCOMPARE(blockSecond.constFirst()->action(), FilterLeaf::Block);
+
+    // domain not on list
+    QVector<const FilterLeaf *> missing = tree.match(QString(), "other.domain");
+    QCOMPARE(missing.length(), 0);
+}
+
+QTEST_GUILESS_MAIN(HostlistTest)
diff --git a/test/hostlist/hostlisttest.h b/test/hostlist/hostlisttest.h
new file mode 100644
index 0000000..9a87e0d
--- /dev/null
+++ b/test/hostlist/hostlisttest.h
@@ -0,0 +1,19 @@
+#ifndef HOSTLIST_TEST
+#define HOSTLIST_TEST
+
+#include <QObject>
+#include "filtertree.h"
+
+class HostlistTest : public QObject
+{
+    Q_OBJECT
+
+private slots:
+    void parseList();
+    void checkRules();
+
+private:
+    FilterTree tree;
+};
+
+#endif
-- 
cgit v1.2.1